dapla-toolbelt-metadata 0.9.7__tar.gz → 0.9.8__tar.gz

{dapla_toolbelt_metadata-0.9.7 → dapla_toolbelt_metadata-0.9.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dapla-toolbelt-metadata
-Version: 0.9.7
+Version: 0.9.8
 Summary: Dapla Toolbelt Metadata
 Project-URL: homepage, https://github.com/statisticsnorway/dapla-toolbelt-metadata
 Project-URL: repository, https://github.com/statisticsnorway/dapla-toolbelt-metadata

{dapla_toolbelt_metadata-0.9.7 → dapla_toolbelt_metadata-0.9.8}/noxfile.py

@@ -31,6 +31,7 @@ TESTS_SESSIONS_DEPENDENCIES = {
     "requests-mock",
     "faker",
     "testcontainers[generic]",
+    "typeguard"
 }
 
 
@@ -152,6 +153,7 @@ def mypy(session: nox.Session) -> None:
         "types-beautifulsoup4",
         "types-docker",
         "faker",
+        "typeguard"
     )
     session.run("mypy", *args)
     if not session.posargs:

{dapla_toolbelt_metadata-0.9.7 → dapla_toolbelt_metadata-0.9.8}/pyproject.toml

@@ -2,7 +2,7 @@
 name = "dapla-toolbelt-metadata"
 description = "Dapla Toolbelt Metadata"
 license = "MIT"
-version = "0.9.7"
+version = "0.9.8"
 classifiers = ["Development Status :: 4 - Beta"]
 readme = "README.md"
 authors = [{ name = "Statistics Norway", email = "metadata@ssb.no" }]

{dapla_toolbelt_metadata-0.9.7 → dapla_toolbelt_metadata-0.9.8}/src/dapla_metadata/_shared/config.py

@@ -13,9 +13,6 @@ from dotenv import load_dotenv
 from dapla_metadata._shared.enums import DaplaEnvironment
 from dapla_metadata._shared.enums import DaplaRegion
 from dapla_metadata._shared.enums import DaplaService
-from dapla_metadata.datasets.utility.constants import (
-    DATADOC_STATISTICAL_SUBJECT_SOURCE_URL,
-)
 
 logger = logging.getLogger(__name__)
 
@@ -28,6 +25,12 @@ DAPLA_SERVICE = "DAPLA_SERVICE"
 DAPLA_GROUP_CONTEXT = "DAPLA_GROUP_CONTEXT"
 OIDC_TOKEN = "OIDC_TOKEN"  # noqa: S105
 
+
+DATADOC_STATISTICAL_SUBJECT_SOURCE_URL_DEFAULT = (
+    "https://www.ssb.no/xp/_/service/mimir/subjectStructurStatistics"
+)
+
+
 env_loaded = False
 
 
@@ -73,7 +76,7 @@ def get_statistical_subject_source_url() -> str | None:
     """Get the URL to the statistical subject source."""
     return (
         get_config_item("DATADOC_STATISTICAL_SUBJECT_SOURCE_URL")
-        or DATADOC_STATISTICAL_SUBJECT_SOURCE_URL
+        or DATADOC_STATISTICAL_SUBJECT_SOURCE_URL_DEFAULT
     )
 
 

{dapla_toolbelt_metadata-0.9.7 → dapla_toolbelt_metadata-0.9.8}/src/dapla_metadata/datasets/utility/constants.py

@@ -94,10 +94,6 @@ DATASET_FIELDS_FROM_EXISTING_METADATA = [
 
 METADATA_DOCUMENT_FILE_SUFFIX = "__DOC.json"
 
-DATADOC_STATISTICAL_SUBJECT_SOURCE_URL = (
-    "https://www.ssb.no/xp/_/service/mimir/subjectStructurStatistics"
-)
-
 PAPIS_STABLE_IDENTIFIER_TYPE = "FREG_SNR"
 PAPIS_ENCRYPTION_KEY_REFERENCE = "papis-common-key-1"
 DAEAD_ENCRYPTION_KEY_REFERENCE = "ssb-common-key-1"

{dapla_toolbelt_metadata-0.9.7 → dapla_toolbelt_metadata-0.9.8}/src/dapla_metadata/datasets/utility/urn.py

@@ -6,9 +6,12 @@ from collections.abc import Iterable
 from dataclasses import dataclass
 from enum import Enum
 from enum import auto
+from typing import Literal
 
 from pydantic import AnyUrl
 
+from dapla_metadata._shared.config import get_dapla_environment
+from dapla_metadata._shared.enums import DaplaEnvironment
 from dapla_metadata.datasets.utility.utils import VariableListType
 
 logger = logging.getLogger(__name__)
@@ -44,6 +47,9 @@ class ReferenceUrlTypes(Enum):
     FRONTEND = auto()
 
 
+UrlVisibility = Literal["public", "internal"]
+
+
 @dataclass
 class UrnConverter:
     """Converts URLs to URNs and vice versa.
@@ -70,11 +76,81 @@ class UrnConverter:
     def _build_pattern(self, url_base: str) -> re.Pattern[str]:
         return re.compile(f"^{url_base}/{self.id_pattern}")
 
-    def build_urn(self, identifier: str) -> str:
+    def get_urn(self, identifier: str) -> str:
         """Build a URN for the given identifier."""
         return f"{self.urn_base}:{identifier}"
 
-    def convert_to_urn(self, url: str | AnyUrl) -> AnyUrl | None:
+    def get_url(
+        self,
+        identifier: str,
+        url_type: ReferenceUrlTypes,
+        visibility: Literal["public", "internal"] = "public",
+    ) -> str | None:
+        """Build concrete URL to reference a resource.
+
+        There are typically multiple URLs used to refer to one resource, this method attempts to support known variations.
+
+        Args:
+            identifier (str): The identifier of the resource the URL refers to.
+            url_type (ReferenceUrlTypes): The representation type of the URL
+            visibility (UrlVisibility, optional): Whether the URL should be that which is publicly available or not. Defaults to "public".
+
+        Returns:
+            str | None: The concrete URL. None if we cannot satisfy the supplied requirements.
+        """
+        candidates = [base[-1] for base in self.url_bases if base[0] == url_type]
+
+        def matches_visibility(url: str, visibility: UrlVisibility):
+            return (".intern." in url) is (visibility == "internal")
+
+        def matches_environment(url: str):
+            current_environment = get_dapla_environment()
+            if current_environment == DaplaEnvironment.TEST:
+                return ".test." in url
+            return ".test." not in url
+
+        if url := next(
+            (
+                url
+                for url in candidates
+                if matches_visibility(url, visibility) and matches_environment(url)
+            ),
+            None,
+        ):
+            return url + "/" + identifier
+        return None
+
+    def get_id(self, urn_or_url: str | AnyUrl) -> str | None:
+        """Get an identifier from a URN or URL.
+
+        Args:
+            urn_or_url (str | AnyUrl): The URN or URL refering to a particular resource
+
+        Returns:
+            str | None: The identifier for the resource, or None if it cannot be extracted.
+        """
+        if str(urn_or_url).startswith(self.urn_base):
+            return str(urn_or_url).removeprefix(self.urn_base + ":")
+        return self._extract_id_from_url(urn_or_url)
+
+    def is_id(self, value: str) -> bool:
+        """Check if the value is an identifier for this URN type.
+
+        Args:
+            value (str): The value to check.
+        """
+        if not isinstance(value, str):
+            # Mypy thinks it's impossible to reach this branch, but there are no guarantees in Python.
+            return False  # type: ignore [unreachable]
+        pattern = re.compile(f"^{self.id_pattern}$")
+        return bool(pattern.match(value))
+
+    def _extract_id_from_url(self, url: str | AnyUrl) -> str | None:
+        patterns = (self._build_pattern(url[-1]) for url in self.url_bases)
+        matches = (self._extract_id(str(url), p) for p in patterns)
+        return next((m for m in matches if m), None)
+
+    def convert_url_to_urn(self, url: str | AnyUrl) -> AnyUrl | None:
         """Convert a URL to a generalized URN for that same resource.
 
         Args:
@@ -86,11 +162,8 @@ class UrnConverter:
         if str(url).startswith(self.urn_base):
             # In this case the value is already in the expected format and nothing needs to be done.
             return AnyUrl(url)
-        patterns = (self._build_pattern(url[-1]) for url in self.url_bases)
-        matches = (self._extract_id(str(url), p) for p in patterns)
-        identifier = next((m for m in matches if m), None)
-        if identifier:
-            return AnyUrl(self.build_urn(identifier))
+        if identifier := self._extract_id_from_url(url):
+            return AnyUrl(self.get_urn(identifier))
 
         return None
 
@@ -149,7 +222,7 @@ def convert_uris_to_urns(
     for v in variables:
         field = getattr(v, field_name, None)
         if field:
-            if urn := next((c.convert_to_urn(field) for c in converters), None):
+            if urn := next((c.convert_url_to_urn(field) for c in converters), None):
                 setattr(v, field_name, urn)
             else:
                 logger.error(

{dapla_toolbelt_metadata-0.9.7 → dapla_toolbelt_metadata-0.9.8}/tests/datasets/test_urn.py

@@ -3,17 +3,20 @@ import logging
 import datadoc_model.all_optional.model as all_optional_model
 import pytest
 from pydantic import AnyUrl
+from typeguard import suppress_type_checks
 
 from dapla_metadata.datasets.core import Datadoc
 from dapla_metadata.datasets.utility.urn import URN_ERROR_MESSAGE_BASE
+from dapla_metadata.datasets.utility.urn import ReferenceUrlTypes
 from dapla_metadata.datasets.utility.urn import SsbNaisDomains
+from dapla_metadata.datasets.utility.urn import UrlVisibility
 from dapla_metadata.datasets.utility.urn import klass_urn_converter
 from dapla_metadata.datasets.utility.urn import vardef_urn_converter
 
 EXAMPLE_KLASS_ID = "91"
-EXAMPLE_KLASS_URN = AnyUrl(klass_urn_converter.build_urn(EXAMPLE_KLASS_ID))
+EXAMPLE_KLASS_URN = AnyUrl(klass_urn_converter.get_urn(EXAMPLE_KLASS_ID))
 EXAMPLE_VARDEF_ID = "hd8sks89"
-EXAMPLE_VARDEF_URN = AnyUrl(vardef_urn_converter.build_urn(EXAMPLE_VARDEF_ID))
+EXAMPLE_VARDEF_URN = AnyUrl(vardef_urn_converter.get_urn(EXAMPLE_VARDEF_ID))
 
 VARIABLE_DEFINITION_URN_TEST_CASES = [
     *[
@@ -52,7 +55,7 @@ def test_convert_to_vardef_urn(
     expected_result: AnyUrl | None,
     expect_warning: bool,  # noqa: ARG001
 ):
-    assert vardef_urn_converter.convert_to_urn(case) == expected_result
+    assert vardef_urn_converter.convert_url_to_urn(case) == expected_result
 
 
 @pytest.mark.parametrize(
@@ -88,6 +91,66 @@ def test_convert_to_vardef_urn_end_to_end(
     assert (URN_ERROR_MESSAGE_BASE in caplog.text) is expect_warning
 
 
+@pytest.mark.parametrize("identifier", [EXAMPLE_VARDEF_ID])
+@pytest.mark.parametrize("visibility", ["internal", "public"])
+@pytest.mark.parametrize(
+    "url_type", [ReferenceUrlTypes.API, ReferenceUrlTypes.FRONTEND]
+)
+def test_vardef_get_url(
+    url_type: ReferenceUrlTypes, visibility: UrlVisibility, identifier: str
+):
+    url = vardef_urn_converter.get_url(identifier, url_type, visibility)
+    assert url is not None
+    assert url.endswith("/" + identifier)
+    if url_type == ReferenceUrlTypes.API:
+        assert url.startswith("https://metadata.")
+    else:
+        assert url.startswith("https://catalog.")
+    if visibility == "internal":
+        assert ".intern." in url
+    else:
+        assert ".intern." not in url
+
+
+@pytest.mark.parametrize(
+    ("urn_or_url", "identifier"),
+    [
+        ("https://www.vg.no", None),
+        (EXAMPLE_VARDEF_URN, EXAMPLE_VARDEF_ID),
+        (
+            f"https://catalog.ssb.no/variable-definitions/{EXAMPLE_VARDEF_ID}",
+            EXAMPLE_VARDEF_ID,
+        ),
+        (
+            AnyUrl(
+                f"https://metadata.intern.test.ssb.no/variable-definitions/{EXAMPLE_VARDEF_ID}"
+            ),
+            EXAMPLE_VARDEF_ID,
+        ),
+    ],
+)
+def test_vardef_get_id(urn_or_url: str | AnyUrl, identifier: str | None):
+    assert vardef_urn_converter.get_id(urn_or_url) == identifier
+
+
+@pytest.mark.parametrize(
+    ("identifier", "expected"),
+    [
+        (EXAMPLE_VARDEF_ID, True),
+        ("invalid-id", False),
+        ("123", False),
+        ("12345678", True),
+        ("123456789", False),
+        ("abcdEFGH", False),
+        ("", False),
+        (None, False),
+    ],
+)
+@suppress_type_checks
+def test_vardef_is_id(identifier: str, expected: bool):
+    assert vardef_urn_converter.is_id(identifier) is expected
+
+
 CLASSIFICATION_URN_TEST_CASES = [
     (
         f"https://www.ssb.no/klass/klassifikasjoner/{EXAMPLE_KLASS_ID}",
@@ -126,7 +189,7 @@ def test_convert_to_klass_urn(
     expected_result: str,
     expect_warning: bool,  # noqa: ARG001
 ):
-    assert klass_urn_converter.convert_to_urn(case) == expected_result
+    assert klass_urn_converter.convert_url_to_urn(case) == expected_result
 
 
 @pytest.mark.parametrize(
@@ -160,3 +223,63 @@ def test_convert_to_klass_urn_end_to_end(
     else:
         assert meta.variables[0].classification_uri == expected_result
     assert (URN_ERROR_MESSAGE_BASE in caplog.text) is expect_warning
+
+
+@pytest.mark.parametrize("identifier", [EXAMPLE_KLASS_ID])
+@pytest.mark.parametrize("visibility", ["internal", "public"])
+@pytest.mark.parametrize(
+    "url_type", [ReferenceUrlTypes.API, ReferenceUrlTypes.FRONTEND]
+)
+def test_klass_get_url(
+    url_type: ReferenceUrlTypes, visibility: UrlVisibility, identifier: str
+):
+    url = klass_urn_converter.get_url(identifier, url_type, visibility)
+    if visibility == "internal":
+        # Klass doesn't have an internal representation
+        assert url is None
+    else:
+        assert url is not None
+        assert url.endswith("/" + identifier)
+        if url_type == ReferenceUrlTypes.API:
+            assert url.startswith("https://data.")
+        else:
+            assert url.startswith("https://www.ssb.")
+
+
+@pytest.mark.parametrize(
+    ("urn_or_url", "identifier"),
+    [
+        ("https://www.vg.no", None),
+        (EXAMPLE_KLASS_URN, EXAMPLE_KLASS_ID),
+        (
+            f"https://www.ssb.no/klass/klassifikasjoner/{EXAMPLE_KLASS_ID}",
+            EXAMPLE_KLASS_ID,
+        ),
+        (
+            AnyUrl(
+                f"https://data.ssb.no/api/klass/v1/classifications/{EXAMPLE_KLASS_ID}"
+            ),
+            EXAMPLE_KLASS_ID,
+        ),
+    ],
+)
+def test_klass_get_id(urn_or_url: str | AnyUrl, identifier: str | None):
+    assert klass_urn_converter.get_id(urn_or_url) == identifier
+
+
+@pytest.mark.parametrize(
+    ("identifier", "expected"),
+    [
+        (EXAMPLE_KLASS_ID, True),
+        ("invalid-id", False),
+        ("123", True),
+        ("1", True),
+        ("123456789", False),
+        ("abcdEFGH", False),
+        ("", False),
+        (None, False),
+    ],
+)
+@suppress_type_checks
+def test_klass_is_id(identifier: str, expected: bool):
+    assert klass_urn_converter.is_id(identifier) is expected

{dapla_toolbelt_metadata-0.9.7 → dapla_toolbelt_metadata-0.9.8}/uv.lock

@@ -525,7 +525,7 @@ wheels = [
 
 [[package]]
 name = "dapla-toolbelt-metadata"
-version = "0.9.7"
+version = "0.9.8"
 source = { editable = "." }
 dependencies = [
     { name = "arrow" },