21 kuvakkeellisen esitystavan tuki templaateille

arho_feature_template/core/feature_template_library.py

@@ -5,7 +5,7 @@
 from qgis.core import QgsFeature, QgsProject, QgsVectorLayer
 from qgis.gui import QgsMapToolDigitizeFeature
 from qgis.PyQt.QtCore import Qt
-from qgis.PyQt.QtGui import QStandardItem, QStandardItemModel
+from qgis.PyQt.QtGui import QIcon, QStandardItem, QStandardItemModel
 from qgis.utils import iface
 
 from arho_feature_template.core.template_library_config import (
@@ -18,10 +18,16 @@
 from arho_feature_template.gui.feature_attribute_form import FeatureAttributeForm
 from arho_feature_template.gui.template_dock import TemplateLibraryDock
 from arho_feature_template.resources.template_libraries import library_config_files
+from arho_feature_template.utils import get_icon_file_path
 
 logger = logging.getLogger(__name__)
 
 
+class IconFileNotFoundError(Exception):
+    def __init__(self, icon_file: str):
+        super().__init__(f"Icon file {icon_file} not found")
+
+
 class LayerNotFoundError(Exception):
     def __init__(self, layer_name: str):
         super().__init__(f"Layer {layer_name} not found")
@@ -58,6 +64,17 @@ def __init__(self, template_config: FeatureTemplate) -> None:
 
         self.setCheckable(True)
 
+        if template_config.icon_file:
+            self.set_icon_from_config()
+
+    def set_icon_from_config(self):
+        icon_file = self.config.icon_file
+        icon_fp = get_icon_file_path(icon_file)
+        if icon_fp:
+            self.setIcon(QIcon(str(icon_fp)))
+        else:
+            raise IconFileNotFoundError(icon_file)
+
     def is_valid(self) -> bool:
         """Check if the template is valid agains current QGIS project
 

arho_feature_template/core/template_library_config.py

@@ -72,13 +72,15 @@ class FeatureTemplate:
     name: str
     description: str | None
     feature: Feature
+    icon_file: str | None
 
     @classmethod
     def from_dict(cls, data: dict) -> FeatureTemplate:
         return cls(
             name=data["name"],
             description=data.get("description"),
             feature=Feature.from_dict(data["feature"]),
+            icon_file=data.get("icon_file"),
         )
 
 

arho_feature_template/resources/template_libraries/asemakaava-sample.yaml

@@ -13,6 +13,7 @@ templates:
           default: 1
   - name: Asuin-, liike- ja toimistorakennusten alue
     description: Aluella kuvataan ...
+    icon_file: ak.jpg
     feature:
       layer: land_use_area
       attributes:

arho_feature_template/utils.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+
+def get_icon_file_path(
+    icon_file: str, icons_directory: Path = Path(__file__).parent / "resources/template_libraries/icons"
+) -> Path | None:
+    icon_full_fp = icons_directory / icon_file
+    if icon_full_fp.exists():
+        return icon_full_fp.resolve()
+    return None