CodeGen from PR 15786 in Azure/azure-rest-api-specs

sql always encrypted add uami support (Azure#15786)

sdk/datafactory/azure-mgmt-datafactory/_meta.json

@@ -4,7 +4,7 @@
     "@autorest/[email protected]",
     "@autorest/[email protected]"
   ],
-  "commit": "406474c3807f2dec010af72286f22aa7a9a54920",
+  "commit": "0676dbbe863c8f83f4e5985091f8422f171d8d89",
   "repository_url": "https://github.com/Azure/azure-rest-api-specs",
   "autorest_command": "autorest specification/datafactory/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --track2 --use=@autorest/[email protected] --use=@autorest/[email protected] --version=3.4.5",
   "readme": "specification/datafactory/resource-manager/readme.md"

sdk/datafactory/azure-mgmt-datafactory/azure/mgmt/datafactory/_version.py

@@ -6,4 +6,4 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "2.0.0"
+VERSION = "1.0.0"

sdk/datafactory/azure-mgmt-datafactory/azure/mgmt/datafactory/aio/operations/_integration_runtimes_operations.py

@@ -455,6 +455,68 @@ async def get_status(
         return deserialized
     get_status.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataFactory/factories/{factoryName}/integrationRuntimes/{integrationRuntimeName}/getStatus'}  # type: ignore
 
+    async def list_outbound_network_dependencies_endpoints(
+        self,
+        resource_group_name: str,
+        factory_name: str,
+        integration_runtime_name: str,
+        **kwargs: Any
+    ) -> "_models.IntegrationRuntimeOutboundNetworkDependenciesEndpointsResponse":
+        """Gets the list of outbound network dependencies for a given Azure-SSIS integration runtime.
+
+        :param resource_group_name: The resource group name.
+        :type resource_group_name: str
+        :param factory_name: The factory name.
+        :type factory_name: str
+        :param integration_runtime_name: The integration runtime name.
+        :type integration_runtime_name: str
+        :keyword callable cls: A custom type or function that will be passed the direct response
+        :return: IntegrationRuntimeOutboundNetworkDependenciesEndpointsResponse, or the result of cls(response)
+        :rtype: ~azure.mgmt.datafactory.models.IntegrationRuntimeOutboundNetworkDependenciesEndpointsResponse
+        :raises: ~azure.core.exceptions.HttpResponseError
+        """
+        cls = kwargs.pop('cls', None)  # type: ClsType["_models.IntegrationRuntimeOutboundNetworkDependenciesEndpointsResponse"]
+        error_map = {
+            401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
+        }
+        error_map.update(kwargs.pop('error_map', {}))
+        api_version = "2018-06-01"
+        accept = "application/json"
+
+        # Construct URL
+        url = self.list_outbound_network_dependencies_endpoints.metadata['url']  # type: ignore
+        path_format_arguments = {
+            'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
+            'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'),
+            'factoryName': self._serialize.url("factory_name", factory_name, 'str', max_length=63, min_length=3, pattern=r'^[A-Za-z0-9]+(?:-[A-Za-z0-9]+)*$'),
+            'integrationRuntimeName': self._serialize.url("integration_runtime_name", integration_runtime_name, 'str', max_length=63, min_length=3, pattern=r'^[A-Za-z0-9]+(?:-[A-Za-z0-9]+)*$'),
+        }
+        url = self._client.format_url(url, **path_format_arguments)
+
+        # Construct parameters
+        query_parameters = {}  # type: Dict[str, Any]
+        query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str')
+
+        # Construct headers
+        header_parameters = {}  # type: Dict[str, Any]
+        header_parameters['Accept'] = self._serialize.header("accept", accept, 'str')
+
+        request = self._client.get(url, query_parameters, header_parameters)
+        pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs)
+        response = pipeline_response.http_response
+
+        if response.status_code not in [200]:
+            map_error(status_code=response.status_code, response=response, error_map=error_map)
+            raise HttpResponseError(response=response, error_format=ARMErrorFormat)
+
+        deserialized = self._deserialize('IntegrationRuntimeOutboundNetworkDependenciesEndpointsResponse', pipeline_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})
+
+        return deserialized
+    list_outbound_network_dependencies_endpoints.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataFactory/factories/{factoryName}/integrationRuntimes/{integrationRuntimeName}/outboundNetworkDependenciesEndpoints'}  # type: ignore
+
     async def get_connection_info(
         self,
         resource_group_name: str,

sdk/datafactory/azure-mgmt-datafactory/azure/mgmt/datafactory/models/__init__.py

@@ -18,6 +18,13 @@
     from ._models_py3 import AmazonMWSLinkedService
     from ._models_py3 import AmazonMWSObjectDataset
     from ._models_py3 import AmazonMWSSource
+    from ._models_py3 import AmazonRdsForOracleLinkedService
+    from ._models_py3 import AmazonRdsForOraclePartitionSettings
+    from ._models_py3 import AmazonRdsForOracleSource
+    from ._models_py3 import AmazonRdsForOracleTableDataset
+    from ._models_py3 import AmazonRdsForSqlServerLinkedService
+    from ._models_py3 import AmazonRdsForSqlServerSource
+    from ._models_py3 import AmazonRdsForSqlServerTableDataset
     from ._models_py3 import AmazonRedshiftLinkedService
     from ._models_py3 import AmazonRedshiftSource
     from ._models_py3 import AmazonRedshiftTableDataset
@@ -243,9 +250,12 @@
     from ._models_py3 import ExcelDataset
     from ._models_py3 import ExcelSource
     from ._models_py3 import ExecuteDataFlowActivity
+    from ._models_py3 import ExecuteDataFlowActivityTypeProperties
     from ._models_py3 import ExecuteDataFlowActivityTypePropertiesCompute
     from ._models_py3 import ExecutePipelineActivity
+    from ._models_py3 import ExecutePowerQueryActivityTypeProperties
     from ._models_py3 import ExecuteSSISPackageActivity
+    from ._models_py3 import ExecuteWranglingDataflowActivity
     from ._models_py3 import ExecutionActivity
     from ._models_py3 import ExportSettings
     from ._models_py3 import ExposureControlBatchRequest
@@ -333,13 +343,18 @@
     from ._models_py3 import IntegrationRuntimeComputeProperties
     from ._models_py3 import IntegrationRuntimeConnectionInfo
     from ._models_py3 import IntegrationRuntimeCustomSetupScriptProperties
+    from ._models_py3 import IntegrationRuntimeCustomerVirtualNetwork
     from ._models_py3 import IntegrationRuntimeDataFlowProperties
     from ._models_py3 import IntegrationRuntimeDataProxyProperties
     from ._models_py3 import IntegrationRuntimeDebugResource
     from ._models_py3 import IntegrationRuntimeListResponse
     from ._models_py3 import IntegrationRuntimeMonitoringData
     from ._models_py3 import IntegrationRuntimeNodeIpAddress
     from ._models_py3 import IntegrationRuntimeNodeMonitoringData
+    from ._models_py3 import IntegrationRuntimeOutboundNetworkDependenciesCategoryEndpoint
+    from ._models_py3 import IntegrationRuntimeOutboundNetworkDependenciesEndpoint
+    from ._models_py3 import IntegrationRuntimeOutboundNetworkDependenciesEndpointDetails
+    from ._models_py3 import IntegrationRuntimeOutboundNetworkDependenciesEndpointsResponse
     from ._models_py3 import IntegrationRuntimeReference
     from ._models_py3 import IntegrationRuntimeRegenerateKeyParameters
     from ._models_py3 import IntegrationRuntimeResource
@@ -480,6 +495,8 @@
     from ._models_py3 import PostgreSqlLinkedService
     from ._models_py3 import PostgreSqlSource
     from ._models_py3 import PostgreSqlTableDataset
+    from ._models_py3 import PowerQuerySink
+    from ._models_py3 import PowerQuerySource
     from ._models_py3 import PrestoLinkedService
     from ._models_py3 import PrestoObjectDataset
     from ._models_py3 import PrestoSource
@@ -591,6 +608,7 @@
     from ._models_py3 import SqlAlwaysEncryptedProperties
     from ._models_py3 import SqlDWSink
     from ._models_py3 import SqlDWSource
+    from ._models_py3 import SqlDWUpsertSettings
     from ._models_py3 import SqlMISink
     from ._models_py3 import SqlMISource
     from ._models_py3 import SqlPartitionSettings
@@ -601,6 +619,7 @@
     from ._models_py3 import SqlServerTableDataset
     from ._models_py3 import SqlSink
     from ._models_py3 import SqlSource
+    from ._models_py3 import SqlUpsertSettings
     from ._models_py3 import SquareLinkedService
     from ._models_py3 import SquareObjectDataset
     from ._models_py3 import SquareSource
@@ -670,6 +689,7 @@
     from ._models_py3 import WebLinkedServiceTypeProperties
     from ._models_py3 import WebSource
     from ._models_py3 import WebTableDataset
+    from ._models_py3 import WranglingDataFlow
     from ._models_py3 import XeroLinkedService
     from ._models_py3 import XeroObjectDataset
     from ._models_py3 import XeroSource
@@ -692,6 +712,13 @@
     from ._models import AmazonMWSLinkedService  # type: ignore
     from ._models import AmazonMWSObjectDataset  # type: ignore
     from ._models import AmazonMWSSource  # type: ignore
+    from ._models import AmazonRdsForOracleLinkedService  # type: ignore
+    from ._models import AmazonRdsForOraclePartitionSettings  # type: ignore
+    from ._models import AmazonRdsForOracleSource  # type: ignore
+    from ._models import AmazonRdsForOracleTableDataset  # type: ignore
+    from ._models import AmazonRdsForSqlServerLinkedService  # type: ignore
+    from ._models import AmazonRdsForSqlServerSource  # type: ignore
+    from ._models import AmazonRdsForSqlServerTableDataset  # type: ignore
     from ._models import AmazonRedshiftLinkedService  # type: ignore
     from ._models import AmazonRedshiftSource  # type: ignore
     from ._models import AmazonRedshiftTableDataset  # type: ignore
@@ -917,9 +944,12 @@
     from ._models import ExcelDataset  # type: ignore
     from ._models import ExcelSource  # type: ignore
     from ._models import ExecuteDataFlowActivity  # type: ignore
+    from ._models import ExecuteDataFlowActivityTypeProperties  # type: ignore
     from ._models import ExecuteDataFlowActivityTypePropertiesCompute  # type: ignore
     from ._models import ExecutePipelineActivity  # type: ignore
+    from ._models import ExecutePowerQueryActivityTypeProperties  # type: ignore
     from ._models import ExecuteSSISPackageActivity  # type: ignore
+    from ._models import ExecuteWranglingDataflowActivity  # type: ignore
     from ._models import ExecutionActivity  # type: ignore
     from ._models import ExportSettings  # type: ignore
     from ._models import ExposureControlBatchRequest  # type: ignore
@@ -1007,13 +1037,18 @@
     from ._models import IntegrationRuntimeComputeProperties  # type: ignore
     from ._models import IntegrationRuntimeConnectionInfo  # type: ignore
     from ._models import IntegrationRuntimeCustomSetupScriptProperties  # type: ignore
+    from ._models import IntegrationRuntimeCustomerVirtualNetwork  # type: ignore
     from ._models import IntegrationRuntimeDataFlowProperties  # type: ignore
     from ._models import IntegrationRuntimeDataProxyProperties  # type: ignore
     from ._models import IntegrationRuntimeDebugResource  # type: ignore
     from ._models import IntegrationRuntimeListResponse  # type: ignore
     from ._models import IntegrationRuntimeMonitoringData  # type: ignore
     from ._models import IntegrationRuntimeNodeIpAddress  # type: ignore
     from ._models import IntegrationRuntimeNodeMonitoringData  # type: ignore
+    from ._models import IntegrationRuntimeOutboundNetworkDependenciesCategoryEndpoint  # type: ignore
+    from ._models import IntegrationRuntimeOutboundNetworkDependenciesEndpoint  # type: ignore
+    from ._models import IntegrationRuntimeOutboundNetworkDependenciesEndpointDetails  # type: ignore
+    from ._models import IntegrationRuntimeOutboundNetworkDependenciesEndpointsResponse  # type: ignore
     from ._models import IntegrationRuntimeReference  # type: ignore
     from ._models import IntegrationRuntimeRegenerateKeyParameters  # type: ignore
     from ._models import IntegrationRuntimeResource  # type: ignore
@@ -1154,6 +1189,8 @@
     from ._models import PostgreSqlLinkedService  # type: ignore
     from ._models import PostgreSqlSource  # type: ignore
     from ._models import PostgreSqlTableDataset  # type: ignore
+    from ._models import PowerQuerySink  # type: ignore
+    from ._models import PowerQuerySource  # type: ignore
     from ._models import PrestoLinkedService  # type: ignore
     from ._models import PrestoObjectDataset  # type: ignore
     from ._models import PrestoSource  # type: ignore
@@ -1265,6 +1302,7 @@
     from ._models import SqlAlwaysEncryptedProperties  # type: ignore
     from ._models import SqlDWSink  # type: ignore
     from ._models import SqlDWSource  # type: ignore
+    from ._models import SqlDWUpsertSettings  # type: ignore
     from ._models import SqlMISink  # type: ignore
     from ._models import SqlMISource  # type: ignore
     from ._models import SqlPartitionSettings  # type: ignore
@@ -1275,6 +1313,7 @@
     from ._models import SqlServerTableDataset  # type: ignore
     from ._models import SqlSink  # type: ignore
     from ._models import SqlSource  # type: ignore
+    from ._models import SqlUpsertSettings  # type: ignore
     from ._models import SquareLinkedService  # type: ignore
     from ._models import SquareObjectDataset  # type: ignore
     from ._models import SquareSource  # type: ignore
@@ -1344,6 +1383,7 @@
     from ._models import WebLinkedServiceTypeProperties  # type: ignore
     from ._models import WebSource  # type: ignore
     from ._models import WebTableDataset  # type: ignore
+    from ._models import WranglingDataFlow  # type: ignore
     from ._models import XeroLinkedService  # type: ignore
     from ._models import XeroObjectDataset  # type: ignore
     from ._models import XeroSource  # type: ignore
@@ -1356,6 +1396,7 @@
     from ._models import ZohoSource  # type: ignore
 
 from ._data_factory_management_client_enums import (
+    AmazonRdsForOraclePartitionOption,
     AvroCompressionCodec,
     AzureFunctionActivityMethod,
     AzureSearchIndexWriteBehaviorType,
@@ -1433,7 +1474,9 @@
     SparkServerType,
     SparkThriftTransportProtocol,
     SqlAlwaysEncryptedAkvAuthType,
+    SqlDWWriteBehaviorEnum,
     SqlPartitionOption,
+    SqlWriteBehaviorEnum,
     SsisLogLocationType,
     SsisObjectMetadataType,
     SsisPackageLocationType,
@@ -1462,6 +1505,13 @@
     'AmazonMWSLinkedService',
     'AmazonMWSObjectDataset',
     'AmazonMWSSource',
+    'AmazonRdsForOracleLinkedService',
+    'AmazonRdsForOraclePartitionSettings',
+    'AmazonRdsForOracleSource',
+    'AmazonRdsForOracleTableDataset',
+    'AmazonRdsForSqlServerLinkedService',
+    'AmazonRdsForSqlServerSource',
+    'AmazonRdsForSqlServerTableDataset',
     'AmazonRedshiftLinkedService',
     'AmazonRedshiftSource',
     'AmazonRedshiftTableDataset',
@@ -1687,9 +1737,12 @@
     'ExcelDataset',
     'ExcelSource',
     'ExecuteDataFlowActivity',
+    'ExecuteDataFlowActivityTypeProperties',
     'ExecuteDataFlowActivityTypePropertiesCompute',
     'ExecutePipelineActivity',
+    'ExecutePowerQueryActivityTypeProperties',
     'ExecuteSSISPackageActivity',
+    'ExecuteWranglingDataflowActivity',
     'ExecutionActivity',
     'ExportSettings',
     'ExposureControlBatchRequest',
@@ -1777,13 +1830,18 @@
     'IntegrationRuntimeComputeProperties',
     'IntegrationRuntimeConnectionInfo',
     'IntegrationRuntimeCustomSetupScriptProperties',
+    'IntegrationRuntimeCustomerVirtualNetwork',
     'IntegrationRuntimeDataFlowProperties',
     'IntegrationRuntimeDataProxyProperties',
     'IntegrationRuntimeDebugResource',
     'IntegrationRuntimeListResponse',
     'IntegrationRuntimeMonitoringData',
     'IntegrationRuntimeNodeIpAddress',
     'IntegrationRuntimeNodeMonitoringData',
+    'IntegrationRuntimeOutboundNetworkDependenciesCategoryEndpoint',
+    'IntegrationRuntimeOutboundNetworkDependenciesEndpoint',
+    'IntegrationRuntimeOutboundNetworkDependenciesEndpointDetails',
+    'IntegrationRuntimeOutboundNetworkDependenciesEndpointsResponse',
     'IntegrationRuntimeReference',
     'IntegrationRuntimeRegenerateKeyParameters',
     'IntegrationRuntimeResource',
@@ -1924,6 +1982,8 @@
     'PostgreSqlLinkedService',
     'PostgreSqlSource',
     'PostgreSqlTableDataset',
+    'PowerQuerySink',
+    'PowerQuerySource',
     'PrestoLinkedService',
     'PrestoObjectDataset',
     'PrestoSource',
@@ -2035,6 +2095,7 @@
     'SqlAlwaysEncryptedProperties',
     'SqlDWSink',
     'SqlDWSource',
+    'SqlDWUpsertSettings',
     'SqlMISink',
     'SqlMISource',
     'SqlPartitionSettings',
@@ -2045,6 +2106,7 @@
     'SqlServerTableDataset',
     'SqlSink',
     'SqlSource',
+    'SqlUpsertSettings',
     'SquareLinkedService',
     'SquareObjectDataset',
     'SquareSource',
@@ -2114,6 +2176,7 @@
     'WebLinkedServiceTypeProperties',
     'WebSource',
     'WebTableDataset',
+    'WranglingDataFlow',
     'XeroLinkedService',
     'XeroObjectDataset',
     'XeroSource',
@@ -2124,6 +2187,7 @@
     'ZohoLinkedService',
     'ZohoObjectDataset',
     'ZohoSource',
+    'AmazonRdsForOraclePartitionOption',
     'AvroCompressionCodec',
     'AzureFunctionActivityMethod',
     'AzureSearchIndexWriteBehaviorType',
@@ -2201,7 +2265,9 @@
     'SparkServerType',
     'SparkThriftTransportProtocol',
     'SqlAlwaysEncryptedAkvAuthType',
+    'SqlDWWriteBehaviorEnum',
     'SqlPartitionOption',
+    'SqlWriteBehaviorEnum',
     'SsisLogLocationType',
     'SsisObjectMetadataType',
     'SsisPackageLocationType',

sdk/datafactory/azure-mgmt-datafactory/azure/mgmt/datafactory/models/_data_factory_management_client_enums.py

@@ -26,6 +26,12 @@ def __getattr__(cls, name):
             raise AttributeError(name)
 
 
+class AmazonRdsForOraclePartitionOption(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)):
+
+    NONE = "None"
+    PHYSICAL_PARTITIONS_OF_TABLE = "PhysicalPartitionsOfTable"
+    DYNAMIC_RANGE = "DynamicRange"
+
 class AvroCompressionCodec(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)):
 
     NONE = "none"
@@ -696,6 +702,14 @@ class SqlAlwaysEncryptedAkvAuthType(with_metaclass(_CaseInsensitiveEnumMeta, str
 
     SERVICE_PRINCIPAL = "ServicePrincipal"
     MANAGED_IDENTITY = "ManagedIdentity"
+    USER_ASSIGNED_MANAGED_IDENTITY = "UserAssignedManagedIdentity"
+
+class SqlDWWriteBehaviorEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)):
+    """Specify the write behavior when copying data into sql dw.
+    """
+
+    INSERT = "Insert"
+    UPSERT = "Upsert"
 
 class SqlPartitionOption(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)):
     """The partition mechanism that will be used for Sql read in parallel.
@@ -705,6 +719,14 @@ class SqlPartitionOption(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)):
     PHYSICAL_PARTITIONS_OF_TABLE = "PhysicalPartitionsOfTable"
     DYNAMIC_RANGE = "DynamicRange"
 
+class SqlWriteBehaviorEnum(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)):
+    """Specify the write behavior when copying data into sql.
+    """
+
+    INSERT = "Insert"
+    UPSERT = "Upsert"
+    STORED_PROCEDURE = "StoredProcedure"
+
 class SsisLogLocationType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)):
     """The type of SSIS log location.
     """