cognite-toolkit 0.7.44__py3-none-any.whl → 0.7.46__py3-none-any.whl

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/data_modeling/_references.py

@@ -2,7 +2,7 @@ from typing import Literal
 
 from pydantic import Field
 
-from cognite_toolkit._cdf_tk.client.data_classes.identifiers import Identifier
+from cognite_toolkit._cdf_tk.client.resource_classes.identifiers import Identifier
 
 
 class SpaceReference(Identifier):

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/data_modeling/_space.py

@@ -1,6 +1,6 @@
 from abc import ABC
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import BaseModelObject, RequestResource, ResponseResource
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestResource, ResponseResource
 
 from ._references import SpaceReference
 

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/data_modeling/_view.py

@@ -4,7 +4,7 @@ from typing import Any, Literal
 from pydantic import Field, JsonValue, field_serializer, model_validator
 from pydantic_core.core_schema import FieldSerializationInfo
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import BaseModelObject, RequestResource, ResponseResource
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestResource, ResponseResource
 
 from ._data_types import DirectNodeRelation
 from ._references import ContainerReference, ViewReference

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/data_modeling/_view_property.py

@@ -4,7 +4,7 @@ from typing import Annotated, Any, Literal
 from pydantic import Field, JsonValue, TypeAdapter, field_serializer
 from pydantic_core.core_schema import FieldSerializationInfo
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import BaseModelObject
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject
 
 from ._data_types import DataType
 from ._references import ContainerDirectReference, ContainerReference, NodeReference, ViewDirectReference, ViewReference

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/dataset.py

@@ -1,6 +1,6 @@
 from typing import ClassVar
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestUpdateable,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/event.py

@@ -1,6 +1,6 @@
 from typing import ClassVar
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestUpdateable,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/extraction_pipeline.py

@@ -1,6 +1,6 @@
 from typing import ClassVar
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import (
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
     BaseModelObject,
     RequestUpdateable,
     ResponseResource,

cognite_toolkit/_cdf_tk/client/{data_classes → resource_classes}/filemetadata.py

@@ -2,7 +2,7 @@ from typing import ClassVar, Literal
 
 from pydantic import Field, JsonValue
 
-from cognite_toolkit._cdf_tk.client.data_classes.base import BaseModelObject, RequestUpdateable, ResponseResource
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestUpdateable, ResponseResource
 
 from .identifiers import ExternalId
 from .instance_api import NodeReference

cognite_toolkit/_cdf_tk/client/resource_classes/function.py

@@ -0,0 +1,53 @@
+from typing import Literal, TypeAlias
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import BaseModelObject, RequestResource, ResponseResource
+
+from .identifiers import ExternalId
+
+FunctionStatus: TypeAlias = Literal["Queued", "Deploying", "Ready", "Failed"]
+
+
+class FunctionBase(BaseModelObject):
+    """Base class for Function with common fields."""
+
+    external_id: str | None = None
+    name: str
+    file_id: int
+    description: str | None = None
+    owner: str | None = None
+    function_path: str | None = None
+    secrets: dict[str, str] | None = None
+    env_vars: dict[str, str] | None = None
+    cpu: float | None = None
+    memory: float | None = None
+    runtime: Literal["py310", "py311", "py312"] | None = None
+    runtime_version: str | None = None
+    metadata: dict[str, str] | None = None
+    index_url: str | None = None
+    extra_index_urls: list[str] | None = None
+
+
+class FunctionRequest(FunctionBase, RequestResource):
+    """Request resource for creating/updating functions."""
+
+    def as_id(self) -> ExternalId:
+        if self.external_id is None:
+            raise ValueError("Cannot create ExternalId: external_id is None")
+        return ExternalId(external_id=self.external_id)
+
+
+class FunctionAPIError(BaseModelObject):
+    code: int
+    message: str
+
+
+class FunctionResponse(FunctionBase, ResponseResource[FunctionRequest]):
+    """Response resource for functions."""
+
+    id: int
+    created_time: int
+    status: FunctionStatus | None = None
+    error: FunctionAPIError | None = None
+
+    def as_request_resource(self) -> FunctionRequest:
+        return FunctionRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/function_schedule.py

@@ -0,0 +1,61 @@
+from pydantic import Field, JsonValue
+
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    Identifier,
+    RequestResource,
+    ResponseResource,
+)
+
+
+class FunctionScheduleId(Identifier):
+    """Identifier for a function schedule."""
+
+    function_external_id: str
+    name: str
+
+    def __str__(self) -> str:
+        return f"functionExternalId='{self.function_external_id}', name='{self.name}'"
+
+
+class FunctionScheduleCredentials(BaseModelObject):
+    """Credentials for function schedule authentication."""
+
+    client_id: str
+    client_secret: str
+
+
+class FunctionSchedule(BaseModelObject):
+    """Base class for function schedule with common fields."""
+
+    name: str
+    cron_expression: str
+    description: str | None = None
+    data: dict[str, JsonValue] | None = None
+
+
+class FunctionScheduleRequest(FunctionSchedule, RequestResource):
+    """Request resource for creating/updating function schedules."""
+
+    function_id: int
+    function_external_id: str | None = Field(None, exclude=True)
+    nonce: str
+
+    def as_id(self) -> FunctionScheduleId:
+        if self.function_external_id is None:
+            raise ValueError("function_external_id must be set to create FunctionScheduleId")
+        return FunctionScheduleId(function_external_id=self.function_external_id, name=self.name)
+
+
+class FunctionScheduleResponse(FunctionSchedule, ResponseResource[FunctionScheduleRequest]):
+    """Response resource for function schedules."""
+
+    id: int
+    created_time: int
+    when: str
+    function_id: int | None = None
+    function_external_id: str | None = None
+    session_id: int | None = None
+
+    def as_request_resource(self) -> FunctionScheduleRequest:
+        raise NotImplementedError("Cannot convert to request resource as 'nonce' is missing.")

cognite_toolkit/_cdf_tk/client/resource_classes/graphql_data_model.py

@@ -0,0 +1,40 @@
+from cognite_toolkit._cdf_tk.client.resource_classes.base import (
+    BaseModelObject,
+    RequestResource,
+    ResponseResource,
+)
+
+from .data_modeling import DataModelReference, ViewReference
+
+
+class GraphQLDataModel(BaseModelObject):
+    """Base class for GraphQL data model with common fields."""
+
+    space: str
+    external_id: str
+    version: str
+    name: str | None = None
+    description: str | None = None
+
+
+class GraphQLDataModelRequest(GraphQLDataModel, RequestResource):
+    """Request resource for creating/updating GraphQL data models."""
+
+    previous_version: str | None = None
+    dml: str | None = None
+    preserve_dml: bool | None = None
+
+    def as_id(self) -> DataModelReference:
+        return DataModelReference(space=self.space, external_id=self.external_id, version=self.version)
+
+
+class GraphQLDataModelResponse(GraphQLDataModel, ResponseResource[GraphQLDataModelRequest]):
+    """Response resource for GraphQL data models."""
+
+    is_global: bool
+    created_time: int
+    last_updated_time: int
+    views: list[ViewReference] | None = None
+
+    def as_request_resource(self) -> GraphQLDataModelRequest:
+        return GraphQLDataModelRequest.model_validate(self.dump(), extra="ignore")

cognite_toolkit/_cdf_tk/client/resource_classes/group/__init__.py

@@ -0,0 +1,187 @@
+"""Group data classes for the Cognite API Groups endpoint.
+
+Based on the API specification at:
+https://api-docs.cognite.com/20230101/tag/Groups/operation/createGroups
+"""
+
+from .acls import (
+    Acl,
+    AgentsAcl,
+    AnalyticsAcl,
+    AnnotationsAcl,
+    AppConfigAcl,
+    AssetsAcl,
+    AuditlogAcl,
+    DataModelInstancesAcl,
+    DataModelsAcl,
+    DataSetsAcl,
+    DiagramParsingAcl,
+    DigitalTwinAcl,
+    DocumentFeedbackAcl,
+    DocumentPipelinesAcl,
+    EntityMatchingAcl,
+    EventsAcl,
+    ExperimentsAcl,
+    ExtractionConfigsAcl,
+    ExtractionPipelinesAcl,
+    ExtractionRunsAcl,
+    FilePipelinesAcl,
+    FilesAcl,
+    FunctionsAcl,
+    GeospatialAcl,
+    GeospatialCrsAcl,
+    GroupsAcl,
+    HostedExtractorsAcl,
+    LabelsAcl,
+    LegacyGenericsAcl,
+    LegacyModelHostingAcl,
+    LocationFiltersAcl,
+    MonitoringTasksAcl,
+    NotificationsAcl,
+    PipelinesAcl,
+    PostgresGatewayAcl,
+    ProjectsAcl,
+    RawAcl,
+    RelationshipsAcl,
+    RoboticsAcl,
+    SAPWritebackAcl,
+    SAPWritebackRequestsAcl,
+    ScheduledCalculationsAcl,
+    SecurityCategoriesAcl,
+    SeismicAcl,
+    SequencesAcl,
+    SessionsAcl,
+    SimulatorsAcl,
+    StreamRecordsAcl,
+    StreamsAcl,
+    TemplateGroupsAcl,
+    TemplateInstancesAcl,
+    ThreeDAcl,
+    TimeSeriesAcl,
+    TimeSeriesSubscriptionsAcl,
+    TransformationsAcl,
+    TypesAcl,
+    UnknownAcl,
+    UserProfilesAcl,
+    VideoStreamingAcl,
+    VisionModelAcl,
+    WellsAcl,
+    WorkflowOrchestrationAcl,
+)
+from .capability import (
+    GroupCapability,
+    ProjectUrlNames,
+)
+from .group import (
+    Group,
+    GroupAttributes,
+    GroupRequest,
+    GroupResponse,
+    TokenAttributes,
+)
+from .scopes import (
+    AllScope,
+    AppConfigScope,
+    AssetRootIDScope,
+    CurrentUserScope,
+    DataSetScope,
+    ExperimentScope,
+    ExtractionPipelineScope,
+    IDScope,
+    IDScopeLowerCase,
+    InstancesScope,
+    PartitionScope,
+    PostgresGatewayUsersScope,
+    Scope,
+    ScopeDefinition,
+    SpaceIDScope,
+    TableScope,
+)
+
+__all__ = [
+    "Acl",
+    "AgentsAcl",
+    "AllScope",
+    "AnalyticsAcl",
+    "AnnotationsAcl",
+    "AppConfigAcl",
+    "AppConfigScope",
+    "AssetRootIDScope",
+    "AssetsAcl",
+    "AuditlogAcl",
+    "CurrentUserScope",
+    "DataModelInstancesAcl",
+    "DataModelsAcl",
+    "DataSetScope",
+    "DataSetsAcl",
+    "DiagramParsingAcl",
+    "DigitalTwinAcl",
+    "DocumentFeedbackAcl",
+    "DocumentPipelinesAcl",
+    "EntityMatchingAcl",
+    "EventsAcl",
+    "ExperimentScope",
+    "ExperimentsAcl",
+    "ExtractionConfigsAcl",
+    "ExtractionPipelineScope",
+    "ExtractionPipelinesAcl",
+    "ExtractionRunsAcl",
+    "FilePipelinesAcl",
+    "FilesAcl",
+    "FunctionsAcl",
+    "GeospatialAcl",
+    "GeospatialCrsAcl",
+    "Group",
+    "GroupAttributes",
+    "GroupCapability",
+    "GroupRequest",
+    "GroupResponse",
+    "GroupsAcl",
+    "HostedExtractorsAcl",
+    "IDScope",
+    "IDScopeLowerCase",
+    "InstancesScope",
+    "LabelsAcl",
+    "LegacyGenericsAcl",
+    "LegacyModelHostingAcl",
+    "LocationFiltersAcl",
+    "MonitoringTasksAcl",
+    "NotificationsAcl",
+    "PartitionScope",
+    "PipelinesAcl",
+    "PostgresGatewayAcl",
+    "PostgresGatewayUsersScope",
+    "ProjectUrlNames",
+    "ProjectsAcl",
+    "RawAcl",
+    "RelationshipsAcl",
+    "RoboticsAcl",
+    "SAPWritebackAcl",
+    "SAPWritebackRequestsAcl",
+    "ScheduledCalculationsAcl",
+    "Scope",
+    "ScopeDefinition",
+    "SecurityCategoriesAcl",
+    "SeismicAcl",
+    "SequencesAcl",
+    "SessionsAcl",
+    "SimulatorsAcl",
+    "SpaceIDScope",
+    "StreamRecordsAcl",
+    "StreamsAcl",
+    "TableScope",
+    "TemplateGroupsAcl",
+    "TemplateInstancesAcl",
+    "ThreeDAcl",
+    "TimeSeriesAcl",
+    "TimeSeriesSubscriptionsAcl",
+    "TokenAttributes",
+    "TransformationsAcl",
+    "TypesAcl",
+    "UnknownAcl",
+    "UserProfilesAcl",
+    "VideoStreamingAcl",
+    "VisionModelAcl",
+    "WellsAcl",
+    "WorkflowOrchestrationAcl",
+]

cognite_toolkit/_cdf_tk/client/resource_classes/group/_constants.py

@@ -0,0 +1,2 @@
+ACL_NAME = "acl_name"
+SCOPE_NAME = "scope_name"