[AutoPR datafactory] Change PartitionOption to object

src/datafactory/HISTORY.rst

@@ -3,10 +3,6 @@
 Release History
 ===============
 
-0.2.0
-++++++
-* add update command for linked services and triggers and datasets
-
 0.1.0
 ++++++
 * Initial release.

src/datafactory/azext_datafactory/generated/_help.py

@@ -19,11 +19,14 @@
 
 helps['datafactory factory list'] = """
     type: command
-    short-summary: "Lists factories under the specified subscription."
+    short-summary: "Lists factories. And Lists factories under the specified subscription."
     examples:
       - name: Factories_ListByResourceGroup
         text: |-
                az datafactory factory list --resource-group "exampleResourceGroup"
+      - name: Factories_List
+        text: |-
+               az datafactory factory list
 """
 
 helps['datafactory factory show'] = """
@@ -37,7 +40,7 @@
 
 helps['datafactory factory create'] = """
     type: command
-    short-summary: "Creates or updates a factory."
+    short-summary: "Create a factory."
     parameters:
       - name: --factory-vsts-configuration
         short-summary: "Factory's VSTS repo information."
@@ -203,7 +206,7 @@
 
 helps['datafactory integration-runtime managed create'] = """
     type: command
-    short-summary: "Creates or updates an integration runtime."
+    short-summary: "Create an integration runtime."
 """
 
 helps['datafactory integration-runtime self-hosted'] = """
@@ -213,7 +216,7 @@
 
 helps['datafactory integration-runtime self-hosted create'] = """
     type: command
-    short-summary: "Creates or updates an integration runtime."
+    short-summary: "Create an integration runtime."
     examples:
       - name: IntegrationRuntimes_Create
         text: |-
@@ -432,7 +435,7 @@
 
 helps['datafactory linked-service create'] = """
     type: command
-    short-summary: "Creates or updates a linked service."
+    short-summary: "Create a linked service."
     examples:
       - name: LinkedServices_Create
         text: |-
@@ -444,7 +447,7 @@
 
 helps['datafactory linked-service update'] = """
     type: command
-    short-summary: "Creates or updates a linked service."
+    short-summary: "Update a linked service."
     examples:
       - name: LinkedServices_Update
         text: |-
@@ -488,7 +491,7 @@
 
 helps['datafactory dataset create'] = """
     type: command
-    short-summary: "Creates or updates a dataset."
+    short-summary: "Create a dataset."
     examples:
       - name: Datasets_Create
         text: |-
@@ -502,7 +505,7 @@
 
 helps['datafactory dataset update'] = """
     type: command
-    short-summary: "Creates or updates a dataset."
+    short-summary: "Update a dataset."
     parameters:
       - name: --folder
         short-summary: "The folder that this Dataset is in. If not specified, Dataset will appear at the root level."
@@ -556,7 +559,7 @@
 
 helps['datafactory pipeline create'] = """
     type: command
-    short-summary: "Creates or updates a pipeline."
+    short-summary: "Create a pipeline."
     examples:
       - name: Pipelines_Create
         text: |-
@@ -576,7 +579,7 @@
 
 helps['datafactory pipeline update'] = """
     type: command
-    short-summary: "Creates or updates a pipeline."
+    short-summary: "Update a pipeline."
     examples:
       - name: Pipelines_Update
         text: |-
@@ -739,7 +742,7 @@
 
 helps['datafactory trigger create'] = """
     type: command
-    short-summary: "Creates or updates a trigger."
+    short-summary: "Create a trigger."
     examples:
       - name: Triggers_Create
         text: |-
@@ -753,7 +756,7 @@
 
 helps['datafactory trigger update'] = """
     type: command
-    short-summary: "Creates or updates a trigger."
+    short-summary: "Update a trigger."
     examples:
       - name: Triggers_Update
         text: |-

src/datafactory/azext_datafactory/generated/_params.py

@@ -57,6 +57,8 @@ def load_arguments(self, _):
                    'GitHub repo information.', arg_group='RepoConfiguration')
         c.argument('global_parameters', type=validate_file_or_dict, help='List of parameters for factory. Expected '
                    'value: json-string/@json-file.')
+        c.argument('public_network_access', arg_type=get_enum_type(['Enabled', 'Disabled']), help='Whether or not '
+                   'public network access is allowed for the data factory.')
 
     with self.argument_context('datafactory factory update') as c:
         c.argument('resource_group_name', resource_group_name_type)

src/datafactory/azext_datafactory/generated/custom.py

@@ -39,7 +39,8 @@ def datafactory_factory_create(client,
                                tags=None,
                                factory_vsts_configuration=None,
                                factory_git_hub_configuration=None,
-                               global_parameters=None):
+                               global_parameters=None,
+                               public_network_access=None):
     all_repo_configuration = []
     if factory_vsts_configuration is not None:
         all_repo_configuration.append(factory_vsts_configuration)
@@ -56,7 +57,8 @@ def datafactory_factory_create(client,
                                    tags=tags,
                                    identity=None,
                                    repo_configuration=repo_configuration,
-                                   global_parameters=global_parameters)
+                                   global_parameters=global_parameters,
+                                   public_network_access=public_network_access)
 
 
 def datafactory_factory_update(client,
@@ -66,7 +68,7 @@ def datafactory_factory_update(client,
     return client.update(resource_group_name=resource_group_name,
                          factory_name=factory_name,
                          tags=tags,
-                         identity={"type": "SystemAssigned"})
+                         identity=json.loads("{\"type\": \"SystemAssigned\"}"))
 
 
 def datafactory_factory_delete(client,

src/datafactory/azext_datafactory/tests/latest/test_datafactory_scenario.py

@@ -17,6 +17,7 @@
 TEST_DIR = os.path.abspath(os.path.join(os.path.abspath(__file__), '..'))
 
 
+# Env setup
 @try_manual
 def setup(test, rg):
     pass
@@ -671,11 +672,13 @@ def step_factories_delete(test, rg):
              checks=[])
 
 
+# Env cleanup
 @try_manual
 def cleanup(test, rg):
     pass
 
 
+# Testcase
 @try_manual
 def call_scenario(test, rg):
     setup(test, rg)

src/datafactory/azext_datafactory/vendored_sdks/datafactory/_configuration.py

@@ -48,8 +48,7 @@ def __init__(
         self.credential = credential
         self.subscription_id = subscription_id
         self.api_version = "2018-06-01"
-        self.credential_scopes = ['https://management.azure.com/.default']
-        self.credential_scopes.extend(kwargs.pop('credential_scopes', []))
+        self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default'])
         kwargs.setdefault('sdk_moniker', 'datafactorymanagementclient/{}'.format(VERSION))
         self._configure(**kwargs)
 

src/datafactory/azext_datafactory/vendored_sdks/datafactory/aio/__init__.py

@@ -6,5 +6,5 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-from ._data_factory_management_client_async import DataFactoryManagementClient
+from ._data_factory_management_client import DataFactoryManagementClient
 __all__ = ['DataFactoryManagementClient']

src/datafactory/azext_datafactory/vendored_sdks/datafactory/aio/_configuration_async.py → src/datafactory/azext_datafactory/vendored_sdks/datafactory/aio/_configuration.py

@@ -45,8 +45,7 @@ def __init__(
         self.credential = credential
         self.subscription_id = subscription_id
         self.api_version = "2018-06-01"
-        self.credential_scopes = ['https://management.azure.com/.default']
-        self.credential_scopes.extend(kwargs.pop('credential_scopes', []))
+        self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default'])
         kwargs.setdefault('sdk_moniker', 'datafactorymanagementclient/{}'.format(VERSION))
         self._configure(**kwargs)
 

src/datafactory/azext_datafactory/vendored_sdks/datafactory/aio/_data_factory_management_client_async.py → src/datafactory/azext_datafactory/vendored_sdks/datafactory/aio/_data_factory_management_client.py

@@ -15,64 +15,64 @@
     # pylint: disable=unused-import,ungrouped-imports
     from azure.core.credentials_async import AsyncTokenCredential
 
-from ._configuration_async import DataFactoryManagementClientConfiguration
-from .operations_async import OperationOperations
-from .operations_async import FactoryOperations
-from .operations_async import ExposureControlOperations
-from .operations_async import IntegrationRuntimeOperations
-from .operations_async import IntegrationRuntimeObjectMetadataOperations
-from .operations_async import IntegrationRuntimeNodeOperations
-from .operations_async import LinkedServiceOperations
-from .operations_async import DatasetOperations
-from .operations_async import PipelineOperations
-from .operations_async import PipelineRunOperations
-from .operations_async import ActivityRunOperations
-from .operations_async import TriggerOperations
-from .operations_async import TriggerRunOperations
-from .operations_async import DataFlowOperations
-from .operations_async import DataFlowDebugSessionOperations
-from .operations_async import ManagedVirtualNetworkOperations
-from .operations_async import ManagedPrivateEndpointOperations
+from ._configuration import DataFactoryManagementClientConfiguration
+from .operations import OperationOperations
+from .operations import FactoryOperations
+from .operations import ExposureControlOperations
+from .operations import IntegrationRuntimeOperations
+from .operations import IntegrationRuntimeObjectMetadataOperations
+from .operations import IntegrationRuntimeNodeOperations
+from .operations import LinkedServiceOperations
+from .operations import DatasetOperations
+from .operations import PipelineOperations
+from .operations import PipelineRunOperations
+from .operations import ActivityRunOperations
+from .operations import TriggerOperations
+from .operations import TriggerRunOperations
+from .operations import DataFlowOperations
+from .operations import DataFlowDebugSessionOperations
+from .operations import ManagedVirtualNetworkOperations
+from .operations import ManagedPrivateEndpointOperations
 from .. import models
 
 
 class DataFactoryManagementClient(object):
     """The Azure Data Factory V2 management API provides a RESTful set of web services that interact with Azure Data Factory V2 services.
 
     :ivar operation: OperationOperations operations
-    :vartype operation: data_factory_management_client.aio.operations_async.OperationOperations
+    :vartype operation: data_factory_management_client.aio.operations.OperationOperations
     :ivar factory: FactoryOperations operations
-    :vartype factory: data_factory_management_client.aio.operations_async.FactoryOperations
+    :vartype factory: data_factory_management_client.aio.operations.FactoryOperations
     :ivar exposure_control: ExposureControlOperations operations
-    :vartype exposure_control: data_factory_management_client.aio.operations_async.ExposureControlOperations
+    :vartype exposure_control: data_factory_management_client.aio.operations.ExposureControlOperations
     :ivar integration_runtime: IntegrationRuntimeOperations operations
-    :vartype integration_runtime: data_factory_management_client.aio.operations_async.IntegrationRuntimeOperations
+    :vartype integration_runtime: data_factory_management_client.aio.operations.IntegrationRuntimeOperations
     :ivar integration_runtime_object_metadata: IntegrationRuntimeObjectMetadataOperations operations
-    :vartype integration_runtime_object_metadata: data_factory_management_client.aio.operations_async.IntegrationRuntimeObjectMetadataOperations
+    :vartype integration_runtime_object_metadata: data_factory_management_client.aio.operations.IntegrationRuntimeObjectMetadataOperations
     :ivar integration_runtime_node: IntegrationRuntimeNodeOperations operations
-    :vartype integration_runtime_node: data_factory_management_client.aio.operations_async.IntegrationRuntimeNodeOperations
+    :vartype integration_runtime_node: data_factory_management_client.aio.operations.IntegrationRuntimeNodeOperations
     :ivar linked_service: LinkedServiceOperations operations
-    :vartype linked_service: data_factory_management_client.aio.operations_async.LinkedServiceOperations
+    :vartype linked_service: data_factory_management_client.aio.operations.LinkedServiceOperations
     :ivar dataset: DatasetOperations operations
-    :vartype dataset: data_factory_management_client.aio.operations_async.DatasetOperations
+    :vartype dataset: data_factory_management_client.aio.operations.DatasetOperations
     :ivar pipeline: PipelineOperations operations
-    :vartype pipeline: data_factory_management_client.aio.operations_async.PipelineOperations
+    :vartype pipeline: data_factory_management_client.aio.operations.PipelineOperations
     :ivar pipeline_run: PipelineRunOperations operations
-    :vartype pipeline_run: data_factory_management_client.aio.operations_async.PipelineRunOperations
+    :vartype pipeline_run: data_factory_management_client.aio.operations.PipelineRunOperations
     :ivar activity_run: ActivityRunOperations operations
-    :vartype activity_run: data_factory_management_client.aio.operations_async.ActivityRunOperations
+    :vartype activity_run: data_factory_management_client.aio.operations.ActivityRunOperations
     :ivar trigger: TriggerOperations operations
-    :vartype trigger: data_factory_management_client.aio.operations_async.TriggerOperations
+    :vartype trigger: data_factory_management_client.aio.operations.TriggerOperations
     :ivar trigger_run: TriggerRunOperations operations
-    :vartype trigger_run: data_factory_management_client.aio.operations_async.TriggerRunOperations
+    :vartype trigger_run: data_factory_management_client.aio.operations.TriggerRunOperations
     :ivar data_flow: DataFlowOperations operations
-    :vartype data_flow: data_factory_management_client.aio.operations_async.DataFlowOperations
+    :vartype data_flow: data_factory_management_client.aio.operations.DataFlowOperations
     :ivar data_flow_debug_session: DataFlowDebugSessionOperations operations
-    :vartype data_flow_debug_session: data_factory_management_client.aio.operations_async.DataFlowDebugSessionOperations
+    :vartype data_flow_debug_session: data_factory_management_client.aio.operations.DataFlowDebugSessionOperations
     :ivar managed_virtual_network: ManagedVirtualNetworkOperations operations
-    :vartype managed_virtual_network: data_factory_management_client.aio.operations_async.ManagedVirtualNetworkOperations
+    :vartype managed_virtual_network: data_factory_management_client.aio.operations.ManagedVirtualNetworkOperations
     :ivar managed_private_endpoint: ManagedPrivateEndpointOperations operations
-    :vartype managed_private_endpoint: data_factory_management_client.aio.operations_async.ManagedPrivateEndpointOperations
+    :vartype managed_private_endpoint: data_factory_management_client.aio.operations.ManagedPrivateEndpointOperations
     :param credential: Credential needed for the client to connect to Azure.
     :type credential: ~azure.core.credentials_async.AsyncTokenCredential
     :param subscription_id: The subscription identifier.