cognite-neat 0.75.6__py3-none-any.whl → 0.75.8__py3-none-any.whl

cognite/neat/rules/models/_entity.py

@@ -1,142 +0,0 @@
-import re
-import sys
-from functools import total_ordering
-from typing import ClassVar
-
-from pydantic import BaseModel
-
-if sys.version_info >= (3, 11):
-    from enum import StrEnum
-    from typing import Self
-else:
-    from backports.strenum import StrEnum
-    from typing_extensions import Self
-
-
-class EntityTypes(StrEnum):
-    subject = "subject"
-    predicate = "predicate"
-    object = "object"
-    class_ = "class"
-    parent_class = "parent_class"
-    property_ = "property"
-    object_property = "ObjectProperty"
-    data_property = "DatatypeProperty"
-    annotation_property = "AnnotationProperty"
-    object_value_type = "object_value_type"
-    data_value_type = "data_value_type"  # these are strings, floats, ...
-    xsd_value_type = "xsd_value_type"
-    dms_value_type = "dms_value_type"
-    view = "view"
-    view_prop = "view_prop"
-    reference_entity = "reference_entity"
-    container = "container"
-    datamodel = "datamodel"
-    undefined = "undefined"
-
-
-# ALLOWED
-ALLOWED_PATTERN = r"[^a-zA-Z0-9-_.]"
-
-# FOR PARSING STRINGS:
-PREFIX_REGEX = r"[a-zA-Z]+[a-zA-Z0-9-_.]*[a-zA-Z0-9]+"
-SUFFIX_REGEX = r"[a-zA-Z0-9-_.]+[a-zA-Z0-9]|[-_.]*[a-zA-Z0-9]+"
-VERSION_REGEX = r"[a-zA-Z0-9]([.a-zA-Z0-9_-]{0,41}[a-zA-Z0-9])?"
-PROPERTY_REGEX = r"[a-zA-Z0-9][a-zA-Z0-9_-]*[a-zA-Z0-9]?"
-ENTITY_ID_REGEX = rf"{PREFIX_REGEX}:({SUFFIX_REGEX})"
-ENTITY_ID_REGEX_COMPILED = re.compile(rf"^(?P<prefix>{PREFIX_REGEX}):(?P<suffix>{SUFFIX_REGEX})$")
-VERSIONED_ENTITY_REGEX_COMPILED = re.compile(
-    rf"^(?P<prefix>{PREFIX_REGEX}):(?P<suffix>{SUFFIX_REGEX})\(version=(?P<version>{VERSION_REGEX})\)$"
-)
-CLASS_ID_REGEX = rf"(?P<{EntityTypes.class_}>{ENTITY_ID_REGEX})"
-CLASS_ID_REGEX_COMPILED = re.compile(rf"^{CLASS_ID_REGEX}$")
-PROPERTY_ID_REGEX = rf"\((?P<{EntityTypes.property_}>{ENTITY_ID_REGEX})\)"
-
-Undefined = type(object())
-Unknown = type(object())
-
-
-# mypy does not like the sentinel value, and it is not possible to ignore only the line with it below.
-# so we ignore all errors beyond this point.
-# mypy: ignore-errors
-@total_ordering
-class Entity(BaseModel, arbitrary_types_allowed=True):
-    """Entity is a class or property in OWL/RDF sense."""
-
-    type_: ClassVar[EntityTypes] = EntityTypes.undefined
-    prefix: str | Undefined = Undefined
-    suffix: str | Unknown
-    version: str | None = None
-    name: str | None = None
-    description: str | None = None
-
-    def __lt__(self, other: object) -> bool:
-        if type(self) is not type(other) or not isinstance(other, Entity):
-            return NotImplemented
-        return self.versioned_id < other.versioned_id
-
-    def __eq__(self, other: object) -> bool:
-        if type(self) is not type(other) or not isinstance(other, Entity):
-            return NotImplemented
-        return self.versioned_id == other.versioned_id
-
-    def __hash__(self) -> int:
-        return hash(self.versioned_id)
-
-    def as_non_versioned_entity(self) -> Self:
-        return self.from_string(f"{self.prefix}:{self.suffix}")
-
-    @property
-    def id(self) -> str:
-        if self.suffix is Unknown:
-            return "#N/A"
-        elif self.prefix is Undefined:
-            return self.suffix
-        else:
-            return f"{self.prefix}:{self.suffix}"
-
-    @property
-    def versioned_id(self) -> str:
-        if self.version is None:
-            return self.id
-        else:
-            return f"{self.id}(version={self.version})"
-
-    @property
-    def space(self) -> str:
-        """Returns entity space in CDF."""
-        return self.prefix
-
-    @property
-    def external_id(self) -> str:
-        """Returns entity external id in CDF."""
-        return self.suffix
-
-    def __repr__(self):
-        return self.versioned_id
-
-    def __str__(self):
-        return self.versioned_id
-
-    @classmethod
-    def from_string(cls, entity_string: str, base_prefix: str | None = None) -> Self:
-        if entity_string == "#N/A":
-            return cls(prefix=Undefined, suffix=Unknown)
-        elif result := VERSIONED_ENTITY_REGEX_COMPILED.match(entity_string):
-            return cls(
-                prefix=result.group("prefix"),
-                suffix=result.group("suffix"),
-                version=result.group("version"),
-            )
-        elif result := ENTITY_ID_REGEX_COMPILED.match(entity_string):
-            return cls(prefix=result.group("prefix"), suffix=result.group("suffix"))
-        elif base_prefix and re.match(SUFFIX_REGEX, entity_string) and re.match(PREFIX_REGEX, base_prefix):
-            return cls(prefix=base_prefix, suffix=entity_string)
-        else:
-            raise ValueError(f"{cls.__name__} is expected to be prefix:suffix, got {entity_string}")
-
-    @classmethod
-    def from_list(cls, entity_strings: list[str], base_prefix: str | None = None) -> list[Self]:
-        return [
-            cls.from_string(entity_string=entity_string, base_prefix=base_prefix) for entity_string in entity_strings
-        ]

cognite/neat/rules/models/rules/_types/_value.py

@@ -1,159 +0,0 @@
-from __future__ import annotations
-
-from datetime import date, datetime
-from typing import ClassVar, cast
-
-from cognite.client.data_classes.data_modeling import (
-    Boolean,
-    Date,
-    FileReference,
-    Float32,
-    Float64,
-    Int32,
-    Int64,
-    Json,
-    SequenceReference,
-    Text,
-    TimeSeriesReference,
-    Timestamp,
-)
-from pydantic import BaseModel
-
-from ._base import Entity, EntityTypes
-
-
-class ValueTypeMapping(BaseModel):
-    """Mapping between XSD, Python, DMS and Graphql types."""
-
-    xsd: str
-    python: type
-    dms: type
-    graphql: str
-    sql: str
-
-
-# mypy: ignore-errors
-class XSDValueType(Entity):
-    """Value type is a data/object type defined as a child of Entity model."""
-
-    type_: ClassVar[EntityTypes] = EntityTypes.xsd_value_type
-    mapping: ValueTypeMapping
-
-    @property
-    def python(self) -> type:
-        """Returns the Python type for a given value type."""
-        return cast(ValueTypeMapping, self.mapping).python
-
-    @property
-    def xsd(self) -> str:
-        """Returns the XSD type for a given value type."""
-        return cast(ValueTypeMapping, self.mapping).xsd
-
-    @property
-    def dms(self) -> type:
-        """Returns the DMS type for a given value type."""
-        return cast(ValueTypeMapping, self.mapping).dms
-
-    @property
-    def graphql(self) -> str:
-        """Returns the Graphql type for a given value type."""
-        return cast(ValueTypeMapping, self.mapping).graphql
-
-    @property
-    def sql(self) -> str:
-        """Returns the SQL type for a given value type."""
-        return cast(ValueTypeMapping, self.mapping).sql
-
-
-class DMSValueType(XSDValueType):
-    type_: ClassVar[EntityTypes] = EntityTypes.dms_value_type
-
-    def __str__(self) -> str:
-        return self.dms._type
-
-
-_DATA_TYPES: list[dict] = [
-    {"name": "boolean", "python": bool, "GraphQL": "Boolean", "dms": Boolean, "SQL": "BOOLEAN"},
-    {"name": "float", "python": float, "GraphQL": "Float", "dms": Float64, "SQL": "FLOAT"},
-    {"name": "double", "python": float, "GraphQL": "Float", "dms": Float64, "SQL": "DOUBLE"},
-    {"name": "integer", "python": int, "GraphQL": "Int", "dms": Int32, "SQL": "INTEGER"},
-    {"name": "nonPositiveInteger", "python": int, "GraphQL": "Int", "dms": Int32, "SQL": "INTEGER"},
-    {"name": "nonNegativeInteger", "python": int, "GraphQL": "Int", "dms": Int32, "SQL": "INTEGER"},
-    {"name": "negativeInteger", "python": int, "GraphQL": "Int", "dms": Int32, "SQL": "INTEGER"},
-    {"name": "long", "python": int, "GraphQL": "Int", "dms": Int64, "SQL": "INTEGER"},
-    {"name": "string", "python": str, "GraphQL": "String", "dms": Text, "SQL": "STRING"},
-    {"name": "langString", "python": str, "GraphQL": "String", "dms": Text, "SQL": "STRING"},
-    {"name": "anyURI", "python": str, "GraphQL": "String", "dms": Text, "SQL": "STRING"},
-    {"name": "normalizedString", "python": str, "GraphQL": "String", "dms": Text, "SQL": "STRING"},
-    {"name": "token", "python": str, "GraphQL": "String", "dms": Text, "SQL": "STRING"},
-    # Graphql does not have a datetime/date type this is CDF specific
-    {"name": "dateTime", "python": datetime, "GraphQL": "Timestamp", "dms": Timestamp, "SQL": "TIMESTAMP"},
-    {"name": "dateTimeStamp", "python": datetime, "GraphQL": "Timestamp", "dms": Timestamp, "SQL": "TIMESTAMP"},
-    {"name": "date", "python": date, "GraphQL": "String", "dms": Date, "SQL": "DATE"},
-    # Some RDF types which are not in XSD
-    {"name": "PlainLiteral", "python": str, "GraphQL": "String", "dms": Text, "SQL": "STRING"},
-    {"name": "Literal", "python": str, "GraphQL": "String", "dms": Text, "SQL": "STRING"},
-    # CDF specific types, not in XSD
-    {
-        "name": "timeseries",
-        "python": TimeSeriesReference,
-        "GraphQL": "TimeSeries",
-        "dms": TimeSeriesReference,
-        "SQL": "STRING",
-    },
-    {"name": "file", "python": FileReference, "GraphQL": "File", "dms": FileReference, "SQL": "STRING"},
-    {"name": "sequence", "python": SequenceReference, "GraphQL": "Sequence", "dms": SequenceReference, "SQL": "STRING"},
-    {"name": "json", "python": Json, "GraphQL": "Json", "dms": Json, "SQL": "STRING"},
-]
-
-_DMS_TYPES: list[dict] = [
-    {"name": "boolean", "python": bool, "GraphQL": "Boolean", "dms": Boolean, "SQL": "BOOLEAN"},
-    {"name": "float", "python": float, "GraphQL": "Float", "dms": Float32, "SQL": "FLOAT"},
-    {"name": "double", "python": float, "GraphQL": "Float", "dms": Float64, "SQL": "DOUBLE"},
-    {"name": "integer", "python": int, "GraphQL": "Int", "dms": Int32, "SQL": "INTEGER"},
-    {"name": "long", "python": int, "GraphQL": "Int", "dms": Int64, "SQL": "INTEGER"},
-    {"name": "string", "python": str, "GraphQL": "String", "dms": Text, "SQL": "STRING"},
-    {"name": "dateTimeStamp", "python": datetime, "GraphQL": "Timestamp", "dms": Timestamp, "SQL": "TIMESTAMP"},
-    {
-        "name": "timeseries",
-        "python": TimeSeriesReference,
-        "GraphQL": "TimeSeries",
-        "dms": TimeSeriesReference,
-        "SQL": "STRING",
-    },
-    {"name": "file", "python": FileReference, "GraphQL": "File", "dms": FileReference, "SQL": "STRING"},
-    {"name": "sequence", "python": SequenceReference, "GraphQL": "Sequence", "dms": SequenceReference, "SQL": "STRING"},
-    {"name": "json", "python": Json, "GraphQL": "Json", "dms": Json, "SQL": "STRING"},
-]
-
-XSD_VALUE_TYPE_MAPPINGS: dict[str, XSDValueType] = {
-    data_type["name"]: XSDValueType(
-        prefix="xsd",
-        suffix=cast(str, data_type["name"]),
-        name=cast(str, data_type["name"]),
-        mapping=ValueTypeMapping(
-            xsd=data_type["name"],
-            python=data_type["python"],
-            dms=data_type["dms"],
-            graphql=data_type["GraphQL"],
-            sql=data_type["SQL"],
-        ),
-    )
-    for data_type in _DATA_TYPES
-}
-
-DMS_VALUE_TYPE_MAPPINGS: dict[str, DMSValueType] = {
-    data_type["dms"]._type.casefold(): DMSValueType(
-        prefix="dms",
-        suffix=data_type["dms"]._type.casefold(),
-        name=data_type["dms"]._type.casefold(),
-        mapping=ValueTypeMapping(
-            xsd=data_type["name"],
-            python=data_type["python"],
-            dms=data_type["dms"],
-            graphql=data_type["GraphQL"],
-            sql=data_type["SQL"],
-        ),
-    )
-    for data_type in _DMS_TYPES
-}