cognite-neat 0.94.0__py3-none-any.whl → 0.95.0__py3-none-any.whl

cognite/neat/_rules/models/dms/_rules.py

@@ -5,6 +5,7 @@ from collections.abc import Hashable
 from datetime import datetime
 from typing import TYPE_CHECKING, Any, ClassVar, Literal
 
+import pandas as pd
 from cognite.client import data_modeling as dm
 from pydantic import Field, field_serializer, field_validator, model_validator
 from pydantic_core.core_schema import SerializationInfo, ValidationInfo
@@ -451,3 +452,18 @@ class DMSRules(BaseRules):
         from ._exporter import _DMSExporter
 
         return _DMSExporter(self, include_pipeline, instance_space).to_schema()
+
+    def _repr_html_(self) -> str:
+        summary = {
+            "type": "Physical Data Model",
+            "intended for": "DMS Architect",
+            "name": self.metadata.name,
+            "space": self.metadata.space,
+            "external_id": self.metadata.external_id,
+            "version": self.metadata.version,
+            "views": len(self.views),
+            "containers": len(self.containers) if self.containers else 0,
+            "properties": len(self.properties),
+        }
+
+        return pd.DataFrame([summary]).T.rename(columns={0: ""})._repr_html_()  # type: ignore

cognite/neat/_rules/models/dms/_rules_input.py

@@ -3,6 +3,7 @@ from dataclasses import dataclass
 from datetime import datetime
 from typing import Any, Literal
 
+import pandas as pd
 from cognite.client import data_modeling as dm
 
 from cognite.neat._rules.models._base_input import InputComponent, InputRules
@@ -291,3 +292,18 @@ class DMSInputRules(InputRules[DMSRules]):
             "Last": last,
             "Reference": reference,
         }
+
+    def _repr_html_(self) -> str:
+        summary = {
+            "type": "Physical Data Model",
+            "intended for": "DMS Architect",
+            "name": self.metadata.name,
+            "space": self.metadata.space,
+            "external_id": self.metadata.external_id,
+            "version": self.metadata.version,
+            "views": len(self.views),
+            "containers": len(self.containers) if self.containers else 0,
+            "properties": len(self.properties),
+        }
+
+        return pd.DataFrame([summary]).T.rename(columns={0: ""})._repr_html_()  # type: ignore

cognite/neat/_rules/models/information/_rules.py

@@ -4,6 +4,7 @@ from collections.abc import Hashable
 from datetime import datetime
 from typing import TYPE_CHECKING, Any, ClassVar
 
+import pandas as pd
 from pydantic import Field, field_serializer, field_validator, model_validator
 from pydantic_core.core_schema import SerializationInfo
 from rdflib import Namespace
@@ -380,3 +381,16 @@ class InformationRules(BaseRules):
         from cognite.neat._rules.transformers._converters import _InformationRulesConverter
 
         return _InformationRulesConverter(self).as_dms_rules()
+
+    def _repr_html_(self) -> str:
+        summary = {
+            "type": "Logical Data Model",
+            "intended for": "Information Architect",
+            "name": self.metadata.name,
+            "external_id": self.metadata.prefix,
+            "version": self.metadata.version,
+            "classes": len(self.classes),
+            "properties": len(self.properties),
+        }
+
+        return pd.DataFrame([summary]).T.rename(columns={0: ""})._repr_html_()  # type: ignore

cognite/neat/_rules/models/information/_rules_input.py

@@ -2,6 +2,7 @@ from dataclasses import dataclass, field
 from datetime import datetime
 from typing import Any, Literal
 
+import pandas as pd
 from rdflib import Namespace
 
 from cognite.neat._rules.models._base_input import InputComponent, InputRules
@@ -144,3 +145,16 @@ class InformationInputRules(InputRules[InformationRules]):
             Last=last,
             Reference=reference,
         )
+
+    def _repr_html_(self) -> str:
+        summary = {
+            "type": "Logical Data Model",
+            "intended for": "Information Architect",
+            "name": self.metadata.name,
+            "external_id": self.metadata.prefix,
+            "version": self.metadata.version,
+            "classes": len(self.classes),
+            "properties": len(self.properties),
+        }
+
+        return pd.DataFrame([summary]).T.rename(columns={0: ""})._repr_html_()  # type: ignore

cognite/neat/_session/_base.py

@@ -1,13 +1,11 @@
 from typing import Literal, cast
 
-import pandas as pd
 from cognite.client import CogniteClient
 
 from cognite.neat._issues import IssueList
 from cognite.neat._rules import importers
 from cognite.neat._rules._shared import ReadRules
 from cognite.neat._rules.models import DMSRules
-from cognite.neat._rules.models._base_input import InputComponent
 from cognite.neat._rules.models.information._rules import InformationRules
 from cognite.neat._rules.models.information._rules_input import InformationInputRules
 from cognite.neat._rules.transformers import ConvertToRules, VerifyAnyRules
@@ -72,13 +70,10 @@ class NeatSession:
 
         output = []
         if state.input_rules and not state.verified_rules:
-            metadata = cast(InputComponent, state.input_rule.rules.metadata)  # type: ignore[union-attr]
-            table = pd.DataFrame([metadata.dump()]).T._repr_html_()  # type: ignore[operator]
-            output.append(f"<H2>Raw Data Model</H2><br />{table}")
+            output.append(f"<H2>Raw Data Model</H2><br />{state.input_rule.rules._repr_html_()}")  # type: ignore
 
         if state.verified_rules:
-            table = pd.DataFrame([state.last_verified_rule.metadata.model_dump()]).T._repr_html_()  # type: ignore[operator]
-            output.append(f"<H2>Data Model</H2><br />{table}")
+            output.append(f"<H2>Data Model</H2><br />{state.last_verified_rule._repr_html_()}")  # type: ignore
 
         if state.has_store:
             output.append(f"<H2>Instances</H2> {state.store._repr_html_()}")

cognite/neat/_session/_prepare.py

@@ -9,8 +9,10 @@ from cognite.neat._rules.models.information._rules_input import InformationInput
 from cognite.neat._rules.transformers import ReduceCogniteModel, ToCompliantEntities, ToExtension
 
 from ._state import SessionState
+from .exceptions import intercept_session_exceptions
 
 
+@intercept_session_exceptions
 class PrepareAPI:
     def __init__(self, state: SessionState, verbose: bool) -> None:
         self._state = state
@@ -18,6 +20,7 @@ class PrepareAPI:
         self.data_model = DataModelPrepareAPI(state, verbose)
 
 
+@intercept_session_exceptions
 class DataModelPrepareAPI:
     def __init__(self, state: SessionState, verbose: bool) -> None:
         self._state = state

cognite/neat/_session/_read.py

@@ -12,8 +12,10 @@ from cognite.neat._rules._shared import ReadRules
 
 from ._state import SessionState
 from ._wizard import NeatObjectType, RDFFileType, object_wizard, rdf_dm_wizard
+from .exceptions import intercept_session_exceptions
 
 
+@intercept_session_exceptions
 class ReadAPI:
     def __init__(self, state: SessionState, client: CogniteClient | None, verbose: bool) -> None:
         self._state = state
@@ -23,6 +25,7 @@ class ReadAPI:
         self.excel = ExcelReadAPI(state, client, verbose)
 
 
+@intercept_session_exceptions
 class BaseReadAPI:
     def __init__(self, state: SessionState, client: CogniteClient | None, verbose: bool) -> None:
         self._state = state
@@ -47,6 +50,7 @@ class BaseReadAPI:
             raise ValueError(f"Expected str or Path, got {type(io)}")
 
 
+@intercept_session_exceptions
 class CDFReadAPI(BaseReadAPI):
     def data_model(self, data_model_id: DataModelIdentifier) -> IssueList:
         if self._client is None:
@@ -58,6 +62,7 @@ class CDFReadAPI(BaseReadAPI):
         return input_rules.issues
 
 
+@intercept_session_exceptions
 class ExcelReadAPI(BaseReadAPI):
     def __call__(self, io: Any) -> IssueList:
         filepath = self._return_filepath(io)
@@ -66,6 +71,7 @@ class ExcelReadAPI(BaseReadAPI):
         return input_rules.issues
 
 
+@intercept_session_exceptions
 class RDFReadAPI(BaseReadAPI):
     def __init__(self, state: SessionState, client: CogniteClient | None, verbose: bool) -> None:
         super().__init__(state, client, verbose)

cognite/neat/_session/_show.py

@@ -13,8 +13,10 @@ from cognite.neat._session.exceptions import NeatSessionError
 from cognite.neat._utils.rdf_ import remove_namespace_from_uri
 
 from ._state import SessionState
+from .exceptions import intercept_session_exceptions
 
 
+@intercept_session_exceptions
 class ShowAPI:
     def __init__(self, state: SessionState) -> None:
         self._state = state
@@ -22,6 +24,7 @@ class ShowAPI:
         self.instances = ShowInstanceAPI(self._state)
 
 
+@intercept_session_exceptions
 class ShowInstanceAPI:
     def __init__(self, state: SessionState) -> None:
         self._state = state
@@ -134,6 +137,7 @@ class ShowInstanceAPI:
         return template
 
 
+@intercept_session_exceptions
 class ShowDataModelAPI:
     def __init__(self, state: SessionState) -> None:
         self._state = state

cognite/neat/_session/_to.py

@@ -8,8 +8,10 @@ from cognite.neat._rules import exporters
 from cognite.neat._session._wizard import space_wizard
 
 from ._state import SessionState
+from .exceptions import intercept_session_exceptions
 
 
+@intercept_session_exceptions
 class ToAPI:
     def __init__(self, state: SessionState, client: CogniteClient | None, verbose: bool) -> None:
         self._state = state
@@ -39,6 +41,7 @@ class ToAPI:
         return None
 
 
+@intercept_session_exceptions
 class CDFToAPI:
     def __init__(self, state: SessionState, client: CogniteClient | None, verbose: bool) -> None:
         self._client = client

cognite/neat/_session/exceptions.py

@@ -23,6 +23,8 @@ def _intercept_session_exceptions(func: Callable):
             return func(*args, **kwargs)
         except NeatSessionError as e:
             action = func.__name__
+            if action == "__call__":
+                action = func.__qualname__.removesuffix(".__call__").removesuffix("API")
             print(f"{_PREFIX} Cannot {action}: {e}")
 
     return wrapper
@@ -33,10 +35,11 @@ def intercept_session_exceptions(cls: type):
     while to_check:
         cls = to_check.pop()
         for attr_name in dir(cls):
-            if not attr_name.startswith("_"):
-                attr = getattr(cls, attr_name)
-                if callable(attr):
-                    setattr(cls, attr_name, _intercept_session_exceptions(attr))
-                elif isinstance(attr, type):
-                    to_check.append(attr)
+            if attr_name.startswith("_") and not attr_name == "__call__":
+                continue
+            attr = getattr(cls, attr_name)
+            if callable(attr):
+                setattr(cls, attr_name, _intercept_session_exceptions(attr))
+            elif isinstance(attr, type):
+                to_check.append(attr)
     return cls

cognite/neat/_version.py

@@ -1 +1 @@
-__version__ = "0.94.0"
+__version__ = "0.95.0"

{cognite_neat-0.94.0.dist-info → cognite_neat-0.95.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cognite-neat
-Version: 0.94.0
+Version: 0.95.0
 Summary: Knowledge graph transformation
 Home-page: https://cognite-neat.readthedocs-hosted.com/
 License: Apache-2.0
@@ -30,7 +30,7 @@ Requires-Dist: google-auth-oauthlib ; extra == "google"
 Requires-Dist: gspread ; extra == "google"
 Requires-Dist: ipycytoscape (>=1.3.3,<2.0.0)
 Requires-Dist: jinja2 (>=3.1.2,<4.0.0) ; extra == "graphql" or extra == "all"
-Requires-Dist: matplotlib (>=3.9.2,<4.0.0)
+Requires-Dist: matplotlib (==3.5.2)
 Requires-Dist: mkdocs ; extra == "docs"
 Requires-Dist: mkdocs-autorefs (>=0.5.0,<0.6.0) ; extra == "docs"
 Requires-Dist: mkdocs-git-authors-plugin ; extra == "docs"

{cognite_neat-0.94.0.dist-info → cognite_neat-0.95.0.dist-info}/RECORD

@@ -186,8 +186,8 @@ cognite/neat/_rules/models/asset/_validation.py,sha256=6u86qLlr1ehG1I5kIZhfCYTqU
 cognite/neat/_rules/models/data_types.py,sha256=ACQjS9vs7EjaNSO8R1kHiiNYbdeVQLCvxJYfZhWF8mM,9620
 cognite/neat/_rules/models/dms/__init__.py,sha256=CUqUlVjz4yZX_-61F-2ofSoV7N9MlSYx2N7vM-omp7E,640
 cognite/neat/_rules/models/dms/_exporter.py,sha256=dZAl5439p2K6LsA4yNpwCsEgNsxVkGjcmV9JGL3JAYE,30345
-cognite/neat/_rules/models/dms/_rules.py,sha256=8VlGchvUHtBrcaP6DhpkQn6qBgd2zlmeQoakh09axuk,19369
-cognite/neat/_rules/models/dms/_rules_input.py,sha256=LgehVseHiKLKfc7Osg3q8n1H64DUIe2iRDayh6Be4io,10919
+cognite/neat/_rules/models/dms/_rules.py,sha256=JCYss9Nqf2tloyZj4gZktaW_45gMCjw7aLf5XeD_kRk,19980
+cognite/neat/_rules/models/dms/_rules_input.py,sha256=MjkkoEa5lRzQE0NUJ7TyQsrelWSoyGCCDFjMVG5pQvc,11530
 cognite/neat/_rules/models/dms/_schema.py,sha256=HSmSDvOm5S0x4Vb9tH9Jvd5i9tXiiM08E_Sdu6q_iA8,50783
 cognite/neat/_rules/models/dms/_validation.py,sha256=2KL1_GNnJ5cjsdk9uJTBV691_o8ZehqmgXSiaN5q8aw,15798
 cognite/neat/_rules/models/domain.py,sha256=HOZ6e6pIWJWWYTTDPDWpuwEHoiUacOdkvjAI81vMdMk,3999
@@ -199,8 +199,8 @@ cognite/neat/_rules/models/entities/_single_value.py,sha256=JLX8t-Uft0qeY_honmv0
 cognite/neat/_rules/models/entities/_types.py,sha256=df9rnXJJKciv2Bp-Ve2q4xdEJt6WWniq12Z0hW2d6sk,1917
 cognite/neat/_rules/models/entities/_wrapped.py,sha256=FxC8HztW_tUUtuArAOwxyFfkdJnSEB4bgZoNmmmfiPk,7137
 cognite/neat/_rules/models/information/__init__.py,sha256=fVvgXt-JuyZCP_mLgIVaeKD9pdAXe2BWUxU_BZs8e5g,480
-cognite/neat/_rules/models/information/_rules.py,sha256=myMmtJLJs1nF81Su96wmDTM9-DP7dF-Ezs41RlmqMzk,14869
-cognite/neat/_rules/models/information/_rules_input.py,sha256=LqWJM7D4kjPPC3Ok7WbsqwYpbSIk8hQun22JK8gVUFw,5277
+cognite/neat/_rules/models/information/_rules.py,sha256=BXsQvDrtHBrLzLZt9Q9QGHfzW8OeGdJ-M2dT_TcdocE,15370
+cognite/neat/_rules/models/information/_rules_input.py,sha256=c7JPGe0n0srF2LM9x5_5EtMx8-8EPEZ_HDImuR9tV2I,5778
 cognite/neat/_rules/models/information/_validation.py,sha256=0ntsrXJV28yOFfhhXcKsP9TUr0LwKG_vaYPMHzHbTEI,9252
 cognite/neat/_rules/models/mapping/__init__.py,sha256=jSn-dCckmVQF0ClSBOBvVacprzNxdhPpdyIlYVajjMY,198
 cognite/neat/_rules/models/mapping/_base.py,sha256=xDjtbNvDXAh0F93WSvgdwoI_5K9XYu_w9hhG02ZZKtg,5922
@@ -212,14 +212,14 @@ cognite/neat/_rules/transformers/_mapping.py,sha256=RWHKPMaP3JdeCNvoDGu9ZGHxfyeI
 cognite/neat/_rules/transformers/_pipelines.py,sha256=-E5Hgitnr6ee8R9_3sqtjmWIPJ0w1xaLErG6Fo6ExVU,2603
 cognite/neat/_rules/transformers/_verification.py,sha256=330I1i3Va7Nau3DK3bTgEEGSHppmLvT-1yBMocDxV1A,4694
 cognite/neat/_session/__init__.py,sha256=fxQ5URVlUnmEGYyB8Baw7IDq-uYacqkigbc4b-Pr9Fw,58
-cognite/neat/_session/_base.py,sha256=9QgOUeizAwG9P7xy5J8zOsRd5cMMkncYgv6PTeWgd_k,3564
-cognite/neat/_session/_prepare.py,sha256=u7dJ4nZqB9j6JWoBx6em9ABKc_PA_yaRQWfiI1n3iks,2484
-cognite/neat/_session/_read.py,sha256=JQyxG1-QyX8fbiUnwVexKzlgWMg9VUeTJvC0uFgAg_U,4401
-cognite/neat/_session/_show.py,sha256=q1cuqpDRaNS1hCwgvpxk-nR1V3pcYPyQdyRn5zUMGWQ,10217
+cognite/neat/_session/_base.py,sha256=VSYrKWCgyzV4RL8X7TESNDEdm-LBs6yz6OKl0EfrxxA,3250
+cognite/neat/_session/_prepare.py,sha256=ZG8bJiLbKsQ5iEwPOnl7rr5pRl0cnVXZ0N-nCSLLrA8,2597
+cognite/neat/_session/_read.py,sha256=ONYiNrGmN6G5iKsk30rg4OgddwLnbPBDMQKamNx3XYA,4604
+cognite/neat/_session/_show.py,sha256=VZfgJIfEmU1lwejxcSW5sax0otLsTXi-waBGATfC6gc,10360
 cognite/neat/_session/_state.py,sha256=ef7OiCxavk4_asIivENaWrvC3HcXZNV6wGldus55JD0,2488
-cognite/neat/_session/_to.py,sha256=u--yGD2y9JKV-sf34PYsMf-nc9Qs2yibpPsv6IG0CLA,2125
+cognite/neat/_session/_to.py,sha256=3StR71QNEb8JKW-O2iI3aNTlT-wRRaabDg6dovMHWRw,2238
 cognite/neat/_session/_wizard.py,sha256=Rdld2eZ-Q5BYbmAwW8FlfAYebdlw_P6L6V2WSDk-dHI,1306
-cognite/neat/_session/exceptions.py,sha256=qT0Mq3h-0NaDatvyr2D476LkSjmvyZtNmmLXtd1SLVg,1106
+cognite/neat/_session/exceptions.py,sha256=tVearBcnz7gq0Wn5f7qssCUML_4oxrEvTCj6QnIcCxo,1265
 cognite/neat/_shared.py,sha256=RSaHm2eJceTlvb-hMMe4nHgoHdPYDfN3XcxDXo24k3A,1530
 cognite/neat/_store/__init__.py,sha256=G-VG_YwfRt1kuPao07PDJyZ3w_0-eguzLUM13n-Z_RA,64
 cognite/neat/_store/_base.py,sha256=xi9uxV12X5XcGW-fzTEeDfaTM6Rkt-KpGaeyLXqep4Q,14205
@@ -240,7 +240,7 @@ cognite/neat/_utils/text.py,sha256=PvTEsEjaTu8SE8yYaKUrce4msboMj933dK7-0Eey_rE,3
 cognite/neat/_utils/time_.py,sha256=O30LUiDH9TdOYz8_a9pFqTtJdg8vEjC3qHCk8xZblG8,345
 cognite/neat/_utils/upload.py,sha256=X5GGWHswnW0BrL2ulmm5MnKKtn-t1C8Ps3gb9Byc914,4016
 cognite/neat/_utils/xml_.py,sha256=FQkq84u35MUsnKcL6nTMJ9ajtG9D5i1u4VBnhGqP2DQ,1710
-cognite/neat/_version.py,sha256=BbCpQB0IaKmq_fOtysTb_iVtvqbH-vHS1DoS1eLd4w8,23
+cognite/neat/_version.py,sha256=1KBvo1_4ig2ZeMfCvFApYtfDll8Ysfu6h45ByRERz1E,23
 cognite/neat/_workflows/__init__.py,sha256=S0fZq7kvoqDKodHu1UIPsqcpdvXoefUWRPt1lqeQkQs,420
 cognite/neat/_workflows/base.py,sha256=O1pcmfbme2gIVF2eOGrKZSUDmhZc8L9rI8UfvLN2YAM,26839
 cognite/neat/_workflows/cdf_store.py,sha256=3pebnATPo6In4-1srpa3wzstynTOi3T6hwFX5uaie4c,18050
@@ -269,8 +269,8 @@ cognite/neat/_workflows/tasks.py,sha256=dr2xuIb8P5e5e9p_fjzRlvDbKsre2xGYrkc3wnRx
 cognite/neat/_workflows/triggers.py,sha256=u69xOsaTtM8_WD6ZeIIBB-XKwvlZmPHAsZQh_TnyHcM,7073
 cognite/neat/_workflows/utils.py,sha256=gKdy3RLG7ctRhbCRwaDIWpL9Mi98zm56-d4jfHDqP1E,453
 cognite/neat/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cognite_neat-0.94.0.dist-info/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
-cognite_neat-0.94.0.dist-info/METADATA,sha256=aKSpijqnnPjjnO0NTC98tdm4YBbP67wwGz8tM_DSnus,9668
-cognite_neat-0.94.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-cognite_neat-0.94.0.dist-info/entry_points.txt,sha256=SsQlnl8SNMSSjE3acBI835JYFtsIinLSbVmHmMEXv6E,51
-cognite_neat-0.94.0.dist-info/RECORD,,
+cognite_neat-0.95.0.dist-info/LICENSE,sha256=W8VmvFia4WHa3Gqxq1Ygrq85McUNqIGDVgtdvzT-XqA,11351
+cognite_neat-0.95.0.dist-info/METADATA,sha256=GGSZ4F6EOmFOFIn6ondR3nEJeOe1LLWolqm48mR4wis,9661
+cognite_neat-0.95.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+cognite_neat-0.95.0.dist-info/entry_points.txt,sha256=SsQlnl8SNMSSjE3acBI835JYFtsIinLSbVmHmMEXv6E,51
+cognite_neat-0.95.0.dist-info/RECORD,,