GeoTIFF: Integrate to River submodel.

src/Model/GeoTIFF/GeoTIFF.py

@@ -35,6 +35,8 @@ try:
     import rasterio.sample
     import rasterio.vrt
     import rasterio._features
+
+    from rasterio.io import MemoryFile
     _rasterio_loaded = True
 except Exception as e:
     print(f"Module 'rasterio' is not available: {e}")
@@ -153,7 +155,7 @@ class GeoTIFF(SQLSubModel):
         if self._file_bytes == b'':
             return None
 
-        if self._memfile == None:
+        if self._memfile is None:
             self._memfile = MemoryFile()
             self._memfile.write(self._file_bytes)
 
@@ -202,9 +204,12 @@ class GeoTIFF(SQLSubModel):
     def _db_update(cls, execute, version, data=None):
         major, minor, release = version.strip().split(".")
 
-        if major == "0" and int(minor) <= 2:
+        if major == "0" and int(minor) < 2:
+            cls._db_create(execute)
+
+        if major == "0" and int(minor) == 2:
             if int(release) < 3:
-                cls._create_submodel(execute)
+                cls._db_create(execute)
 
         return True
 
@@ -228,6 +233,7 @@ class GeoTIFF(SQLSubModel):
             f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))})"
         )
 
+        if table is not None:
             for row in table:
                 it = iter(row)
 
@@ -287,7 +293,7 @@ class GeoTIFF(SQLSubModel):
             self.name,
             self.description,
             self.file_name,
-            self.file_bytes
+            self.file_bytes,
             self.coordinates['bottom'],
             self.coordinates['top'],
             self.coordinates['left'],

src/Model/GeoTIFF/GeoTIFFList.py

@@ -20,7 +20,7 @@ from tools import trace, timer
 
 from Model.Except import NotImplementedMethodeError
 from Model.Tools.PamhyrListExt import PamhyrModelList
-from Model.AdditionalFile.GeoTIFF import GeoTIFF
+from Model.GeoTIFF.GeoTIFF import GeoTIFF
 
 
 class GeoTIFFList(PamhyrModelList):

src/Model/River.py

@@ -59,6 +59,7 @@ from Model.LateralContributionsAdisTS.LateralContributionsAdisTSList \
     import LateralContributionsAdisTSList
 from Model.D90AdisTS.D90AdisTSList import D90AdisTSList
 from Model.DIFAdisTS.DIFAdisTSList import DIFAdisTSList
+from Model.GeoTIFF.GeoTIFFList import GeoTIFFList
 from Model.Results.Results import Results
 
 logger = logging.getLogger()
@@ -468,6 +469,7 @@ class River(Graph):
         LateralContributionsAdisTSList,
         D90AdisTSList,
         DIFAdisTSList,
+        GeoTIFFList,
         Results
     ]
 
@@ -505,6 +507,8 @@ class River(Graph):
         self._D90AdisTS = D90AdisTSList(status=self._status)
         self._DIFAdisTS = DIFAdisTSList(status=self._status)
 
+        self._geo_tiff = GeoTIFFList(status=self._status)
+
         self._results = {}
 
     @classmethod
@@ -617,6 +621,8 @@ class River(Graph):
 
         new._DIFAdisTS = DIFAdisTSList._db_load(execute, data)
 
+        new._geo_tiff = GeoTIFFList._db_load(execute, data)
+
         return new
 
     def _db_load_results(self, execute, data=None):
@@ -726,6 +732,7 @@ class River(Graph):
             self._BoundaryConditionsAdisTS,
             self._LateralContributionsAdisTS,
             self._D90AdisTS, self._DIFAdisTS,
+            self._geo_tiff,
         ]
 
         for solver in self._parameters:

src/Model/Study.py

@@ -37,7 +37,7 @@ logger = logging.getLogger()
 
 
 class Study(SQLModel):
-    _version = "0.2.2"
+    _version = "0.2.3"
 
     _sub_classes = [
         Scenario,