qgepdss/export.py - rohrprofil_geometry etc. add if filtered

qgepqwat2ili/gui/__init__.py

@@ -440,7 +440,8 @@ def action_do_export():
                 return
 
         # 2. check wastewater_structure for all data models
-        flag_test = False
+        # flag_test - set to true if this test should be applied also
+        flag_test = True
         if flag_test:
             check_wastewater_structure = False
             check_wastewater_structure = check_wastewater_structure_subclass_data()
@@ -599,6 +600,10 @@ def action_do_export():
 #18.3.2023
 # 22.3.2023 added try - seems not to work
 #            try:
+
+                # add logger info to check what selection is used
+                logger.info(f"Start Exporting with selection {str(export_dialog.selected_ids)}")
+
                 if emodel == "VSA_KEK_2019_LV95":
                     logger.info("Start Exporting VSA_KEK_2019_LV95 - qgep_export")
                     #qgep_export(selection=export_dialog.selected_ids, labels_file=labels_file_path) 

qgepqwat2ili/postimport.py

@@ -41,6 +41,17 @@ def qgep_postimport():
     post_session.execute("SELECT qgep_od.wastewater_structure_update_fk_main_wastewater_node('', True);"
     )
 
+    # add symbology update queries - backporting from tww https://github.com/teksi/wastewater/pull/263
+    logger.info("update_wastewater_node_symbology for all datasets - please be patient")
+    post_session.execute("SELECT qgep_od.update_wastewater_node_symbology(NULL, True);")
+    logger.info("update_wastewater_node_symbology for all datasets - please be patient")
+    logger.info("update_wastewater_structure_label for all datasets - please be patient")
+    post_session.execute("SELECT qgep_od.update_wastewater_structure_label(NULL, True);")
+    logger.info("update_wastewater_node_symbology for all datasets - please be patient")
+    logger.info("update_wn_symbology_by_overflow for all datasets - please be patient")
+    post_session.execute("SELECT qgep_od.update_wn_symbology_by_overflow(NULL, True);")
+
+
     logger.info("Refresh materialized views")
     post_session.execute("SELECT qgep_network.refresh_network_simple();")
 

qgepqwat2ili/qgep/export.py

@@ -3,13 +3,14 @@
 from geoalchemy2.functions import ST_Force2D, ST_GeomFromGeoJSON
 from sqlalchemy import or_
 from sqlalchemy.orm import Session
+from sqlalchemy.sql import text
 
 from .. import utils
 from ..utils.various import logger
 from .model_abwasser import get_abwasser_model
 from .model_qgep import get_qgep_model
 
-#def qgep_export(selection=None, labels_file=None):
+
 def qgep_export(selection=None, labels_file=None, orientation=None):
     """
     Export data from the QGEP model into the ili2pg model.
@@ -28,6 +29,9 @@ def qgep_export(selection=None, labels_file=None, orientation=None):
     abwasser_session = Session(utils.sqlalchemy.create_engine(), autocommit=False, autoflush=False)
     tid_maker = utils.ili2db.TidMaker(id_attribute="obj_id")
 
+    # backport from tww https://github.com/teksi/wastewater/blob/3acfba249866d299f8a22e249d9f1e475fe7b88d/plugin/teksi_wastewater/interlis/interlis_model_mapping/interlis_exporter_to_intermediate_schema.py#L83
+    abwasser_session.execute(text("SET CONSTRAINTS ALL DEFERRED;"))
+
     # Filtering
     filtered = selection is not None
     subset_ids = selection if selection is not None else []
@@ -96,7 +100,7 @@ def modulo_angle(val):
         if val is None:
             return None
 
-        # 5.4.2023 add orientation 
+        # add orientation 
         val = val +  float(labelorientation)
 
         val = val % 360.0
@@ -108,6 +112,27 @@ def modulo_angle(val):
 
         return val
 
+    def check_fk_in_subsetid (subset, relation):
+        """
+        checks, whether foreignkey is in the subset_ids - if yes it return the tid of the foreignkey, if no it will return None
+        """
+        # first check for None, as is get_tid
+        if relation is None:
+            return None
+
+        logger.info(f"check_fk_in_subsetid -  Subset ID's '{subset}'")
+        # get the value of the fk_ attribute as str out of the relation to be able to check whether it is in the subset
+        fremdschluesselstr = getattr(relation, "obj_id")
+        logger.info(f"check_fk_in_subsetid -  fremdschluesselstr '{fremdschluesselstr}'")
+
+        if fremdschluesselstr in subset:
+            logger.info(f"check_fk_in_subsetid - '{fremdschluesselstr}' is in subset ")
+            logger.info(f"check_fk_in_subsetid - tid = '{tid_maker.tid_for_row(relation)}' ")
+            return tid_maker.tid_for_row(relation)
+        else:
+            logger.info(f"check_fk_in_subsetid - '{fremdschluesselstr}' is not in subset - replaced with None instead!")
+            return None
+
     def create_metaattributes(row):
         metaattribute = ABWASSER.metaattribute(
             # FIELDS TO MAP TO ABWASSER.metaattribute
@@ -487,7 +512,9 @@ def textpos_common(row, t_type, geojson_crs_def):
                 QGEP.reach_point.obj_id == QGEP.reach.fk_reach_point_from,
                 QGEP.reach_point.obj_id == QGEP.reach.fk_reach_point_to,
             ),
-        ).filter(QGEP.wastewater_networkelement.obj_id.in_(subset_ids))
+        ).filter(
+            QGEP.wastewater_networkelement.obj_id.in_(subset_ids)
+        )
     for row in query:
 
         # AVAILABLE FIELDS IN QGEP.reach_point
@@ -507,7 +534,10 @@ def textpos_common(row, t_type, geojson_crs_def):
             # --- sia405_baseclass ---
             **base_common(row, "haltungspunkt"),
             # --- haltungspunkt ---
-            abwassernetzelementref=get_tid(row.fk_wastewater_networkelement__REL),
+
+            # changed call from get_tid to check_fk_in_subsetid so it does not wirte foreignkeys on elements that do not exist
+            #abwassernetzelementref=get_tid(row.fk_wastewater_networkelement__REL),
+            abwassernetzelementref=check_fk_in_subsetid(subset_ids, row.fk_wastewater_networkelement__REL),
             auslaufform=get_vl(row.outlet_shape__REL),
             bemerkung=truncate(emptystr_to_null(row.remark), 80),
             bezeichnung=null_to_emptystr(row.identifier),