Merge pull request #140 from tosh2230/get-uri

func get_uri()
tosh2230 · Sep 26, 2023 · edc0d79 · edc0d79
2 parents d2df900 + 6a65831
commit edc0d79
Showing 1 changed file with 22 additions and 14 deletions.
diff --git a/src/stairlight/stairlight.py b/src/stairlight/stairlight.py
@@ -4,7 +4,7 @@
 import os
 from dataclasses import asdict
 from logging import getLogger
-from typing import Any
+from typing import Any, OrderedDict
 
 import src.stairlight.util as sl_util
 from src.stairlight.configurator import Configurator
@@ -196,26 +196,34 @@ def _write_map(self) -> None:
 
     def get_templates_not_found(self) -> list[str]:
         not_found: set[str] = set()
+
+        # Uris that stairlight knows
         mapped_urls: list[str] = self.list_uris()
-        for config in self._mapping_config.Mapping:
-            uri: str = ""
-            if config.get(MappingConfigKey.TEMPLATE_SOURCE_TYPE) in [
-                TemplateSourceType.FILE.value,
-                TemplateSourceType.DBT.value,
-            ]:
-                file_suffix = config.get(MappingConfigKey.File.FILE_SUFFIX)
-                uri = os.path.abspath(f"./{file_suffix}")
-            elif config.get(MappingConfigKey.TEMPLATE_SOURCE_TYPE) in [
-                TemplateSourceType.GCS.value,
-                TemplateSourceType.S3.value,
-            ]:
-                uri = config.get(MappingConfigKey.Gcs.URI)
 
+        for config in self._mapping_config.Mapping:
+            # Uri that mapping config set
+            uri: str = self.get_uri(config=config)
             if uri not in mapped_urls:
                 not_found.add(uri)
 
         return sorted(not_found)
 
+    @staticmethod
+    def get_uri(config: OrderedDict[str, Any]) -> str:
+        uri: str = ""
+        if config.get(MappingConfigKey.TEMPLATE_SOURCE_TYPE) in [
+            TemplateSourceType.FILE.value,
+            TemplateSourceType.DBT.value,
+        ]:
+            file_suffix = config.get(MappingConfigKey.File.FILE_SUFFIX)
+            uri = os.path.abspath(f"./{file_suffix}")
+        elif config.get(MappingConfigKey.TEMPLATE_SOURCE_TYPE) in [
+            TemplateSourceType.GCS.value,
+            TemplateSourceType.S3.value,
+        ]:
+            uri = config.get(MappingConfigKey.Gcs.URI)
+        return uri
+
     def init(self, prefix: str = STAIRLIGHT_CONFIG_PREFIX_DEFAULT) -> str:
         """Create Stairlight template file