src/Model/BoundaryCondition/BoundaryCondition.py

@@ -329,8 +329,7 @@ class BoundaryCondition(SQLSubModel):
             return new
 
         table = execute(
-            "SELECT pamhyr_id, deleted, name, type, node, d50, sigma, " +
-            "scenario " +
+            "SELECT pamhyr_id, deleted, name, type, node, d50, sigma, scenario " +
             "FROM boundary_condition " +
             f"WHERE tab = '{tab}' " +
             f"AND scenario = {scenario.id} " +
@@ -356,14 +355,11 @@ class BoundaryCondition(SQLSubModel):
                 status=data['status'],
                 owner_scenario=owner_scenario
             )
-
             if deleted:
                 bc.set_as_deleted()
-
-            if t == "SL":
+            if t=="SL":
                 bc.d50 = d50
                 bc.sigma = sigma
-
             bc.node = None
             if node != -1:
                 bc.node = next(filter(lambda n: n.id == node, nodes), None)
@@ -395,13 +391,13 @@ class BoundaryCondition(SQLSubModel):
         node = -1
         if self._node is not None:
             node = self._node.id
-
+        
         d50 = 0.002
         sigma = 1
         if self._type == "SL":
             d50 = self._d50
             sigma = self._sigma
-
+        
         execute(
             "INSERT INTO " +
             "boundary_condition(" +
@@ -495,7 +491,7 @@ class BoundaryCondition(SQLSubModel):
 
     def has_node(self):
         return self._node is not None
-
+    
     @property
     def d50(self):
         return self._d50

src/Model/BoundaryConditionsAdisTS/BoundaryConditionAdisTS.py

@@ -302,8 +302,7 @@ class BoundaryConditionAdisTS(SQLSubModel):
 
             if pol_id not in pid_pol:
                 # ⚠️ cas important : probablement déjà migré
-                print(f"[WARN] pol_id {pol_id} not in pid_pol " +
-                      f"→ probably already migrated")
+                print(f"[WARN] pol_id {pol_id} not in pid_pol → probably already migrated")
                 continue
 
             execute(

src/Model/LateralContributionsAdisTS/LateralContributionAdisTS.py

@@ -132,13 +132,11 @@ class Data(SQLSubModel):
         if scenario is None:
             return new
 
-        lca = data["lca"]
-
         table = execute(
             "SELECT pamhyr_id, deleted, data0, data1 " +
             "FROM lateral_contribution_data_adists " +
             f"WHERE scenario = {scenario.id} " +
-            f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))}) " +
+            f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))}) "
             f"AND lca = '{lca.id}'"
         )
 
@@ -374,12 +372,12 @@ class LateralContributionAdisTS(SQLSubModel):
 
         execute(
             "INSERT INTO " +
-            "lateral_contribution_adists(pamhyr_id, deleted, " +
+            "lateral_contribution_adists(pamhyr_id, deleted," +
             "pollutant, reach, begin_rk, end_rk, scenario) " +
             "VALUES (" +
             f"{self.id}, {self._db_format(self.is_deleted())}, " +
             f"{self._pollutant}, {self.reach}, " +
-            f"{self._begin_rk}, {self._end_rk}, " +
+            f"{self._begin_rk}, {self._end_rk}" +
             f"{self._status.scenario_id}" +
             ")"
         )

src/Model/Pollutants/Pollutants.py

@@ -28,9 +28,7 @@ from tools import (
 from Model.Tools.PamhyrDB import SQLSubModel
 from Model.Except import NotImplementedMethodeError
 from Model.Scenario import Scenario
-from Model.BoundaryConditionsAdisTS.BoundaryConditionAdisTS import (
-    BoundaryConditionAdisTS,
-)
+from Model.BoundaryConditionsAdisTS.BoundaryConditionAdisTS import BoundaryConditionAdisTS
 
 logger = logging.getLogger()
 
@@ -424,12 +422,9 @@ class Pollutants(SQLSubModel):
                 new_pollutant._data = PollutantCharacteristics._db_load(
                     execute, data=data
                 )
-
-                new_pollutant._boundary_conditions_adists = (
-                    BoundaryConditionAdisTS._db_load(
-                        execute,
-                        data=data
-                    )
+                
+                new_pollutant._boundary_conditions_adists = BoundaryConditionAdisTS._db_load(
+                    execute, data=data
                 )
 
                 loaded.add(pid)

src/Model/Tools/PamhyrDB.py

@@ -313,8 +313,7 @@ class SQLSubModel(PamhyrID):
 
             if node_id not in nodes:
                 # ⚠️ cas important : probablement déjà migré
-                print(f"[WARN] node_id {node_id} not in nodes " +
-                      f"→ probably already migrated")
+                print(f"[WARN] node_id {node_id} not in nodes → probably already migrated")
                 continue
 
             execute(

src/View/BoundaryConditionsAdisTS/Table.py

@@ -112,8 +112,7 @@ class ComboBoxDelegate(QItemDelegate):
 
 
 class TableModel(PamhyrTableModel):
-    def __init__(self, bc_list=None, pollutant_bc_list=None,
-                 trad=None, **kwargs):
+    def __init__(self, bc_list=None, pollutant_bc_list=None, trad=None, **kwargs):
         self._trad = trad
         self._bc_list = bc_list
         self._pollutant = pollutant_bc_list.id

src/View/BoundaryConditionsAdisTS/Window.py

@@ -58,18 +58,17 @@ class BoundaryConditionAdisTSWindow(PamhyrWindow):
     _pamhyr_ui = "BoundaryConditionsAdisTS"
     _pamhyr_name = "Boundary conditions AdisTS"
 
-    def __init__(self, data=None, pollutant_id=None, study=None,
-                 config=None, parent=None):
+    def __init__(self, data=None, pollutant_id=None, study=None, config=None, parent=None):
         self._data = data
         self._pollutant_id = pollutant_id
-        trad = BCAdisTSTranslate()
 
+        _pollutants_lst = study._river._Pollutants.Pollutants_List
         self._pollutant_name = next(
-            (x.name for x in study._river._Pollutants.Pollutants_List
-             if x.id == self._pollutant_id),
+            (x.name for x in _pollutants_lst if x.id == self._pollutant_id),
             None
         )
 
+        trad = BCAdisTSTranslate()
         name = (
             trad[self._pamhyr_name] +
             " - " + study.name +

src/View/D90AdisTS/Table.py

@@ -120,7 +120,7 @@ class D90TableModel(PamhyrTableModel):
     def _setup_lst(self):
         self._lst = list(
             filter(
-                lambda d90: d90._deleted is False,
+                lambda d90: d90._deleted == False,
                 self._data._data
             )
         )

src/View/Geometry/ShiftDialog.py

@@ -20,7 +20,6 @@ from View.Tools.PamhyrWindow import PamhyrDialog
 from PyQt5.QtWidgets import QDoubleSpinBox
 from View.Tools.FlexibleDoubleSpinBox import FlexibleDoubleSpinBox
 
-
 class ShiftDialog(PamhyrDialog):
     _pamhyr_ui = "GeometryReachShift"
     _pamhyr_name = "Shift"
@@ -32,7 +31,7 @@ class ShiftDialog(PamhyrDialog):
             options=[],
             parent=parent
         )
-
+        
         self._replace_spinboxes()
         self._init_default_values()
 

src/View/Geometry/Translate.py

@@ -91,9 +91,3 @@ class GeometryTranslate(MainTranslate):
         self._dict["Shift"] = _translate(
             "Geometry", "Shift"
         )
-        self._dict["warning"] = _translate(
-            "Geometry", "Warning"
-        )
-        self._dict["format_not_exportable"] = _translate(
-            "Geometry", "The format of the file is not exportable."
-        )

src/View/Geometry/Window.py

@@ -691,13 +691,7 @@ class GeometryWindow(PamhyrWindow):
                 if suffix == ".st" or suffix == ".ST":
                     self._export_to_file_st(filename[:-3])
                 else:
-                    # Warning popup when the format is not exportable
-                    win = QtWidgets.QMessageBox()
-                    win.setIcon(QtWidgets.QMessageBox.Warning)
-                    win.setWindowTitle(self._trad["warning"])
-                    win.setText(self._trad["format_not_exportable"])
-                    win.exec()
-
+                    # TODO : prévenir l'utilisateur que le format n'est pas exportable
                     pass
             else:
                 self._export_to_file_st(filename)

src/View/InitialConditionsAdisTS/Table.py

@@ -119,7 +119,7 @@ class InitialConditionTableModel(PamhyrTableModel):
     def _setup_lst(self):
         self._lst = list(
             filter(
-                lambda ica: ica._deleted is False,
+                lambda ica: ica._deleted == False,
                 self._data._data
             )
         )

src/View/InitialConditionsAdisTS/Window.py

@@ -68,22 +68,15 @@ class InitialConditionsAdisTSWindow(PamhyrWindow):
     _pamhyr_ui = "InitialConditionsAdisTS"
     _pamhyr_name = "Initial condition AdisTS"
 
-    def __init__(self, data=None, pollutant_id=None, study=None,
-                 config=None, parent=None):
+    def __init__(self, data=None, study=None, config=None, parent=None):
         self._data = []
         self._data.append(data)
-        self._pollutant_id = pollutant_id
         trad = IcAdisTSTranslate()
 
-        self._pollutant_name = next(
-            (x.name for x in study._river._Pollutants.Pollutants_List
-             if x.id == self._pollutant_id),
-            None
-        )
         name = (
             trad[self._pamhyr_name] +
             " - " + study.name +
-            " - " + self._pollutant_name
+            " - " + self._data[0].name
         )
 
         super(InitialConditionsAdisTSWindow, self).__init__(

src/View/LateralContributionsAdisTS/Table.py

@@ -116,22 +116,10 @@ class TableModel(PamhyrTableModel):
         self._pollutant = pollutant
 
         super(TableModel, self).__init__(trad=trad, **kwargs)
-        self._setup_lst()
 
     def _setup_lst(self):
-        if self._lcs_list is not None:
-            self._lcs_pol_list = [
-                lcs for lcs in self._lcs_list.lst
-                if lcs.pollutant == self._pollutant
-            ]
-
-        self._lst = list(
-            filter(
-                lambda x: x._deleted is False,
-                self._lcs_pol_list
-            )
-        )
-        # self._tab = self._opt_data
+        self._lst = self._data.lateral_contributions_adists.lst
+        self._tab = self._opt_data
         self._long_types = self._trad.get_dict("long_types")
 
     def rowCount(self, parent):
@@ -202,43 +190,17 @@ class TableModel(PamhyrTableModel):
             )
         )
 
-        self._setup_lst()
         self.endInsertRows()
         self.layoutChanged.emit()
 
     def delete(self, rows, parent=QModelIndex()):
         self.beginRemoveRows(parent, rows[0], rows[-1])
 
-        global_rows = list(
-            map(self._global_row, rows)
-        )
         self._undo.push(
             DelCommand(
-                self._lcs_list, global_rows
+                self._lst, rows
             )
         )
 
-        self._setup_lst()
         self.endRemoveRows()
         self.layoutChanged.emit()
-
-    def undo(self):
-        self._undo.undo()
-        self._setup_lst()
-        self.layoutChanged.emit()
-
-    def redo(self):
-        self._undo.redo()
-        self._setup_lst()
-        self.layoutChanged.emit()
-
-    def get(self, row):
-        if row < 0 or row >= len(self._lst):
-            return None
-        return self._lst[row]
-
-    def _global_row(self, row):
-        bc = self.get(row)
-        if bc is None:
-            return None
-        return self._lcs_list.index(bc)

src/View/LateralContributionsAdisTS/UndoCommand.py

@@ -91,13 +91,13 @@ class AddCommand(QUndoCommand):
         self._new = None
 
     def undo(self):
-        self._lcs.delete_i(self._index)
+        del self._lcs[self._index]
 
     def redo(self):
         if self._new is None:
             self._new = self._lcs.new(self._index, self._pollutant)
         else:
-            self._lcs.insert(self._index, self._new)
+            self._lcs_lst.insert(self._index, self._new)
 
 
 class DelCommand(QUndoCommand):
@@ -107,14 +107,15 @@ class DelCommand(QUndoCommand):
         self._lcs = lcs
         self._rows = rows
 
-        self._lc = []
+        self._bc = []
         for row in rows:
-            self._lc.append((row, self._lcs.get(row)))
-        self._lc.sort()
+            self._bc.append((row, self._lcs[row]))
+        self._bc.sort()
 
     def undo(self):
-        for row, el in self._lc:
+        for row, el in self._bc:
             self._lcs.insert(row, el)
 
     def redo(self):
-        self._lcs.delete_i(self._rows)
+        for row in self._rows:
+            del self._lcs[row]

src/View/LateralContributionsAdisTS/Window.py

@@ -57,24 +57,11 @@ class LateralContributionAdisTSWindow(PamhyrWindow):
     _pamhyr_ui = "LateralContributionsAdisTS"
     _pamhyr_name = "Lateral contribution AdisTS"
 
-    def __init__(self, data=None, study=None, pollutant_id=None,
-                 config=None, parent=None):
-        self._pollutant = pollutant_id
-        self._study = study
-        self._data = data
+    def __init__(self, study=None, pollutant=None, config=None, parent=None):
         trad = LCTranslate()
+        name = trad[self._pamhyr_name] + " - " + study.name
 
-        self._pollutant_name = next(
-            (x.name for x in study._river._Pollutants.Pollutants_List
-             if x.id == self._pollutant),
-            None
-        )
-
-        name = (
-            trad[self._pamhyr_name] +
-            " - " + study.name +
-            " - " + self._pollutant_name
-        )
+        self._pollutant = pollutant
 
         super(LateralContributionAdisTSWindow, self).__init__(
             title=name,
@@ -123,7 +110,7 @@ class LateralContributionAdisTSWindow(PamhyrWindow):
             data=self._study.river,
             undo=self._undo_stack,
             trad=self._trad,
-            # opt_data="liquid",
+            opt_data="liquid",
             pollutant=self._pollutant,
             lcs_list=self._lcs,
         )

src/View/Pollutants/Window.py

@@ -217,8 +217,7 @@ class PollutantsWindow(PamhyrWindow):
             initial = InitialConditionsAdisTSWindow(
                 study=self._study,
                 parent=self,
-                data=ics_adists,
-                pollutant_id=pollutant_id
+                data=ics_adists
             )
             initial.show()
 
@@ -230,8 +229,7 @@ class PollutantsWindow(PamhyrWindow):
         for row in rows:
             pollutant_id = self._pollutants_lst.get(row).id
 
-            river = self._study.river
-            bclist = river.boundary_conditions_adists.BCs_AdisTS_List
+            bclist = self._study.river.boundary_conditions_adists.BCs_AdisTS_List
             bcs_adists = [
                 x for x in bclist
                 if x.pollutant == pollutant_id
@@ -258,9 +256,6 @@ class PollutantsWindow(PamhyrWindow):
 
         pollutant_id = self._pollutants_lst.get(rows[0]).id
 
-        lclist = self._study.river.lateral_contributions_adists.Lat_Cont_List
-        lcs_adists = [x for x in lclist if x.pollutant == pollutant_id]
-
         if self.sub_window_exists(
             LateralContributionAdisTSWindow,
             data=[self._study, pollutant_id, None]
@@ -269,9 +264,8 @@ class PollutantsWindow(PamhyrWindow):
 
         lateral = LateralContributionAdisTSWindow(
             study=self._study,
-            parent=self,
-            data=lcs_adists,
-            pollutant_id=pollutant_id,
+            pollutant=pollutant_id,
+            parent=self
         )
         lateral.show()
 

src/View/Tools/FlexibleDoubleSpinBox.py

@@ -23,7 +23,7 @@ from PyQt5.QtWidgets import QDoubleSpinBox
 
 
 class FlexibleDoubleSpinBox(QDoubleSpinBox):
-
+    
     def keyPressEvent(self, event):
         if event.text() == ".":
             # Simule une virgule à la place du point
@@ -34,4 +34,4 @@ class FlexibleDoubleSpinBox(QDoubleSpinBox):
                 ","
             )
 
-        super().keyPressEvent(event)
+        super().keyPressEvent(event)