hestia-earth-utils 0.15.14__tar.gz → 0.15.16__tar.gz

{hestia_earth_utils-0.15.14 → hestia_earth_utils-0.15.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hestia_earth_utils
-Version: 0.15.14
+Version: 0.15.16
 Summary: HESTIA's utils library
 Home-page: https://gitlab.com/hestia-earth/hestia-utils
 Author: HESTIA Team

{hestia_earth_utils-0.15.14 → hestia_earth_utils-0.15.16}/hestia_earth/utils/emission.py

@@ -9,7 +9,11 @@ from .lookup_utils import (
     is_product_termType_allowed,
     is_product_id_allowed,
     is_input_termType_allowed,
-    is_input_id_allowed
+    is_input_id_allowed,
+    is_practice_termType_allowed,
+    is_practice_id_allowed,
+    is_transformation_termType_allowed,
+    is_transformation_id_allowed,
 )
 
 
@@ -35,7 +39,11 @@ def cycle_emission_is_in_system_boundary(cycle: dict):
             is_product_termType_allowed,
             is_product_id_allowed,
             is_input_termType_allowed,
-            is_input_id_allowed
+            is_input_id_allowed,
+            is_practice_termType_allowed,
+            is_practice_id_allowed,
+            is_transformation_termType_allowed,
+            is_transformation_id_allowed,
         ]))
 
     return filter_term

{hestia_earth_utils-0.15.14 → hestia_earth_utils-0.15.16}/hestia_earth/utils/lookup_utils.py

@@ -1,5 +1,6 @@
 from functools import lru_cache
 import json
+from hestia_earth.schema import SchemaType
 
 from .lookup import _download_lookup_data, download_lookup, get_table_value, column_name
 from .api import download_hestia
@@ -35,6 +36,10 @@ def _get_site_measurements(node: dict):
     return flatten([non_empty_list(site.get('measurements', [])) for site in sites])
 
 
+def _blank_node_term_values(blank_nodes: list, key: str = '@id'):
+    return non_empty_list([v.get('term', {}).get(key) for v in blank_nodes])
+
+
 @lru_cache()
 def _allowed_model_mapping(model: str, term_id: str, column: str):
     mapping = _allowed_mapping_data()
@@ -45,22 +50,27 @@ def _allowed_model_mapping(model: str, term_id: str, column: str):
     return (value or _ALLOW_ALL).split(';') if isinstance(value, str) else _ALLOW_ALL
 
 
+def _is_model_value_allowed(model: str, term_id: str, values: list, lookup_column: str):
+    allowed_values = _allowed_model_mapping(model, term_id, lookup_column)
+    return any([
+        _ALLOW_ALL in allowed_values,
+        len(values) == 0
+    ]) or any([value in allowed_values for value in values])
+
+
 def is_model_siteType_allowed(model: str, term_id: str, data: dict):
-    site_types = _get_site_types(data)
-    allowed_values = _allowed_model_mapping(model, term_id, 'siteTypesAllowed')
-    return True if _ALLOW_ALL in allowed_values or not site_types else any([
-        (site_type in allowed_values) for site_type in site_types
-    ])
+    values = _get_site_types(data)
+    return _is_model_value_allowed(model, term_id, values, 'siteTypesAllowed')
 
 
 def is_model_product_id_allowed(model: str, term_id: str, data: dict):
-    products = data.get('products', [])
-    values = non_empty_list([p.get('term', {}).get('@id') for p in products])
-    allowed_values = _allowed_model_mapping(model, term_id, 'productTermIdsAllowed')
-    return True if any([
-        _ALLOW_ALL in allowed_values,
-        len(values) == 0
-    ]) else any([value in allowed_values for value in values])
+    values = _blank_node_term_values(data.get('products', []))
+    return _is_model_value_allowed(model, term_id, values, 'productTermIdsAllowed')
+
+
+def is_model_measurement_id_allowed(model: str, term_id: str, data: dict):
+    values = _blank_node_term_values(data.get('measurements', []))
+    return _is_model_value_allowed(model, term_id, values, 'measurementIdsAllowed')
 
 
 @lru_cache()
@@ -73,68 +83,76 @@ def _allowed_mapping(term_id: str, column: str):
     return (value or _ALLOW_ALL).split(';') if isinstance(value, str) else _ALLOW_ALL
 
 
+def _is_term_value_allowed(term_id: str, values: list, lookup_column: str):
+    allowed_values = _allowed_mapping(term_id, lookup_column)
+    return any([
+        _ALLOW_ALL in allowed_values,
+        len(values) == 0
+    ]) or any([value in allowed_values for value in values])
+
+
 def is_siteType_allowed(data: dict, term_id: str):
-    site_types = _get_site_types(data)
-    allowed_values = _allowed_mapping(term_id, 'siteTypesAllowed')
-    return True if _ALLOW_ALL in allowed_values or not site_types else any([
-        (site_type in allowed_values) for site_type in site_types
-    ])
+    values = _get_site_types(data)
+    return _is_term_value_allowed(term_id, values, 'siteTypesAllowed')
 
 
 def is_site_measurement_id_allowed(data: dict, term_id: str):
     measurements = _get_site_measurements(data)
-    values = non_empty_list([v.get('term', {}).get('@id') for v in measurements])
-    allowed_values = _allowed_mapping(term_id, 'siteMeasurementIdsAllowed')
-    return True if any([
-        _ALLOW_ALL in allowed_values,
-        len(values) == 0
-    ]) else any([value in allowed_values for value in values])
+    values = _blank_node_term_values(measurements, key='@id')
+    return _is_term_value_allowed(term_id, values, 'siteMeasurementIdsAllowed')
 
 
 def is_product_termType_allowed(data: dict, term_id: str):
     products = data.get('products', [])
-    values = non_empty_list([p.get('term', {}).get('termType') for p in products])
-    allowed_values = _allowed_mapping(term_id, 'productTermTypesAllowed')
-    return True if any([
-        _ALLOW_ALL in allowed_values,
-        len(values) == 0
-    ]) else any([value in allowed_values for value in values])
+    values = _blank_node_term_values(products, key='termType')
+    return _is_term_value_allowed(term_id, values, 'productTermTypesAllowed')
 
 
 def is_product_id_allowed(data: dict, term_id: str):
     products = data.get('products', [])
-    values = non_empty_list([p.get('term', {}).get('@id') for p in products])
-    allowed_values = _allowed_mapping(term_id, 'productTermIdsAllowed')
-    return True if any([
-        _ALLOW_ALL in allowed_values,
-        len(values) == 0
-    ]) else any([value in allowed_values for value in values])
+    values = _blank_node_term_values(products, key='@id')
+    return _is_term_value_allowed(term_id, values, 'productTermIdsAllowed')
 
 
 def is_input_termType_allowed(data: dict, term_id: str):
     inputs = data.get('inputs', [])
-    values = non_empty_list([p.get('term', {}).get('termType') for p in inputs])
-    allowed_values = _allowed_mapping(term_id, 'inputTermTypesAllowed')
-    return True if any([
-        _ALLOW_ALL in allowed_values,
-        len(values) == 0
-    ]) else any([value in allowed_values for value in values])
+    values = _blank_node_term_values(inputs, key='termType')
+    return _is_term_value_allowed(term_id, values, 'inputTermTypesAllowed')
 
 
 def is_input_id_allowed(data: dict, term_id: str):
     inputs = data.get('inputs', [])
-    values = non_empty_list([p.get('term', {}).get('@id') for p in inputs])
-    allowed_values = _allowed_mapping(term_id, 'inputTermIdsAllowed')
-    return True if any([
-        _ALLOW_ALL in allowed_values,
-        len(values) == 0
-    ]) else any([value in allowed_values for value in values])
+    values = _blank_node_term_values(inputs, key='@id')
+    return _is_term_value_allowed(term_id, values, 'inputTermIdsAllowed')
+
+
+def is_practice_termType_allowed(data: dict, term_id: str):
+    practices = data.get('practices', [])
+    values = _blank_node_term_values(practices, key='termType')
+    return _is_term_value_allowed(term_id, values, 'practiceTermTypesAllowed')
+
+
+def is_practice_id_allowed(data: dict, term_id: str):
+    practices = data.get('practices', [])
+    values = _blank_node_term_values(practices, key='@id')
+    return _is_term_value_allowed(term_id, values, 'practiceTermIdsAllowed')
+
+
+def is_transformation_termType_allowed(data: dict, term_id: str):
+    is_transformation = data.get('@type', data.get('type')) == SchemaType.TRANSFORMATION.value
+    values = non_empty_list([data.get('term', {}).get('termType')])
+    return not is_transformation or _is_term_value_allowed(term_id, values, 'transformationTermTypesAllowed')
+
+
+def is_transformation_id_allowed(data: dict, term_id: str):
+    is_transformation = data.get('@type', data.get('type')) == SchemaType.TRANSFORMATION.value
+    values = non_empty_list([data.get('term', {}).get('@id')])
+    return not is_transformation or _is_term_value_allowed(term_id, values, 'transformationTermIdsAllowed')
 
 
 def is_node_type_allowed(data: dict, term_id: str):
-    node_type = data.get('@type', data.get('type'))
-    allowed_types = _allowed_mapping(term_id, 'typesAllowed')
-    return True if _ALLOW_ALL in allowed_types or not node_type else node_type in allowed_types
+    values = non_empty_list([data.get('@type', data.get('type'))])
+    return _is_term_value_allowed(term_id, values, 'typesAllowed')
 
 
 @lru_cache()

hestia_earth_utils-0.15.16/hestia_earth/utils/version.py

@@ -0,0 +1 @@
+VERSION = '0.15.16'

{hestia_earth_utils-0.15.14 → hestia_earth_utils-0.15.16}/hestia_earth_utils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hestia-earth-utils
-Version: 0.15.14
+Version: 0.15.16
 Summary: HESTIA's utils library
 Home-page: https://gitlab.com/hestia-earth/hestia-utils
 Author: HESTIA Team

hestia_earth_utils-0.15.14/hestia_earth/utils/version.py

@@ -1 +0,0 @@
-VERSION = '0.15.14'