Refactor code structure for improved readability and maintainability

Author: Suyash Choudhary

src/storage-mover/azext_storage_mover/_params.py

@@ -70,7 +70,7 @@ def load_arguments(self, _):  # pylint: disable=unused-argument
         c.argument('source_uri', help='The URI which points to the S3-compatible source.')
         c.argument('source_type', arg_type=get_enum_type(['ALIBABA', 'DELL_EMC', 'GCS', 'IBM', 'MINIO', 'OTHER']),
                    help='The source type of the S3WithHmac endpoint.')
-        c.argument('other_source_type_description',
+        c.argument('other_source_type_description', options_list=('--other-source-type-desc', '--other-source-type-description'),
                    help='The description for other source type of S3WithHmac endpoint.')
 
     for command in ['create-for-s3-with-hmac', 'update-for-s3-with-hmac']:

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/agent/_update.py

@@ -16,15 +16,6 @@
 )
 class Update(AAZCommand):
     """Update an Agent resource, which references a hybrid compute machine that can run jobs.
-
-    :example: agent update
-        az storage-mover agent update -g {rg} -n {agent_name} --storage-mover-name {mover_name} --description 123
-
-    :example: add upload-limit-schedule
-        az storage-mover agent update -g test-storagemover-rg2 -n agent2 --storage-mover-name teststoragemover2 --upload-limit-schedule "{weekly-recurrences:[{days:[Monday,Wednesday],start-time:{hour:10,minute:0},end-time:{hour:12,minute:30},limit-in-mbps:20}]}"
-
-    :example: add another weekly-recurrence to existing upload-limit-schedule list
-        az storage-mover agent update -g test-storagemover-rg2 -n agent2 --storage-mover-name teststoragemover2 --upload-limit-schedule weekly-recurrences[1]="{days:[Tuesday,Thursday],start-time:{hour:10,minute:0},end-time:{hour:12,minute:30},limit-in-mbps:20}"
     """
 
     _aaz_info = {

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/connection/_create.py

@@ -75,7 +75,7 @@ def _build_arguments_schema(cls, *args, **kwargs):
             help="List of job definitions associated with this connection.",
         )
         _args_schema.private_link_service_id = AAZResourceIdArg(
-            options=["--private-link-service-id"],
+            options=["--private-link-service-id", "--pls-id"],
             arg_group="Properties",
             help="The PrivateLinkServiceId for the connection.",
             required=True,

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/connection/_update.py

@@ -81,7 +81,7 @@ def _build_arguments_schema(cls, *args, **kwargs):
             nullable=True,
         )
         _args_schema.private_link_service_id = AAZResourceIdArg(
-            options=["--private-link-service-id"],
+            options=["--private-link-service-id", "--pls-id"],
             arg_group="Properties",
             help="The PrivateLinkServiceId for the connection.",
         )

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/endpoint/_create.py

@@ -194,7 +194,7 @@ def _build_arguments_schema(cls, *args, **kwargs):
         s3_with_hmac.source_type = AAZStrArg(
             options=["source-type"],
             help="The source type of S3WithHmac endpoint.",
-            enum={"BACKBLAZE": "BACKBLAZE", "CLOUDFLARE": "CLOUDFLARE", "GCS": "GCS", "IBM": "IBM", "MINIO": "MINIO"},
+            enum={"ALIBABA": "ALIBABA", "DELL_EMC": "DELL_EMC", "GCS": "GCS", "IBM": "IBM", "MINIO": "MINIO", "OTHER": "OTHER"},
         )
         s3_with_hmac.source_uri = AAZStrArg(
             options=["source-uri"],
@@ -466,6 +466,7 @@ def _build_schema_on_200(cls):
             )
             identity.user_assigned_identities = AAZDictType(
                 serialized_name="userAssignedIdentities",
+                nullable=True,
             )
 
             user_assigned_identities = cls._schema_on_200.identity.user_assigned_identities

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/endpoint/_list.py

@@ -197,6 +197,7 @@ def _build_schema_on_200(cls):
             )
             identity.user_assigned_identities = AAZDictType(
                 serialized_name="userAssignedIdentities",
+                nullable=True,
             )
 
             user_assigned_identities = cls._schema_on_200.value.Element.identity.user_assigned_identities

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/endpoint/_show.py

@@ -195,6 +195,7 @@ def _build_schema_on_200(cls):
             )
             identity.user_assigned_identities = AAZDictType(
                 serialized_name="userAssignedIdentities",
+                nullable=True,
             )
 
             user_assigned_identities = cls._schema_on_200.identity.user_assigned_identities

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/endpoint/_update.py

@@ -417,6 +417,7 @@ def _build_schema_endpoint_read(cls, _schema):
         )
         identity.user_assigned_identities = AAZDictType(
             serialized_name="userAssignedIdentities",
+            nullable=True,
         )
 
         user_assigned_identities = _schema_endpoint_read.identity.user_assigned_identities

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/endpoint/_wait.py

@@ -194,6 +194,7 @@ def _build_schema_on_200(cls):
             )
             identity.user_assigned_identities = AAZDictType(
                 serialized_name="userAssignedIdentities",
+                nullable=True,
             )
 
             user_assigned_identities = cls._schema_on_200.identity.user_assigned_identities

src/storage-mover/azext_storage_mover/aaz/latest/storage_mover/endpoint/identity/_assign.py

@@ -377,6 +377,7 @@ def _build_schema_endpoint_read(cls, _schema):
         )
         identity.user_assigned_identities = AAZDictType(
             serialized_name="userAssignedIdentities",
+            nullable=True,
         )
 
         user_assigned_identities = _schema_endpoint_read.identity.user_assigned_identities