from cognite.client.data_classes.data_modeling.query import Query, Select, NodeResultSetExpression, EdgeResultSetExpression, SourceSelector
from cognite.client.data_classes.filters import Range, Equals
from cognite.client.data_classes.data_modeling.ids import ViewId
movie_id = ViewId("mySpace", "MovieView", "v1")
actor_id = ViewId("mySpace", "ActorView", "v1")
query = Query(
    with_ = {
        "movies": NodeResultSetExpression(filter=Range(movie_id.as_property_ref("releaseYear"), lt=2000)),
        "actors_in_movie": EdgeResultSetExpression(from_="movies", filter=Equals(["edge", "type"], {"space": movie_id.space, "externalId": "Movie.actors"})),
        "actors": NodeResultSetExpression(from_="actors_in_movie"),
    },
    select = {
        "actors": Select(
            [SourceSelector(actor_id, ["name"])], sort=[InstanceSort(actor_id.as_property_ref("name"))]),
    },
)
res = client.data_modeling.instances.query(query)

from cognite.client.data_classes.data_modeling.data_types import UnitReference, UnitSystemReference
selected_source = SourceSelector(
    source=ViewId("my-space", "my-xid", "v1"),
    properties=["f32_prop1", "f32_prop2", "f64_prop1", "f64_prop2"],
    target_units=[
        TargetUnit("f32_prop1", UnitReference("pressure:kilopa")),
        TargetUnit("f32_prop2", UnitReference("pressure:barg")),
        TargetUnit("f64_prop1", UnitSystemReference("SI")),
        TargetUnit("f64_prop2", UnitSystemReference("Imperial")),
    ],
)

SourceSelector(source=ViewId("my-space", "my-xid", "v1"), properties=["*"])

{
  "items": {},
  "nextCursor": {},
  "typing": {},
  "debug": {
    "notices": [
      {
        "code": "excessiveTimeout",
        "category": "invalidDebugOptions",
        "level": "warning",
        "hint": "<string>",
        "timeout": 123
      }
    ],
    "translatedQuery": {
      "query": {},
      "parameters": {}
    },
    "plan": {
      "fullPlan": {},
      "profiled": true,
      "byIdentifier": {}
    },
    "llmPrompt": "<string>"
  }
}