Moved lambda request and response handling to their own module.

Mtk112 · Mtk112 · commit 7bc6a773853e · 2024-11-28T16:09:59.000+02:00
- Moved lambda request and response handling to their own module.
- Added capability to use local docker lambdas.
- Added utility to get plan name with plan_id from "Kaava" layer.
diff --git a/arho_feature_template/core/lambda_service.py b/arho_feature_template/core/lambda_service.py
@@ -0,0 +1,95 @@
+from __future__ import annotations
+
+import json
+
+from qgis.PyQt.QtCore import QByteArray, QObject, QUrl, pyqtSignal
+from qgis.PyQt.QtNetwork import QNetworkAccessManager, QNetworkProxy, QNetworkReply, QNetworkRequest
+from qgis.PyQt.QtWidgets import QMessageBox
+
+from arho_feature_template.utils.misc_utils import get_active_plan_id, get_plan_name, get_settings
+
+
+class LambdaService(QObject):
+    jsons_received = pyqtSignal(dict, dict)
+
+    def __init__(self):
+        super().__init__()  # Ensure QObject initialization
+        # Init network manager
+        self.network_manager = QNetworkAccessManager()
+
+        # Get settings
+        proxy_host, proxy_port, self.lambda_url = get_settings()
+
+        if proxy_host and proxy_port:
+            # Set up SOCKS5 Proxy if values are provided
+            proxy = QNetworkProxy()
+            proxy.setType(QNetworkProxy.Socks5Proxy)
+            proxy.setHostName(proxy_host)
+            proxy.setPort(int(proxy_port))
+            self.network_manager.setProxy(proxy)
+
+    def send_request(self, action: str, plan_id: str):
+        """Sends a request to the lambda function."""
+        payload = {"action": action, "plan_uuid": plan_id}
+        payload_bytes = QByteArray(json.dumps(payload).encode("utf-8"))
+
+        request = QNetworkRequest(QUrl(self.lambda_url))
+        request.setHeader(QNetworkRequest.ContentTypeHeader, "application/json")
+
+        reply = self.network_manager.post(request, payload_bytes)
+
+        # Connect reply signal to handle the response
+        reply.finished.connect(lambda: self._process_reply(reply))
+
+    def _process_reply(self, reply: QNetworkReply):
+        """Processes the reply from the lambda and emits signal."""
+        plan_id = get_active_plan_id()
+
+        if reply.error() != QNetworkReply.NoError:
+            error_string = reply.errorString()
+            QMessageBox.critical(None, "API Virhe", f"Lambdan kutsu epäonnistui: {error_string}")
+            reply.deleteLater()
+            return
+
+        try:
+            response_data = reply.readAll().data().decode("utf-8")
+
+            response_json = json.loads(response_data)
+
+            # Determine if the proxy is set up.
+            if hasattr(self, 'network_manager') and self.network_manager.proxy().type() == QNetworkProxy.Socks5Proxy:
+                # If proxy has been set up, retrieve 'details' directly
+                details = response_json.get("details", {})
+            else:
+                # If proxy has not been set up (using local docker lambda), the response includes 'body'.
+                # In this case we need to retrieve 'details' from 'body' first.
+                body = response_json.get("body", {})
+                details = body.get("details", {})
+
+            # Extract the plan JSON for the given plan_id
+            plan_json = details.get(plan_id, {})
+            if not isinstance(plan_json, dict):
+                plan_json = {}
+
+            outline_json = None
+            if plan_json:
+                geographical_area = plan_json.get("geographicalArea")
+                if geographical_area:
+                    outline_name = get_plan_name(plan_id, language="fin")
+                    outline_json = {
+                        "type": "Feature",
+                        "properties": {"name": outline_name},
+                        "srid": geographical_area.get("srid"),
+                        "geometry": geographical_area.get("geometry"),
+                    }
+
+            if outline_json is None:
+                outline_json = {}  # Fallback to empty dictionary if no outline JSON is created
+
+            # Emit the signal with the two JSONs
+            self.jsons_received.emit(plan_json, outline_json)
+
+        except json.JSONDecodeError as e:
+            QMessageBox.critical(None, "JSON Virhe", f"Failed to parse response JSON: {e}")
+        finally:
+            reply.deleteLater()
diff --git a/arho_feature_template/core/plan_manager.py b/arho_feature_template/core/plan_manager.py
@@ -1,36 +1,29 @@
+from __future__ import annotations
+
 import json
 
 from qgis.core import QgsExpressionContextUtils, QgsProject, QgsVectorLayer
-from qgis.PyQt.QtCore import QByteArray, QUrl
-from qgis.PyQt.QtNetwork import QNetworkAccessManager, QNetworkProxy, QNetworkReply, QNetworkRequest
 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,
-    get_settings,
     handle_unsaved_changes,
 )
 
 
 class PlanManager:
     def __init__(self):
-        # Init network manager
-        self.network_manager = QNetworkAccessManager()
-        proxy_host, proxy_port, self.lambda_url = get_settings()
-
-        # SOCKS5 Proxy
-        proxy = QNetworkProxy()
-        proxy.setType(QNetworkProxy.Socks5Proxy)
-        proxy.setHostName(proxy_host)
-        proxy.setPort(int(proxy_port))
-        self.network_manager.setProxy(proxy)
-
+        self.lambda_service = LambdaService()
+        self.lambda_service.jsons_received.connect(self.save_plan_jsons)
+        self.json_plan_path = None
+        self.json_plan_outline_path = None
         self.kaava_layer = get_layer_by_name("Kaava")
 
     def add_new_plan(self):
@@ -124,82 +117,32 @@ def get_plan_json(self):
         """Serializes plan and plan outline to JSON"""
         dialog = SerializePlan()
         if dialog.exec_() == QDialog.Accepted:
-            json_plan_path = dialog.plan_path_edit.text()
-            json_plan_outline_path = dialog.plan_outline_path_edit.text()
+            self.json_plan_path = dialog.plan_path_edit.text()
+            self.json_plan_outline_path = dialog.plan_outline_path_edit.text()
 
             plan_id = get_active_plan_id()
-
             if not plan_id:
-                QMessageBox.critical(None, "Virhe", "Ei aktiivista kaavaa. Luo tai avaa kaava.")
+                QMessageBox.critical(None, "Virhe", "Ei aktiivista kaavaa.")
                 return
 
-            payload = {"action": "get_plans", "plan_uuid": plan_id}
-            payload_bytes = QByteArray(json.dumps(payload).encode("utf-8"))
+            self.lambda_service.send_request("get_plans", plan_id)
 
-            request = QNetworkRequest(QUrl(self.lambda_url))
-            request.setHeader(QNetworkRequest.ContentTypeHeader, "application/json")
-
-            reply = self.network_manager.post(request, payload_bytes)
-
-            # Connect the finished signal to a lambda to pass extra arguments
-            reply.finished.connect(lambda: self._handle_response(reply, json_plan_path, json_plan_outline_path))
-
-    def _handle_response(self, reply: QNetworkReply, json_plan_path: str, json_plan_outline_path: str):
-        """Handle the API response and process the plan and outline data."""
-        if reply.error() != QNetworkReply.NoError:
-            error_string = reply.errorString()
-            QMessageBox.critical(None, "Virhe API kutsussa", f"Virhe: {error_string}")
+    def save_plan_jsons(self, plan_json, outline_json):
+        """This slot saves the plan and outline JSONs to files."""
+        if plan_json is None or outline_json is None:
+            QMessageBox.critical(None, "Virhe", "Kaava tai sen ulkoraja ei löytynyt.")
             return
 
-        try:
-            response_data = reply.readAll().data().decode("utf-8")
-            response_json = json.loads(response_data)
-
-            details = response_json.get("details")
-            if not details:
-                QMessageBox.warning(None, "Varoitus", "Vastauksesta puuttuu 'details' kenttä.")
-                return
-
-            # Extract plan and write it to file
-            plan_id = get_active_plan_id()
-            if not plan_id:
-                QMessageBox.critical(None, "Virhe", "Ei aktiivista kaavaa.")
-                return
+        # Retrieve paths
+        if self.json_plan_path is None or self.json_plan_outline_path is None:
+            QMessageBox.critical(None, "Virhe", "Tiedostopolut eivät ole saatavilla.")
+            return
 
-            plan_json = details.get(plan_id)
-            if not plan_json:
-                QMessageBox.warning(None, "Varoitus", f"Aktiiviselle kaavalle (id: {plan_id}) ei löydy tietoja.")
-                return
+        # Save the JSONs
+        with open(self.json_plan_path, "w", encoding="utf-8") as full_file:
+            json.dump(plan_json, full_file, ensure_ascii=False, indent=2)
 
-            with open(json_plan_path, "w", encoding="utf-8") as full_file:
-                json.dump(plan_json, full_file, ensure_ascii=False, indent=2)
-
-            # Process the geographicalArea for the outline
-            geographical_area = plan_json.get("geographicalArea")
-            if geographical_area:
-                try:
-                    # Build the structured outline JSON and write to file
-                    outline_json = {
-                        "type": "Feature",
-                        "properties": {"name": "Example Polygon"},
-                        "srid": geographical_area.get("srid"),
-                        "geometry": geographical_area.get("geometry"),
-                    }
-
-                    with open(json_plan_outline_path, "w", encoding="utf-8") as outline_file:
-                        json.dump(outline_json, outline_file, ensure_ascii=False, indent=2)
-
-                    QMessageBox.information(None, "Success", "Kaava ja sen ulkorja tallennettu onnistuneesti.")
-                except KeyError as e:
-                    QMessageBox.critical(
-                        None,
-                        "Virhe",
-                        f"'geographicalArea' ei sisällä vaadittuja tietoja: {e}",
-                    )
-            else:
-                QMessageBox.warning(None, "Varoitus", "Kenttä 'geographicalArea' puuttuu kaavan tiedoista.")
-        except json.JSONDecodeError as e:
-            QMessageBox.critical(None, "Virhe", f"Vastaus JSON:in purkaminen epäonnistui: {e}")
+        with open(self.json_plan_outline_path, "w", encoding="utf-8") as outline_file:
+            json.dump(outline_json, outline_file, ensure_ascii=False, indent=2)
 
-        # Clean up the reply
-        reply.deleteLater()
+        QMessageBox.information(None, "Tallennus onnistui", "Kaava ja sen ulkoraja tallennettu onnistuneesti.")
diff --git a/arho_feature_template/gui/serialize_plan.py b/arho_feature_template/gui/serialize_plan.py
@@ -8,15 +8,15 @@
 
 
 class SerializePlan(QDialog, FormClass):  # type: ignore
+    plan_outline_path_edit: QLineEdit
+    plan_path_edit: QLineEdit
+    plan_outline_select_button: QPushButton
+    plan_select_button: QPushButton
+
     def __init__(self):
         super().__init__()
         self.setupUi(self)
 
-        self.plan_outline_path_edit = self.findChild(QLineEdit, "plan_outline_path_edit")
-        self.plan_path_edit = self.findChild(QLineEdit, "plan_path_edit")
-        self.plan_outline_select_button = self.findChild(QPushButton, "plan_outline_select_button")
-        self.plan_select_button = self.findChild(QPushButton, "plan_select_button")
-
         self.plan_outline_select_button.clicked.connect(self.select_plan_outline_file)
         self.plan_select_button.clicked.connect(self.select_plan_file)
 
diff --git a/arho_feature_template/utils/misc_utils.py b/arho_feature_template/utils/misc_utils.py
@@ -102,7 +102,6 @@ def handle_unsaved_changes() -> bool:
 
 def get_active_plan_id():
     """Retrieve the active plan ID stored as a project variable."""
-    # return QgsExpressionContextUtils.projectScope(QgsProject.instance(), "active_plan_id")
     return QgsExpressionContextUtils.projectScope(QgsProject.instance()).variable("active_plan_id")
 
 
@@ -113,3 +112,16 @@ def get_settings():
     proxy_port = settings.value("proxy_port", "5443")
     lambda_url = settings.value("lambda_url", "https://t5w26iqnsf.execute-api.eu-central-1.amazonaws.com/v0/ryhti")
     return proxy_host, proxy_port, lambda_url
+
+
+def get_plan_name(plan_id: str, language: Literal["fin", "eng", "swe"] = "fin") -> str:
+    """Retrieve the name of a plan from the 'Kaava' layer based on its ID."""
+    layer = get_layer_by_name("Kaava")
+    if layer:
+        for feature in layer.getFeatures():
+            if feature["id"] == plan_id:
+                name_field = feature["name"]
+                name = name_field.get(language, "")
+                # Return "Nimetön" if the name is an empty string
+                return name if name.strip() else "Nimetön"
+    return "Nimetön"
