azureml-registry-tools 0.1.0a4__py3-none-any.whl → 0.1.0a6__py3-none-any.whl

azureml/registry/_cli/registry_syndication_cli.py

@@ -3,10 +3,12 @@
 # ---------------------------------------------------------
 import argparse
 import json
+import sys
+from uuid import UUID
 from azureml.registry._rest_client.registry_management_client import RegistryManagementClient
 from azureml.registry._rest_client.arm_client import ArmClient
 from azureml.registry.mgmt.create_manifest import generate_syndication_manifest
-from azureml.registry.mgmt.syndication_manifest import SyndicationManifest
+from azureml.registry.mgmt.syndication_manifest import SyndicationManifest, ResyncAssetsInManifestDto
 
 
 def syndication_manifest_show(registry_name: str) -> dict:
@@ -56,6 +58,31 @@ def syndication_manifest_delete(registry_name: str, dry_run: bool) -> None:
         RegistryManagementClient(registry_name=registry_name).delete_manifest()
 
 
+def syndication_manifest_sync(registry_name: str, manifest_value: str, tenant_id: str, asset_ids: list, dry_run: bool) -> None:
+    """Sync assets in the syndication manifest for the specified registry.
+
+    Args:
+        registry_name (str): Name of the AzureML registry.
+        manifest_value (str): Manifest value as a string (JSON).
+        tenant_id (str): Tenant ID to use for all assets.
+        asset_ids (list): List of asset IDs to sync.
+        dry_run (bool): If True, do not perform any changes.
+    """
+    if manifest_value:
+        # for inline manifest value allow different casing of keys
+        sync_dto = ResyncAssetsInManifestDto.from_dict(json.loads(manifest_value), normalize_keys=True)
+    else:
+        # Create from tenant ID and asset IDs
+        sync_dto = ResyncAssetsInManifestDto.from_asset_list(UUID(tenant_id), asset_ids)
+
+    dto = sync_dto.to_dict()
+    if dry_run:
+        print(f"Dry run: Would sync assets for registry {registry_name} with data: {json.dumps(dto, indent=2)}")
+    else:
+        client = RegistryManagementClient(registry_name=registry_name)
+        client.sync_assets_in_manifest(dto)
+
+
 def syndication_target_show(registry_name: str) -> object:
     """Show the current syndication target(s) for the specified registry and region.
 
@@ -129,6 +156,12 @@ def main() -> None:
       # Delete the current manifest
       registry-mgmt syndication manifest delete --registry-name myreg
 
+      # Sync assets from a JSON value
+      registry-mgmt syndication manifest sync --registry-name myreg --value '{"AssetsToResync": [...]}'
+
+      # Sync assets from tenant ID and asset list
+      registry-mgmt syndication manifest sync --registry-name myreg --tenant-id "12345678-1234-1234-1234-123456789012" --asset-ids asset1 asset2 asset3
+
       # Show the current target
       registry-mgmt syndication target show --registry-name myreg
 
@@ -184,6 +217,13 @@ def main() -> None:
     manifest_delete_parser = manifest_subparsers.add_parser("delete", help="Delete the current manifest.")
     _add_common_args(manifest_delete_parser, [registry_name_arg, dry_run_arg])
 
+    manifest_sync_parser = manifest_subparsers.add_parser("sync", help="Sync assets in the manifest.")
+    _add_common_args(manifest_sync_parser, [registry_name_arg, dry_run_arg])
+    sync_group = manifest_sync_parser.add_mutually_exclusive_group(required=True)
+    sync_group.add_argument("-v", "--value", type=str, help="Manifest sync value as JSON string.")
+    sync_group.add_argument("-t", "--tenant-id", type=str, help="Tenant ID to use for all assets.")
+    manifest_sync_parser.add_argument("-a", "--asset-ids", type=str, nargs="+", help="List of asset IDs to sync (only valid with --tenant-id).")
+
     # syndication target
     target_parser = synd_subparsers.add_parser("target", help="Manage syndication target.")
     target_subparsers = target_parser.add_subparsers(dest="target_subcommand", required=True)
@@ -215,6 +255,18 @@ def main() -> None:
                     syndication_manifest_delete(args.registry_name, args.dry_run)
                 else:
                     print("Manifest deletion cancelled.")
+            elif args.manifest_subcommand == "sync":
+                # Validate constraints for sync command
+                if args.tenant_id and not args.asset_ids:
+                    print("Error: --asset-ids is required when using --tenant-id", file=sys.stderr)
+                    sys.exit(1)
+                elif args.asset_ids and not args.tenant_id:
+                    print("Error: --asset-ids can only be used with --tenant-id", file=sys.stderr)
+                    sys.exit(1)
+                elif not args.value and not args.tenant_id:
+                    print("Error: Either --value or --tenant-id must be specified", file=sys.stderr)
+                    sys.exit(1)
+                syndication_manifest_sync(args.registry_name, args.value, args.tenant_id, args.asset_ids, args.dry_run)
         elif args.synd_subcommand == "target":
             if args.target_subcommand == "show":
                 print(syndication_target_show(args.registry_name))

azureml/registry/_rest_client/registry_management_client.py

@@ -57,7 +57,7 @@ class RegistryManagementClient(BaseAzureRestClient):
         response = self.post(url, json=manifest_dto)
         return response.json()
 
-    def resync_assets_in_manifest(self, resync_assets_dto: dict) -> dict:
+    def sync_assets_in_manifest(self, resync_assets_dto: dict) -> dict:
         """
         Resynchronize assets in the syndication manifest.
 

azureml/registry/mgmt/syndication_manifest.py

@@ -148,3 +148,79 @@ class SyndicationManifest:
             tenant_id=tenant_id,
             source_registries=source_registries
         )
+
+
+@dataclass
+class AssetToResync:
+    """Represents an asset to resync with its associated tenant ID."""
+
+    asset_id: str
+    tenant_id: UUID
+
+    def to_dict(self) -> dict:
+        """Convert the AssetToResync instance to a dictionary."""
+        return {
+            "AssetIds": self.asset_id,
+            "TenantId": str(self.tenant_id)
+        }
+
+    @staticmethod
+    def from_dict(d: dict, normalize_keys=False) -> "AssetToResync":
+        """Create an AssetToResync instance from a dictionary."""
+        asset_id = _get_key(d, "AssetIds", "asset_ids", normalize_keys=normalize_keys)
+        if asset_id is None:
+            raise ValueError("Missing required field 'AssetIds' in dictionary.")
+
+        tenant_id_str = _get_key(d, "TenantId", "tenant_id", normalize_keys=normalize_keys)
+        if tenant_id_str is None:
+            raise ValueError("Missing required field 'TenantId' in dictionary.")
+
+        try:
+            tenant_id = UUID(str(tenant_id_str))
+        except ValueError:
+            raise ValueError("TenantId must be a valid UUID string.")
+
+        return AssetToResync(
+            asset_id=str(asset_id),
+            tenant_id=tenant_id
+        )
+
+
+@dataclass
+class ResyncAssetsInManifestDto:
+    """Represents a collection of assets to resync in a manifest."""
+
+    assets_to_resync: List[AssetToResync]
+
+    def to_dict(self) -> dict:
+        """Convert the ResyncAssetsInManifestDto instance to a dictionary."""
+        return {
+            "AssetsToResync": [asset.to_dict() for asset in self.assets_to_resync]
+        }
+
+    @staticmethod
+    def from_dict(d: dict, normalize_keys=False) -> "ResyncAssetsInManifestDto":
+        """Create a ResyncAssetsInManifestDto instance from a dictionary."""
+        assets_list = _get_key(d, "AssetsToResync", "assets_to_resync", normalize_keys=normalize_keys)
+        if not isinstance(assets_list, list):
+            raise ValueError("AssetsToResync must be a list.")
+        return ResyncAssetsInManifestDto(
+            assets_to_resync=[AssetToResync.from_dict(asset, normalize_keys=normalize_keys) for asset in assets_list]
+        )
+
+    @staticmethod
+    def from_asset_list(tenant_id: UUID, asset_ids: List[str]) -> "ResyncAssetsInManifestDto":
+        """Create a ResyncAssetsInManifestDto instance from a tenant ID and list of asset ID strings.
+
+        Args:
+            tenant_id (UUID): The tenant ID to use for all assets.
+            asset_ids (List[str]): List of asset ID strings.
+
+        Returns:
+            ResyncAssetsInManifestDto: Instance with assets created from the provided IDs and tenant ID.
+        """
+        assets_to_resync = [
+            AssetToResync(asset_id=asset_id, tenant_id=tenant_id)
+            for asset_id in asset_ids
+        ]
+        return ResyncAssetsInManifestDto(assets_to_resync=assets_to_resync)

{azureml_registry_tools-0.1.0a4.dist-info → azureml_registry_tools-0.1.0a6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a4
+Version: 0.1.0a6
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license

{azureml_registry_tools-0.1.0a4.dist-info → azureml_registry_tools-0.1.0a6.dist-info}/RECORD

@@ -1,23 +1,23 @@
 azureml/__init__.py,sha256=_gYz_vABVrp9EyOZEYn4Ya8XotJC2rZn-b9kFTEj1Dk,266
 azureml/registry/__init__.py,sha256=qjP86n1Yz6hdZb3az-wjfi7LNndsFjMQI6lXmKfXapM,266
 azureml/registry/_cli/__init__.py,sha256=dJ020ynrregpUaNGu8xVLLmX3cC9DAjaBMP1qnljLEE,208
-azureml/registry/_cli/registry_syndication_cli.py,sha256=87hxSu4iwFOdwfukSGqc0fjJEpSpQhDRlthiCGCDwQ8,10003
+azureml/registry/_cli/registry_syndication_cli.py,sha256=3PpXkq0EF6-adCMnz6HDALeugqfS9l86oIvh4isZyvw,12962
 azureml/registry/_cli/repo2registry_cli.py,sha256=4CDv4tYWLTjVgdXIcoKA9iu_gOv_Z_xn05wSANkdmdg,5378
 azureml/registry/_rest_client/__init__.py,sha256=Eh0vB8AVlwkZlHLO4DCGHyyfRYYgeTeVjiUAX_WujG0,219
 azureml/registry/_rest_client/arm_client.py,sha256=KVSj0V1bN-vCUV3fBbIdZDpWTLKYT6qdWIZtDjx2la4,4516
 azureml/registry/_rest_client/base_rest_client.py,sha256=JcmhckfR-ZdUtEh0EeQIKfe63Zjowa1fLzmVeWpeLno,8173
-azureml/registry/_rest_client/registry_management_client.py,sha256=P2aijKHdHvsSZ507-06Rq_n9pNJ_xci1oQKDGBJjm-I,4986
+azureml/registry/_rest_client/registry_management_client.py,sha256=3-PBwj91lkG1yrlB-s7jaYj0w5j177oSHZHvcoO8icQ,4984
 azureml/registry/mgmt/__init__.py,sha256=LMhqcEC8ItmmpKZljElGXH-6olHlT3SLl0dJU01OvuM,226
 azureml/registry/mgmt/create_manifest.py,sha256=N9wRmjAKO09A3utN_lCUsM_Ufpj7PL0SJz-XHPHWuyM,9528
-azureml/registry/mgmt/syndication_manifest.py,sha256=ma1_vUEHHtkRcgiFMQanlXZajvKodynylxfblTHuleE,6195
+azureml/registry/mgmt/syndication_manifest.py,sha256=8Sfd49QuCA5en5_mIOLE21kZVpnReUXowx_g0TVRgWg,9025
 azureml/registry/tools/__init__.py,sha256=IAuWWpGfZm__pAkBIxmpJz84QskpkxBr0yDk1TUSnkE,223
 azureml/registry/tools/config.py,sha256=tjPaoBsWtPXBL8Ww1hcJtsr2SuIjPKt79dR8iovcebg,3639
 azureml/registry/tools/create_or_update_assets.py,sha256=Q-_BV7KWn1huQn5JriKT_8xJNoQQ_HK5wCftrq9DepA,15988
 azureml/registry/tools/registry_utils.py,sha256=zgYlCiOONtQJ4yZ9wg8tKVoE8dh6rrjB8hYBGhpV9-0,1403
 azureml/registry/tools/repo2registry_config.py,sha256=eXp_tU8Jyi30g8xGf7wbpLgKEPpieohBANKxMSLzq7s,4873
-azureml_registry_tools-0.1.0a4.dist-info/licenses/LICENSE.txt,sha256=n20rxwp7_NGrrShv9Qvcs90sjI1l3Pkt3m-5OPCWzgs,845
-azureml_registry_tools-0.1.0a4.dist-info/METADATA,sha256=XCJDpX9oa2LTU8pUOVQOGjizcLtrlRYa1f0fECIa9tM,521
-azureml_registry_tools-0.1.0a4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-azureml_registry_tools-0.1.0a4.dist-info/entry_points.txt,sha256=iRUkAeQidMnO6RQzpLqMUBTcyYtNzAfSin9WnSdVGLw,147
-azureml_registry_tools-0.1.0a4.dist-info/top_level.txt,sha256=ZOeEa0TAXo6i5wOjwBoqfIGEuxOcKuscGgNSpizqREY,8
-azureml_registry_tools-0.1.0a4.dist-info/RECORD,,
+azureml_registry_tools-0.1.0a6.dist-info/licenses/LICENSE.txt,sha256=n20rxwp7_NGrrShv9Qvcs90sjI1l3Pkt3m-5OPCWzgs,845
+azureml_registry_tools-0.1.0a6.dist-info/METADATA,sha256=cHaW5YddH_-oZuk3WdrCjAH5xTaFKcKZZDWuBpj-B0A,521
+azureml_registry_tools-0.1.0a6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+azureml_registry_tools-0.1.0a6.dist-info/entry_points.txt,sha256=iRUkAeQidMnO6RQzpLqMUBTcyYtNzAfSin9WnSdVGLw,147
+azureml_registry_tools-0.1.0a6.dist-info/top_level.txt,sha256=ZOeEa0TAXo6i5wOjwBoqfIGEuxOcKuscGgNSpizqREY,8
+azureml_registry_tools-0.1.0a6.dist-info/RECORD,,