diff --git a/src/Model/AdditionalFile/AddFile.py b/src/Model/AdditionalFile/AddFile.py index afde5d59..32d5d355 100644 --- a/src/Model/AdditionalFile/AddFile.py +++ b/src/Model/AdditionalFile/AddFile.py @@ -130,6 +130,7 @@ class AddFile(SQLSubModel): if release < 8: cls._db_create(execute) + return True if 8 < release < 13: cls._db_update_to_0_0_13(execute, data) diff --git a/src/Model/Friction/Friction.py b/src/Model/Friction/Friction.py index ac8594d0..52e1b360 100644 --- a/src/Model/Friction/Friction.py +++ b/src/Model/Friction/Friction.py @@ -59,6 +59,9 @@ class Friction(SQLSubModel): ) """) + if ext == "_tmp": + return True + return cls._create_submodel(execute) @classmethod diff --git a/src/Model/Geometry/ProfileXYZ.py b/src/Model/Geometry/ProfileXYZ.py index ef4d80c9..d1b04ba0 100644 --- a/src/Model/Geometry/ProfileXYZ.py +++ b/src/Model/Geometry/ProfileXYZ.py @@ -95,6 +95,8 @@ class ProfileXYZ(Profile, SQLSubModel): if ext == "_tmp": return True + input() + return cls._create_submodel(execute) @classmethod diff --git a/src/Model/HydraulicStructures/Basic/HydraulicStructures.py b/src/Model/HydraulicStructures/Basic/HydraulicStructures.py index 8b7d0982..28e6cda7 100644 --- a/src/Model/HydraulicStructures/Basic/HydraulicStructures.py +++ b/src/Model/HydraulicStructures/Basic/HydraulicStructures.py @@ -75,6 +75,7 @@ class BasicHS(SQLSubModel): if major == minor == "0": if int(release) < 6: cls._db_create(execute) + return True else: cls._db_update_to_0_1_0(execute, data) diff --git a/src/Model/HydraulicStructures/Basic/Value.py b/src/Model/HydraulicStructures/Basic/Value.py index 60a5a0df..fd5b2727 100644 --- a/src/Model/HydraulicStructures/Basic/Value.py +++ b/src/Model/HydraulicStructures/Basic/Value.py @@ -59,6 +59,7 @@ class BHSValue(SQLSubModel): if major == minor == "0": if int(release) < 6: cls._db_create(execute) + return True else: cls._db_update_to_0_1_0(execute, data) diff --git a/src/Model/HydraulicStructures/HydraulicStructures.py b/src/Model/HydraulicStructures/HydraulicStructures.py index 6b3f6294..d29a070e 100644 --- a/src/Model/HydraulicStructures/HydraulicStructures.py +++ b/src/Model/HydraulicStructures/HydraulicStructures.py @@ -85,8 +85,9 @@ class HydraulicStructure(SQLSubModel): if rl < 6: cls._db_create(execute) + return True - if 6 < rl < 11: + if rl < 11: for v in ["input", "output"]: execute( f""" @@ -95,8 +96,7 @@ class HydraulicStructure(SQLSubModel): """ ) - if 6 < rl: - cls._db_update_to_0_1_0(execute, data) + cls._db_update_to_0_1_0(execute, data) return cls._update_submodel(execute, version, data) diff --git a/src/Model/LateralContribution/LateralContribution.py b/src/Model/LateralContribution/LateralContribution.py index 1854e69e..3d5337c6 100644 --- a/src/Model/LateralContribution/LateralContribution.py +++ b/src/Model/LateralContribution/LateralContribution.py @@ -202,6 +202,9 @@ class LateralContribution(SQLSubModel): ) """) + if ext == "_tmp": + return True + return cls._create_submodel(execute) @classmethod diff --git a/src/Model/REPLine/REPLine.py b/src/Model/REPLine/REPLine.py index 6823b646..2f6867d8 100644 --- a/src/Model/REPLine/REPLine.py +++ b/src/Model/REPLine/REPLine.py @@ -128,6 +128,7 @@ class REPLine(SQLSubModel): if major == minor == "0": if int(release) < 9: cls._db_create(execute) + return True else: cls._db_update_to_0_0_14(execute, data) diff --git a/src/Model/Reservoir/Reservoir.py b/src/Model/Reservoir/Reservoir.py index 386958cb..1b1b082b 100644 --- a/src/Model/Reservoir/Reservoir.py +++ b/src/Model/Reservoir/Reservoir.py @@ -49,6 +49,9 @@ class Data(SQLSubModel): ) """) + if ext == "_tmp": + return True + return cls._create_submodel(execute) @classmethod @@ -187,6 +190,9 @@ class Reservoir(SQLSubModel): ) """) + if ext == "_tmp": + return True + return cls._create_submodel(execute) @classmethod @@ -195,6 +201,7 @@ class Reservoir(SQLSubModel): if major == minor == "0": if int(release) < 5: cls._db_create(execute) + return True else: cls._db_update_to_0_1_0(execute, data) diff --git a/src/Model/River.py b/src/Model/River.py index bd66b031..8c06bdde 100644 --- a/src/Model/River.py +++ b/src/Model/River.py @@ -81,8 +81,10 @@ class RiverNode(Node, SQLSubModel): ) """) - cls._create_submodel(execute) - return True + if ext == "_tmp": + return True + + return cls._create_submodel(execute) @classmethod def _db_update(cls, execute, version, data=None): @@ -184,8 +186,10 @@ class RiverReach(Edge, SQLSubModel): ) """) - cls._create_submodel(execute) - return True + if ext == "_tmp": + return True + + return cls._create_submodel(execute) @classmethod def _db_update(cls, execute, version, data=None): diff --git a/src/Model/SedimentLayer/SedimentLayer.py b/src/Model/SedimentLayer/SedimentLayer.py index 9e621830..7c32ec94 100644 --- a/src/Model/SedimentLayer/SedimentLayer.py +++ b/src/Model/SedimentLayer/SedimentLayer.py @@ -125,6 +125,7 @@ class Layer(SQLSubModel): if major == minor == "0": if int(release) < 2: cls._db_create(execute) + return True else: cls._db_update_to_0_1_0(execute, data) @@ -307,6 +308,7 @@ class SedimentLayer(SQLSubModel): if major == minor == "0": if int(release) < 2: cls._db_create(execute) + return True else: cls._db_update_to_0_1_0(execute, data) diff --git a/src/SQL.py b/src/SQL.py index 39ccb8bc..b1a64e76 100644 --- a/src/SQL.py +++ b/src/SQL.py @@ -22,7 +22,7 @@ import sqlite3 from pathlib import Path -from tools import timer +from tools import timer, logger_exception logger = logging.getLogger() @@ -115,11 +115,19 @@ class SQL(object): self._db.commit() value = self._fetch(res, fetch_one) + except sqlite3.Warning as w: + logger.warning(f"SQL - {cmd}") + logger.warning(w) + except sqlite3.Error as er: + logger.error(f"SQL - {cmd}") + logger_exception(er) except Exception as e: logger_exception(e) - finally: - logger.debug(f" - {value}") - return value + except: + logger.error("ERROR") + + logger.debug(f" - {value}") + return value def _create(self): logger.warning("TODO: Create") diff --git a/src/tools.py b/src/tools.py index d7ba936d..af9592df 100644 --- a/src/tools.py +++ b/src/tools.py @@ -77,7 +77,7 @@ def logger_exception(exception): f"[{logger_color_red()}ERROR{logger_color_reset()}] " + f"{logger_color_red()}{exception}{logger_color_reset()}" ) - logger.debug( + logger.info( f"{logger_color_blue()}{exception}{logger_color_reset()}\n" + f"{logger_color_red()}{traceback.format_exc()}{logger_color_reset()}" )