cognite-neat 0.115.0__tar.gz → 0.115.1__tar.gz

{cognite_neat-0.115.0 → cognite_neat-0.115.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: cognite-neat
-Version: 0.115.0
+Version: 0.115.1
 Summary: Knowledge graph transformation
 License: Apache-2.0
 Author: Nikola Vasiljevic

{cognite_neat-0.115.0 → cognite_neat-0.115.1}/cognite/neat/_session/_base.py

@@ -165,6 +165,9 @@ class NeatSession:
             neat.convert()
             ```
         """
+        self._state._raise_exception_if_condition_not_met(
+            "Convert to physical", has_dms_rules=False, has_information_rules=True
+        )
         converter = InformationToDMS(reserved_properties=reserved_properties)
 
         issues = self._state.rule_transform(converter)

{cognite_neat-0.115.0 → cognite_neat-0.115.1}/cognite/neat/_session/_read.py

@@ -362,6 +362,11 @@ class ExcelReadAPI(BaseReadAPI):
         if enable_manual_edit:
             warnings.filterwarnings("default")
             AlphaFlags.manual_rules_edit.warn()
+        else:
+            self._state._raise_exception_if_condition_not_met(
+                "Read Excel Rules",
+                empty_rules_store_required=True,
+            )
 
         return self._state.rule_import(importers.ExcelImporter(path), enable_manual_edit)
 
@@ -381,6 +386,10 @@ class YamlReadAPI(BaseReadAPI):
             neat.read.yaml("path_to_toolkit_yamls")
             ```
         """
+        self._state._raise_exception_if_condition_not_met(
+            "Read YAML data model",
+            empty_rules_store_required=True,
+        )
         reader = NeatReader.create(io)
         path = reader.materialize_path()
         importer: BaseImporter
@@ -634,6 +643,10 @@ class RDFReadAPI(BaseReadAPI):
         return self._state.rule_import(importer)
 
     def instances(self, io: Any) -> IssueList:
+        self._state._raise_exception_if_condition_not_met(
+            "Read RDF Instances",
+            empty_rules_store_required=True,
+        )
         reader = NeatReader.create(io)
         self._state.instances.store.write(extractors.RdfFileExtractor(reader.materialize_path()))
         return IssueList()

{cognite_neat-0.115.0 → cognite_neat-0.115.1}/cognite/neat/_session/_state.py

@@ -71,14 +71,23 @@ class SessionState:
         empty_instances_store_required: bool = False,
         instances_required: bool = False,
         client_required: bool = False,
+        has_information_rules: bool | None = None,
+        has_dms_rules: bool | None = None,
     ) -> None:
         """Set conditions for raising an error in the session that are used by various methods in the session."""
         condition = set()
         suggestion = set()
-
+        try_again = True
         if client_required and not self.client:
             condition.add(f"{activity} expects a client in NEAT session")
             suggestion.add("Please provide a client")
+        if has_information_rules is True and self.rule_store.try_get_last_information_rules is None:
+            condition.add(f"{activity} expects information rules in NEAT session")
+            suggestion.add("Read in information rules to neat session")
+        if has_dms_rules is False and self.rule_store.try_get_last_dms_rules is not None:
+            condition.add(f"{activity} expects no DMS data model in NEAT session")
+            suggestion.add("You already have a DMS data model in the session")
+            try_again = False
         if empty_rules_store_required and not self.rule_store.empty:
             condition.add(f"{activity} expects no data model in NEAT session")
             suggestion.add("Start new session")
@@ -90,7 +99,10 @@ class SessionState:
             suggestion.add("Read in instances to neat session")
 
         if condition:
-            raise NeatSessionError(". ".join(condition) + ". " + ". ".join(suggestion) + ". And try again.")
+            message = ". ".join(condition) + ". " + ". ".join(suggestion) + "."
+            if try_again:
+                message += " And try again."
+            raise NeatSessionError(message)
 
 
 class InstancesState:

{cognite_neat-0.115.0 → cognite_neat-0.115.1}/cognite/neat/_store/_rules_store.py

@@ -421,6 +421,20 @@ class NeatRulesStore:
         self._iteration_by_id[identifier] += 1
         return identifier + f"/Iteration_{self._iteration_by_id[identifier]}"
 
+    @property
+    def try_get_last_dms_rules(self) -> DMSRules | None:
+        if not self.provenance:
+            return None
+        if self.provenance[-1].target_entity.dms is None:
+            return None
+        return self.provenance[-1].target_entity.dms
+
+    @property
+    def try_get_last_information_rules(self) -> InformationRules | None:
+        if not self.provenance:
+            return None
+        return self.provenance[-1].target_entity.information
+
     @property
     def last_verified_dms_rules(self) -> DMSRules:
         if not self.provenance:

cognite_neat-0.115.1/cognite/neat/_version.py

@@ -0,0 +1,2 @@
+__version__ = "0.115.1"
+__engine__ = "^2.0.4"

{cognite_neat-0.115.0 → cognite_neat-0.115.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cognite-neat"
-version = "0.115.0"
+version = "0.115.1"
 readme = "README.md"
 description = "Knowledge graph transformation"
 authors = [

cognite_neat-0.115.0/cognite/neat/_version.py

@@ -1,2 +0,0 @@
-__version__ = "0.115.0"
-__engine__ = "^2.0.4"