ogc-na 0.2.10__py3-none-any.whl → 0.2.11__py3-none-any.whl

ogc/na/annotate_schema.py

@@ -402,12 +402,16 @@ class SchemaAnnotator:
                     prop_value[ANNOTATION_ID] = terms[prop]
                     if prop in types:
                         prop_value[ANNOTATION_TYPE] = types[prop]
-                self._follow_ref(prop_value, fn, url, base_url)
+
+                process_subschema(prop_value)
 
             properties.update({p: {ANNOTATION_ID: terms[p]} for p in empty_properties if p in terms})
 
         def process_subschema(subschema):
 
+            if not subschema:
+                return
+
             self._follow_ref(subschema, fn, url, base_url)
 
             # Annotate oneOf, allOf, anyOf
@@ -426,7 +430,7 @@ class SchemaAnnotator:
                 process_properties(subschema)
             elif schema_type == 'array':
                 for k in ('prefixItems', 'items', 'contains'):
-                    process_properties(subschema.get(k))
+                    process_subschema(subschema.get(k))
 
             # Annotate $defs
             for defs_prop in ('$defs', 'definitions'):
@@ -453,7 +457,7 @@ class ContextBuilder:
     """
 
     def __init__(self, fn: Path | str | None = None, url: str | None = None,
-                 compact: bool = True):
+                 compact: bool = True, ref_mapper: Callable[[str], str] | None = None):
         """
         :param fn: file to load the annotated schema from
         :param url: URL to load the annotated schema from
@@ -461,6 +465,7 @@ class ContextBuilder:
         self.context = {'@context': {}}
         self._parsed_schemas: dict[str | Path, dict] = {}
         self.compact = compact
+        self._ref_mapper = ref_mapper
 
         context = self._build_context(fn, url)
         self.context = {'@context': context}
@@ -502,7 +507,7 @@ class ContextBuilder:
         if prefixes:
             own_context.update(prefixes)
 
-        def read_properties(where: dict):
+        def read_properties(where: dict, into_context: dict):
             if not isinstance(where, dict):
                 return
             for prop, prop_val in where.get('properties', {}).items():
@@ -513,56 +518,56 @@ class ContextBuilder:
                     if ANNOTATION_TYPE in prop_val:
                         prop_context['@type'] = compact_uri(prop_val[ANNOTATION_TYPE])
 
-                    if '$ref' in prop_val:
-                        ref_fn, ref_url = resolve_ref(prop_val['$ref'], fn, url, base_url)
-                        prop_context['@context'] = self._build_context(ref_fn, ref_url)
+                    process_subschema(prop_val, prop_context.setdefault('@context', {}))
+
+                    if not prop_context['@context']:
+                        prop_context.pop('@context', None)
 
-                    if len(prop_context) == 1:
+                    if isinstance(prop_context, dict) and len(prop_context) == 1:
                         # shorten to just the id
                         prop_context = next(iter(prop_context.values()))
 
-                    own_context[prop] = prop_context
+                    into_context[prop] = prop_context
+
+        def process_subschema(ss, into_context: dict):
 
-        def process_subschema(ss):
             if isinstance(ss, dict):
                 if '$ref' in ss:
                     ref_fn, ref_url = resolve_ref(ss['$ref'], fn, url, base_url)
                     merge_dicts(self._build_context(ref_fn, ref_url), own_context)
-                read_properties(ss)
+                read_properties(ss, into_context)
 
-        for i in ('allOf', 'anyOf', 'oneOf'):
-            l = schema.get(i)
-            if isinstance(l, list):
-                for sub_schema in l:
-                    process_subschema(sub_schema)
+                for i in ('allOf', 'anyOf', 'oneOf'):
+                    l = ss.get(i)
+                    if isinstance(l, list):
+                        for sub_schema in l:
+                            process_subschema(sub_schema, into_context)
 
-        for i in ('$defs', 'definitions'):
-            d = schema.get(i)
-            if isinstance(d, dict):
-                for sub_schema in d.values():
-                    process_subschema(sub_schema)
+                for i in ('$defs', 'definitions'):
+                    d = ss.get(i)
+                    if isinstance(d, dict):
+                        for sub_schema in d.values():
+                            process_subschema(sub_schema, into_context)
 
-        read_properties(schema)
+        process_subschema(schema, own_context)
 
         if self.compact:
-            context_props = set(own_context.keys())
-            compact_context = {}
-            for prop, prop_val in own_context.items():
-                compact_context[prop] = prop_val
-                if not isinstance(prop_val, dict):
-                    continue
-                prop_context = prop_val.pop('@context', None)
-                if not isinstance(prop_context, dict):
-                    continue
-                for term, term_val in prop_context.items():
-                    if not term.startswith('@') and term in context_props:
-                        compact_context.setdefault(prop, {}).setdefault('@context', {})[term] = term_val
-                    else:
-                        compact_context[term] = term_val
-                if len(compact_context[prop]) == 1 and '@id' in compact_context[prop]:
-                    compact_context[prop] = compact_context[prop]['@id']
-
-            own_context = compact_context
+
+            def compact_branch(branch, existing_terms):
+
+                for term in list(branch.keys()):
+                    if term[0] == '@':
+                        # skip special terms
+                        continue
+                    if term in existing_terms and existing_terms[term] == branch[term]:
+                        # same term exists in ancestor -> delete
+                        del branch[term]
+
+                for term, term_value in branch.items():
+                    if isinstance(term_value, dict) and '@context' in term_value:
+                        compact_branch(term_value['@context'], {**existing_terms, **branch})
+
+            compact_branch(own_context, {})
 
         self._parsed_schemas[fn or url] = own_context
         return own_context

{ogc_na-0.2.10.dist-info → ogc_na-0.2.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ogc-na
-Version: 0.2.10
+Version: 0.2.11
 Summary: OGC Naming Authority tools
 Author-email: Rob Atkinson <ratkinson@ogc.org>, Piotr Zaborowski <pzaborowski@ogc.org>, Alejandro Villar <avillar@ogc.org>
 Project-URL: Homepage, https://github.com/opengeospatial/ogc-na-tools/

{ogc_na-0.2.10.dist-info → ogc_na-0.2.11.dist-info}/RECORD

@@ -1,5 +1,5 @@
 ogc/na/__init__.py,sha256=uzcNiJ3uKFNJ1HBfKxIwgAy2HMUFsLAe5RkrUg8ncac,464
-ogc/na/annotate_schema.py,sha256=XJQD1UoEglcN78W6OQ7azpBjgKNETuqbT0WsIq6d1p0,22411
+ogc/na/annotate_schema.py,sha256=wEvqVbdRXpAKSLQz231H7B11m-mZekDrtULjXPMMIVk,22473
 ogc/na/domain_config.py,sha256=x_X4YiNo1etaa4N6KrZFRsFgWPPKk1rLqVE3ZP3rrps,13030
 ogc/na/download.py,sha256=2afrLyl4WsAlxkCgXsl47fs9mNKfDmhVpeT2iwNSoq0,3354
 ogc/na/ingest_json.py,sha256=3g47V01eKAPGK6EVGxyzhJhp4uE3mSfvI5gV4L3awZQ,32327
@@ -10,7 +10,7 @@ ogc/na/util.py,sha256=ySu2mSAhUYwYFH78Bmo9Z35DM4fHqb1ExLEYPjno3fs,7990
 ogc/na/validation.py,sha256=FkXx1Pwot4ztg9Vv2LrODfYxpknG9-67BmY3Ep7avd4,3535
 ogc/na/input_filters/__init__.py,sha256=PvWEI-rj5NNR_Cl903wHtS08oTywjDYEfappY5JtVkI,1169
 ogc/na/input_filters/csv.py,sha256=O2E3ivjP9i5tGtyvb6UjmR4eM7toVkzP58EUm6bxvfw,2530
-ogc_na-0.2.10.dist-info/METADATA,sha256=P5Venhx2oDit9KwrAu62Txxj7tQMUmNZGul8FTdpw3s,3528
-ogc_na-0.2.10.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-ogc_na-0.2.10.dist-info/top_level.txt,sha256=Kvy3KhzcIhNPT4_nZuJCmS946ptRr_MDyU4IIhZJhCY,4
-ogc_na-0.2.10.dist-info/RECORD,,
+ogc_na-0.2.11.dist-info/METADATA,sha256=2VI9Kdkv6M2EkPv3AtvUe2uUwz5mu10zWa7SVaMW8b0,3528
+ogc_na-0.2.11.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+ogc_na-0.2.11.dist-info/top_level.txt,sha256=Kvy3KhzcIhNPT4_nZuJCmS946ptRr_MDyU4IIhZJhCY,4
+ogc_na-0.2.11.dist-info/RECORD,,