cognite-toolkit 0.7.59__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/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.59
+      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.59
+      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.59"
+version = "0.7.60"
 
 
 [plugins]

cognite_toolkit/_version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognite_toolkit
-Version: 0.7.59
+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.59.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
@@ -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=ex-Bd4HDkooDO2fjZJCwuk2vLf_1KYby4Ibsv_gU7IM,667
-cognite_toolkit/_repo_files/GitHub/.github/workflows/dry-run.yaml,sha256=E_8yqti8A2eG-RoQfMH8737-Jc8m6K5xjYASiQmdDCA,2430
-cognite_toolkit/_resources/cdf.toml,sha256=wrdnH7iLwAHLpu2pLLdSRfEv-vhfx9P3h09tP79aJkg,475
-cognite_toolkit/_version.py,sha256=t6AOVchfJc437zHhyoE43W1_Pn_n39nbWFfgh6rtifE,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.59.dist-info/WHEEL,sha256=e_m4S054HL0hyR3CpOk-b7Q7fDX6BuFkgL5OjAExXas,80
-cognite_toolkit-0.7.59.dist-info/entry_points.txt,sha256=EtZ17K2mUjh-AY0QNU1CPIB_aDSSOdmtNI_4Fj967mA,84
-cognite_toolkit-0.7.59.dist-info/METADATA,sha256=YgfkGwmfX0hsLHvf71-JrFcR9ZtFazes2G_f5OH1rL0,5026
-cognite_toolkit-0.7.59.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,,