Add filtering to new layers

Author: LKajan

arho_feature_template/core/plan_manager.py

@@ -1,20 +1,72 @@
 from __future__ import annotations
 
 import json
+import logging
+from string import Template
+from textwrap import dedent
 
 from qgis.core import QgsExpressionContextUtils, QgsProject, QgsVectorLayer
 from qgis.PyQt.QtWidgets import QDialog, QMessageBox
 from qgis.utils import iface
 
 from arho_feature_template.core.lambda_service import LambdaService
-from arho_feature_template.core.update_plan import LandUsePlan, update_selected_plan
 from arho_feature_template.gui.load_plan_dialog import LoadPlanDialog
 from arho_feature_template.gui.serialize_plan import SerializePlan
 from arho_feature_template.utils.db_utils import get_existing_database_connection_names
 from arho_feature_template.utils.misc_utils import get_active_plan_id, get_layer_by_name, handle_unsaved_changes
 
+logger = logging.getLogger(__name__)
+
 LAYER_NAME_PLAN = "Kaava"
 
+PLAN_FILTER_TEMPLATES = {
+    "Kaava": Template("id = '$plan_id'"),
+    "Maankäytön kohteet": Template("plan_id = '$plan_id'"),
+    "Muut pisteet": Template("plan_id = '$plan_id'"),
+    "Viivat": Template("plan_id = '$plan_id'"),
+    "Aluevaraus": Template("plan_id = '$plan_id'"),
+    "Osa-alue": Template("plan_id = '$plan_id'"),
+    "Kaavamääräysryhmät": Template("plan_id = '$plan_id'"),
+    "Kaavamääräysryhmien assosiaatiot": Template(
+        dedent(
+            """\
+            EXISTS (
+                SELECT 1
+                FROM hame.plan_regulation_group prg
+                WHERE
+                    hame.regulation_group_association.plan_regulation_group_id = prg.id
+                    AND prg.plan_id = '$plan_id'
+            )"""
+        )
+    ),
+    "Kaavamääräys": Template(
+        dedent(
+            """\
+            EXISTS (
+                SELECT 1
+                FROM hame.plan_regulation_group prg
+                WHERE
+                    hame.plan_regulation.plan_regulation_group_id = prg.id
+                    AND prg.plan_id = '$plan_id'
+            )"""
+        )
+    ),
+    "Kaavasuositus": Template(
+        dedent(
+            """\
+            EXISTS (
+                SELECT 1
+                FROM hame.plan_regulation_group rg
+                WHERE
+                    hame.plan_proposition.plan_regulation_group_id = rg.id
+                    AND rg.plan_id = '$plan_id'
+            )"""
+        )
+    ),
+    "Asiakirjat": Template("plan_id = '$plan_id'"),
+    "Lähtötietoaineistot": Template("plan_id = '$plan_id'"),
+}
+
 
 class PlanManager:
     def __init__(self):
@@ -27,10 +79,10 @@ def add_new_plan(self):
             return
 
         plan_layer = get_layer_by_name(LAYER_NAME_PLAN)
-        self.clear_all_filters()
 
         if not plan_layer:
             return
+        self.set_active_plan(None)
 
         if not plan_layer.isEditable():
             plan_layer.startEditing()
@@ -61,18 +113,40 @@ def _feature_added(self):
             return
 
         feature_ids_after_commit = plan_layer.allFeatureIds()
-        new_feature_id = next((fid for fid in feature_ids_after_commit if fid not in feature_ids_before_commit), None)
+        new_feature_id = next(
+            (fid for fid in feature_ids_after_commit if fid not in feature_ids_before_commit),
+            None,
+        )
 
         if new_feature_id is not None:
             new_feature = plan_layer.getFeature(new_feature_id)
             if new_feature.isValid():
                 feature_id_value = new_feature["id"]
-                update_selected_plan(LandUsePlan(feature_id_value))
+                self.set_active_plan(feature_id_value)
             else:
                 iface.messageBar().pushMessage("Error", "Invalid feature retrieved.", level=3)
         else:
             iface.messageBar().pushMessage("Error", "No new feature was added.", level=3)
 
+    def set_active_plan(self, plan_id: str | None):
+        """Update the project layers based on the selected land use plan."""
+        QgsExpressionContextUtils.setProjectVariable(QgsProject.instance(), "active_plan_id", plan_id)
+
+        for layer_name, filter_template in PLAN_FILTER_TEMPLATES.items():
+            """Set a filter for the given vector layer."""
+            filter_expression = filter_template.substitute(plan_id=plan_id) if plan_id else None
+            layer = get_layer_by_name(layer_name)
+            if not layer:
+                logger.warning("Layer %s not found", layer_name)
+                continue
+            result = layer.setSubsetString(filter_expression)
+            if result is False:
+                iface.messageBar().pushMessage(
+                    "Error",
+                    f"Failed to filter layer {layer_name} with query {filter_expression}",
+                    level=3,
+                )
+
     def load_land_use_plan(self):
         """Load an existing land use plan using a dialog selection."""
         connections = get_existing_database_connection_names()
@@ -94,15 +168,7 @@ def load_land_use_plan(self):
                 QMessageBox.critical(None, "Error", "No plan was selected.")
                 return
 
-            plan = LandUsePlan(selected_plan_id)
-            update_selected_plan(plan)
-
-    def clear_all_filters(self):
-        """Clear active_plan_id and filters for all vector layers in the project."""
-        QgsExpressionContextUtils.setProjectVariable(QgsProject.instance(), "active_plan_id", None)
-        for layer in QgsProject.instance().mapLayers().values():
-            if isinstance(layer, QgsVectorLayer):
-                layer.setSubsetString("")
+            self.set_active_plan(selected_plan_id)
 
     def commit_all_editable_layers(self):
         """Commit all changes in any editable layers."""