cognite-toolkit 0.7.58__py3-none-any.whl → 0.7.60__py3-none-any.whl

cognite_toolkit/_cdf_tk/apps/_dump_app.py

@@ -17,6 +17,7 @@ from cognite_toolkit._cdf_tk.commands.dump_resource import (
     GroupFinder,
     LocationFilterFinder,
     NodeFinder,
+    ResourceViewMappingFinder,
     SearchConfigFinder,
     SpaceFinder,
     StreamlitFinder,
@@ -52,6 +53,8 @@ class DumpApp(typer.Typer):
         self.command("agents")(DumpConfigApp.dump_agents)
 
         self.command("search-config")(DumpConfigApp.dump_search_config)
+        if Flags.MIGRATE.is_enabled():
+            self.command("resource-view-mapping")(DumpConfigApp.dump_resource_view_mapping)
 
     @staticmethod
     def dump_main(ctx: typer.Context) -> None:
@@ -79,6 +82,8 @@ class DumpConfigApp(typer.Typer):
         self.command("streamlit")(DumpConfigApp.dump_streamlit)
         self.command("agents")(self.dump_agents)
         self.command("search-config")(self.dump_search_config)
+        if Flags.MIGRATE.is_enabled():
+            self.command("resource-view-mapping")(self.dump_resource_view_mapping)
 
     @staticmethod
     def dump_config_main(ctx: typer.Context) -> None:
@@ -758,6 +763,54 @@ class DumpConfigApp(typer.Typer):
             )
         )
 
+    @staticmethod
+    def dump_resource_view_mapping(
+        ctx: typer.Context,
+        external_id: Annotated[
+            list[str] | None,
+            typer.Argument(
+                help="The external ID(s) of the resource view mapping(s) you want to dump. "
+                "If nothing is provided, an interactive prompt will be shown to select the resource view mappings.",
+            ),
+        ] = None,
+        output_dir: Annotated[
+            Path,
+            typer.Option(
+                "--output-dir",
+                "-o",
+                help="Where to dump the resource view mapping files.",
+                allow_dash=True,
+            ),
+        ] = Path("tmp"),
+        clean: Annotated[
+            bool,
+            typer.Option(
+                "--clean",
+                "-c",
+                help="Delete the output directory before dumping the resource view mappings.",
+            ),
+        ] = False,
+        verbose: Annotated[
+            bool,
+            typer.Option(
+                "--verbose",
+                "-v",
+                help="Turn on to get more verbose output when running the command",
+            ),
+        ] = False,
+    ) -> None:
+        """[MIGRATION] This command will dump the selected resource view mapping(s) as yaml to the folder specified, defaults to /tmp."""
+        client = EnvironmentVariables.create_from_environment().get_client()
+        cmd = DumpResourceCommand(client=client)
+        cmd.run(
+            lambda: cmd.dump_to_yamls(
+                ResourceViewMappingFinder(client, tuple(external_id) if external_id else None),
+                output_dir=output_dir,
+                clean=clean,
+                verbose=verbose,
+            )
+        )
+
 
 class DumpDataApp(typer.Typer):
     def __init__(self, *args: Any, **kwargs: Any) -> None:

cognite_toolkit/_cdf_tk/client/http_client/_data_classes.py

@@ -1,10 +1,10 @@
 import gzip
 from abc import ABC, abstractmethod
-from typing import Any, Literal
+from typing import TYPE_CHECKING, Any, Literal
 
 import httpx
 from cognite.client import global_config
-from pydantic import TYPE_CHECKING, BaseModel, JsonValue, TypeAdapter, model_validator
+from pydantic import BaseModel, JsonValue, TypeAdapter, model_validator
 
 from cognite_toolkit._cdf_tk.client.http_client._exception import ToolkitAPIError
 from cognite_toolkit._cdf_tk.utils.useful_types import PrimitiveType

cognite_toolkit/_cdf_tk/commands/dump_resource.py

@@ -28,7 +28,7 @@ from cognite.client.data_classes._base import (
 from cognite.client.data_classes.agents import (
     AgentList,
 )
-from cognite.client.data_classes.data_modeling import DataModelId
+from cognite.client.data_classes.data_modeling import DataModelId, NodeList
 from cognite.client.data_classes.documents import SourceFileProperty
 from cognite.client.data_classes.extractionpipelines import ExtractionPipelineConfigList
 from cognite.client.data_classes.functions import (
@@ -53,6 +53,7 @@ from rich.panel import Panel
 
 from cognite_toolkit._cdf_tk.client import ToolkitClient
 from cognite_toolkit._cdf_tk.client.resource_classes.legacy.location_filters import LocationFilterList
+from cognite_toolkit._cdf_tk.client.resource_classes.legacy.migration import ResourceViewMapping
 from cognite_toolkit._cdf_tk.client.resource_classes.legacy.search_config import SearchConfigList
 from cognite_toolkit._cdf_tk.client.resource_classes.legacy.search_config import ViewId as SearchConfigViewId
 from cognite_toolkit._cdf_tk.client.resource_classes.legacy.streamlit_ import Streamlit, StreamlitList
@@ -69,6 +70,7 @@ from cognite_toolkit._cdf_tk.cruds import (
     LocationFilterCRUD,
     NodeCRUD,
     ResourceCRUD,
+    ResourceViewMappingCRUD,
     SearchConfigCRUD,
     SpaceCRUD,
     StreamlitCRUD,
@@ -797,6 +799,44 @@ class SearchConfigFinder(ResourceFinder[tuple[SearchConfigViewId, ...]]):
             yield list(self.identifier), None, loader, None
 
 
+class ResourceViewMappingFinder(ResourceFinder[tuple[str, ...]]):
+    def __init__(self, client: ToolkitClient, identifier: tuple[str, ...] | None = None):
+        super().__init__(client, identifier)
+        self.resource_view_mappings: list[ResourceViewMapping] | None = None
+
+    def _interactive_select(self) -> tuple[str, ...]:
+        mappings = self.client.migration.resource_view_mapping.list(limit=-1)
+        if not mappings:
+            raise ToolkitMissingResourceError("No resource view mappings found")
+        self.resource_view_mappings = list(mappings)
+        choices = [
+            Choice(
+                f"{mapping.external_id} ({mapping.resource_type} -> {mapping.view_id.external_id})",
+                value=mapping.external_id,
+            )
+            for mapping in sorted(mappings, key=lambda m: m.external_id)
+        ]
+        selected_ids: list[str] | None = questionary.checkbox(
+            "Which resource view mapping(s) would you like to dump?",
+            choices=choices,
+            validate=lambda choices: True if choices else "You must select at least one resource view mapping.",
+        ).unsafe_ask()
+        if not selected_ids:
+            raise ToolkitValueError(f"No resource view mappings selected for dumping.{_INTERACTIVE_SELECT_HELPER_TEXT}")
+        return tuple(selected_ids)
+
+    def __iter__(self) -> Iterator[tuple[list[Hashable], CogniteResourceList | None, ResourceCRUD, None | str]]:
+        self.identifier = self._selected()
+        loader = ResourceViewMappingCRUD.create_loader(self.client)
+        if self.resource_view_mappings:
+            selected_mappings = NodeList[ResourceViewMapping](
+                [m for m in self.resource_view_mappings if m.external_id in self.identifier]
+            )
+            yield [], selected_mappings, loader, None
+        else:
+            yield list(self.identifier), None, loader, None
+
+
 class DumpResourceCommand(ToolkitCommand):
     def dump_to_yamls(
         self,

cognite_toolkit/_repo_files/GitHub/.github/workflows/deploy.yaml

@@ -12,7 +12,7 @@ jobs:
     environment: dev
     name: Deploy
     container:
-      image: cognite/toolkit:0.7.58
+      image: cognite/toolkit:0.7.60
       env:
         CDF_CLUSTER: ${{ vars.CDF_CLUSTER }}
         CDF_PROJECT: ${{ vars.CDF_PROJECT }}

cognite_toolkit/_repo_files/GitHub/.github/workflows/dry-run.yaml

@@ -10,7 +10,7 @@ jobs:
     environment: dev
     name: Deploy Dry Run
     container:
-      image: cognite/toolkit:0.7.58
+      image: cognite/toolkit:0.7.60
       env:
         CDF_CLUSTER: ${{ vars.CDF_CLUSTER }}
         CDF_PROJECT: ${{ vars.CDF_PROJECT }}

cognite_toolkit/_resources/cdf.toml

@@ -4,7 +4,7 @@ default_env = "<DEFAULT_ENV_PLACEHOLDER>"
 [modules]
 # This is the version of the modules. It should not be changed manually.
 # It will be updated by the 'cdf modules upgrade' command.
-version = "0.7.58"
+version = "0.7.60"
 
 
 [plugins]

cognite_toolkit/_version.py

@@ -1 +1 @@
-__version__ = "0.7.58"
+__version__ = "0.7.60"

{cognite_toolkit-0.7.58.dist-info → cognite_toolkit-0.7.60.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognite_toolkit
-Version: 0.7.58
+Version: 0.7.60
 Summary: Official Cognite Data Fusion tool for project templates and configuration deployment
 Author: Cognite AS
 Author-email: Cognite AS <support@cognite.com>

{cognite_toolkit-0.7.58.dist-info → cognite_toolkit-0.7.60.dist-info}/RECORD

@@ -7,7 +7,7 @@ cognite_toolkit/_cdf_tk/apps/_core_app.py,sha256=ogHzZCTILj6bHQn0GmLZRKldFHlxL33
 cognite_toolkit/_cdf_tk/apps/_data_app.py,sha256=LeplXlxXtyIymRPgbatQrRFodU4VZBFxI0bqDutLSbg,806
 cognite_toolkit/_cdf_tk/apps/_dev_app.py,sha256=zae8O55y4Gp-I1DLqh_yCRbRPOa-OlSVqXgrx_w-UV0,3313
 cognite_toolkit/_cdf_tk/apps/_download_app.py,sha256=Iw-q4lIMagFQ7b8RVRubUQNAPbqFKP2NmPi30yI7qG0,40455
-cognite_toolkit/_cdf_tk/apps/_dump_app.py,sha256=TE8NhCsR-iUvp2Ij7K5RPqYh2VkNmYjIOMyHyxcmN2Y,37307
+cognite_toolkit/_cdf_tk/apps/_dump_app.py,sha256=8Z0D2snV48FC5eMBYJQ9fDnaNQCII3yrgrGolRtSOSc,39325
 cognite_toolkit/_cdf_tk/apps/_import_app.py,sha256=5n5AF40HJ0Q_3LENCknG0MxH4pMUS8OwsqvtCYj_t7w,2086
 cognite_toolkit/_cdf_tk/apps/_landing_app.py,sha256=qZqO0qN33-6TTF9O4fYuNF6zhJH-QyUWsuElWr8sH6w,1365
 cognite_toolkit/_cdf_tk/apps/_migrate_app.py,sha256=X_HZ0wmRlL9-iuEZ5lZszEl-BwkzdyCKuTNbQ_qnWz8,47296
@@ -111,7 +111,7 @@ cognite_toolkit/_cdf_tk/client/cdf_client/responses.py,sha256=UVhvU_8UKCLsjCFaSi
 cognite_toolkit/_cdf_tk/client/config.py,sha256=weMR43z-gqHMn-Jqvfmh_nJ0HbgEdyeCGtISuEf3OuY,4269
 cognite_toolkit/_cdf_tk/client/http_client/__init__.py,sha256=_B_eB9yZ97ctfWFNrrK722R-WkMbktZECeiqZlUUee8,674
 cognite_toolkit/_cdf_tk/client/http_client/_client.py,sha256=wPq3Sya_a-SbiMNoAGWFgdefeQN4a16xZ7ytS0F0TV0,15860
-cognite_toolkit/_cdf_tk/client/http_client/_data_classes.py,sha256=qBXasZtaNTi9IRN07poW5JCunyliuY-3sKuJlTMDvfY,5414
+cognite_toolkit/_cdf_tk/client/http_client/_data_classes.py,sha256=BX7_itzaKm9oXl-cQroTn6wZHb-b8EEFddsEYxfMuw4,5414
 cognite_toolkit/_cdf_tk/client/http_client/_exception.py,sha256=9dE5tm1qTviG93JppLql6ltlmAgm8TxWh9wWPdm0ITA,428
 cognite_toolkit/_cdf_tk/client/http_client/_item_classes.py,sha256=Jjbvt0HjR9kcDpmFXsVRtwJvQqvEq6UOJO9xutSibr8,4454
 cognite_toolkit/_cdf_tk/client/http_client/_tracker.py,sha256=pu6oA-XpOeaOLdoeD_mGfJXC3BFGWfh5oGcRDpb6maw,1407
@@ -253,7 +253,7 @@ cognite_toolkit/_cdf_tk/commands/build_v2/data_classes/_resource.py,sha256=4tNKk
 cognite_toolkit/_cdf_tk/commands/clean.py,sha256=bsrT-W__Vd4c9rbGC8p-KosdcdH7519ipIwad1dpuqo,16716
 cognite_toolkit/_cdf_tk/commands/collect.py,sha256=zBMKhhvjOpuASMnwP0eeHRI02tANcvFEZgv0CQO1ECc,627
 cognite_toolkit/_cdf_tk/commands/deploy.py,sha256=GhVHFXbuAdMM_ax0wU-c8ALlkmU2DyEHoqIP7xpE1XQ,23354
-cognite_toolkit/_cdf_tk/commands/dump_resource.py,sha256=E7fX0TGwoVkj6neRcoF3QnLB92JLocZc-WQPb8eOm_M,40891
+cognite_toolkit/_cdf_tk/commands/dump_resource.py,sha256=V6SRvzBur-P946PoZDMF2pxQ9wo2H-KvpfAHHDzIGrg,42926
 cognite_toolkit/_cdf_tk/commands/init.py,sha256=og4PDPkRmjJ1sLq6cZ1Ozk6m7XnQPcE4I4RBtyrA0Yc,9427
 cognite_toolkit/_cdf_tk/commands/modules.py,sha256=9s9pMHT6ySoO5TXBmG8Qe7z_iQka4ctLSE_wpf191MU,41447
 cognite_toolkit/_cdf_tk/commands/pull.py,sha256=p2Per2Wpb5FmZU1qlsU1V-4PHWWbbLhkyv1XiAdDUSA,44086
@@ -432,13 +432,13 @@ cognite_toolkit/_repo_files/.gitignore,sha256=ip9kf9tcC5OguF4YF4JFEApnKYw0nG0vPi
 cognite_toolkit/_repo_files/AzureDevOps/.devops/README.md,sha256=OLA0D7yCX2tACpzvkA0IfkgQ4_swSd-OlJ1tYcTBpsA,240
 cognite_toolkit/_repo_files/AzureDevOps/.devops/deploy-pipeline.yml,sha256=brULcs8joAeBC_w_aoWjDDUHs3JheLMIR9ajPUK96nc,693
 cognite_toolkit/_repo_files/AzureDevOps/.devops/dry-run-pipeline.yml,sha256=OBFDhFWK1mlT4Dc6mDUE2Es834l8sAlYG50-5RxRtHk,723
-cognite_toolkit/_repo_files/GitHub/.github/workflows/deploy.yaml,sha256=Vc6lbsJshjcbdnk0m0L0I1lgq9D1zFdnivVs9rU4Mu0,667
-cognite_toolkit/_repo_files/GitHub/.github/workflows/dry-run.yaml,sha256=llgGZbNXpV9ks1z-FDTYxye_poxmqUHX7PhhzuS5PeQ,2430
-cognite_toolkit/_resources/cdf.toml,sha256=f6Y3dP4CQCvse7TjH-J_Oya939-iwEEK1-JO3j_acts,475
-cognite_toolkit/_version.py,sha256=p1Yn12O0jCcfUb2apLoojUHCZKX1Ej8wVvNv4Xo_4_I,23
+cognite_toolkit/_repo_files/GitHub/.github/workflows/deploy.yaml,sha256=IaQ-Z7WcOSO1MBiqG4R_RnFKK3egcpV-etbbgslJDBw,667
+cognite_toolkit/_repo_files/GitHub/.github/workflows/dry-run.yaml,sha256=y3VQL0JlzWQMjudfZSX9EMcx4EscgEQXD2uCTSS9Hu0,2430
+cognite_toolkit/_resources/cdf.toml,sha256=w078At6q9-z6dYhPwFnBveWZsyJq2nO9Gf-AZwrV_i0,475
+cognite_toolkit/_version.py,sha256=OD9is93QVR83jKhdg4JiAv63SOR_32toBDAsTuloB6o,23
 cognite_toolkit/demo/__init__.py,sha256=-m1JoUiwRhNCL18eJ6t7fZOL7RPfowhCuqhYFtLgrss,72
 cognite_toolkit/demo/_base.py,sha256=6xKBUQpXZXGQ3fJ5f7nj7oT0s2n7OTAGIa17ZlKHZ5U,8052
-cognite_toolkit-0.7.58.dist-info/WHEEL,sha256=e_m4S054HL0hyR3CpOk-b7Q7fDX6BuFkgL5OjAExXas,80
-cognite_toolkit-0.7.58.dist-info/entry_points.txt,sha256=EtZ17K2mUjh-AY0QNU1CPIB_aDSSOdmtNI_4Fj967mA,84
-cognite_toolkit-0.7.58.dist-info/METADATA,sha256=--_9cyLbtX5DNroh8enBrrFVxqRfr1WJHuAElNw2VLU,5026
-cognite_toolkit-0.7.58.dist-info/RECORD,,
+cognite_toolkit-0.7.60.dist-info/WHEEL,sha256=e_m4S054HL0hyR3CpOk-b7Q7fDX6BuFkgL5OjAExXas,80
+cognite_toolkit-0.7.60.dist-info/entry_points.txt,sha256=EtZ17K2mUjh-AY0QNU1CPIB_aDSSOdmtNI_4Fj967mA,84
+cognite_toolkit-0.7.60.dist-info/METADATA,sha256=baWSItas1tNH2vLmN58Vdxaaag-UrGqEwLXpjda_kMY,5026
+cognite_toolkit-0.7.60.dist-info/RECORD,,