[AutoPR azure-mgmt-batch] Updates for Batch Management 2020-03 API version

sdk/batch/azure-mgmt-batch/azure/mgmt/batch/_batch_management_client.py

@@ -19,6 +19,8 @@
 from .operations import LocationOperations
 from .operations import Operations
 from .operations import CertificateOperations
+from .operations import PrivateLinkResourceOperations
+from .operations import PrivateEndpointConnectionOperations
 from .operations import PoolOperations
 from . import models
 
@@ -41,6 +43,10 @@ class BatchManagementClient(SDKClient):
     :vartype operations: azure.mgmt.batch.operations.Operations
     :ivar certificate: Certificate operations
     :vartype certificate: azure.mgmt.batch.operations.CertificateOperations
+    :ivar private_link_resource: PrivateLinkResource operations
+    :vartype private_link_resource: azure.mgmt.batch.operations.PrivateLinkResourceOperations
+    :ivar private_endpoint_connection: PrivateEndpointConnection operations
+    :vartype private_endpoint_connection: azure.mgmt.batch.operations.PrivateEndpointConnectionOperations
     :ivar pool: Pool operations
     :vartype pool: azure.mgmt.batch.operations.PoolOperations
 
@@ -60,7 +66,7 @@ def __init__(
         super(BatchManagementClient, self).__init__(self.config.credentials, self.config)
 
         client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
-        self.api_version = '2019-08-01'
+        self.api_version = '2020-03-01'
         self._serialize = Serializer(client_models)
         self._deserialize = Deserializer(client_models)
 
@@ -76,5 +82,9 @@ def __init__(
             self._client, self.config, self._serialize, self._deserialize)
         self.certificate = CertificateOperations(
             self._client, self.config, self._serialize, self._deserialize)
+        self.private_link_resource = PrivateLinkResourceOperations(
+            self._client, self.config, self._serialize, self._deserialize)
+        self.private_endpoint_connection = PrivateEndpointConnectionOperations(
+            self._client, self.config, self._serialize, self._deserialize)
         self.pool = PoolOperations(
             self._client, self.config, self._serialize, self._deserialize)

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

@@ -41,10 +41,13 @@
     from ._models_py3 import DataDisk
     from ._models_py3 import DeleteCertificateError
     from ._models_py3 import DeploymentConfiguration
+    from ._models_py3 import DiskEncryptionConfiguration
+    from ._models_py3 import EncryptionProperties
     from ._models_py3 import EnvironmentSetting
     from ._models_py3 import FixedScaleSettings
     from ._models_py3 import ImageReference
     from ._models_py3 import InboundNatPool
+    from ._models_py3 import KeyVaultProperties
     from ._models_py3 import KeyVaultReference
     from ._models_py3 import LinuxUserConfiguration
     from ._models_py3 import MetadataItem
@@ -56,7 +59,12 @@
     from ._models_py3 import OperationDisplay
     from ._models_py3 import Pool
     from ._models_py3 import PoolEndpointConfiguration
+    from ._models_py3 import PrivateEndpoint
+    from ._models_py3 import PrivateEndpointConnection
+    from ._models_py3 import PrivateLinkResource
+    from ._models_py3 import PrivateLinkServiceConnectionState
     from ._models_py3 import ProxyResource
+    from ._models_py3 import PublicIPAddressConfiguration
     from ._models_py3 import ResizeError
     from ._models_py3 import ResizeOperationStatus
     from ._models_py3 import Resource
@@ -103,10 +111,13 @@
     from ._models import DataDisk
     from ._models import DeleteCertificateError
     from ._models import DeploymentConfiguration
+    from ._models import DiskEncryptionConfiguration
+    from ._models import EncryptionProperties
     from ._models import EnvironmentSetting
     from ._models import FixedScaleSettings
     from ._models import ImageReference
     from ._models import InboundNatPool
+    from ._models import KeyVaultProperties
     from ._models import KeyVaultReference
     from ._models import LinuxUserConfiguration
     from ._models import MetadataItem
@@ -118,7 +129,12 @@
     from ._models import OperationDisplay
     from ._models import Pool
     from ._models import PoolEndpointConfiguration
+    from ._models import PrivateEndpoint
+    from ._models import PrivateEndpointConnection
+    from ._models import PrivateLinkResource
+    from ._models import PrivateLinkServiceConnectionState
     from ._models import ProxyResource
+    from ._models import PublicIPAddressConfiguration
     from ._models import ResizeError
     from ._models import ResizeOperationStatus
     from ._models import Resource
@@ -139,9 +155,15 @@
 from ._paged_models import CertificatePaged
 from ._paged_models import OperationPaged
 from ._paged_models import PoolPaged
+from ._paged_models import PrivateEndpointConnectionPaged
+from ._paged_models import PrivateLinkResourcePaged
 from ._batch_management_client_enums import (
+    KeySource,
     PoolAllocationMode,
+    PublicNetworkAccessType,
     ProvisioningState,
+    PrivateEndpointConnectionProvisioningState,
+    PrivateLinkServiceConnectionStatus,
     AccountKeyType,
     PackageState,
     CertificateFormat,
@@ -150,10 +172,12 @@
     AllocationState,
     CachingType,
     StorageAccountType,
+    DiskEncryptionTarget,
     ComputeNodeDeallocationOption,
     InterNodeCommunicationState,
     InboundEndpointProtocol,
     NetworkSecurityGroupRuleAccess,
+    IPAddressProvisioningType,
     ComputeNodeFillType,
     ElevationLevel,
     LoginMode,
@@ -196,10 +220,13 @@
     'DataDisk',
     'DeleteCertificateError',
     'DeploymentConfiguration',
+    'DiskEncryptionConfiguration',
+    'EncryptionProperties',
     'EnvironmentSetting',
     'FixedScaleSettings',
     'ImageReference',
     'InboundNatPool',
+    'KeyVaultProperties',
     'KeyVaultReference',
     'LinuxUserConfiguration',
     'MetadataItem',
@@ -211,7 +238,12 @@
     'OperationDisplay',
     'Pool',
     'PoolEndpointConfiguration',
+    'PrivateEndpoint',
+    'PrivateEndpointConnection',
+    'PrivateLinkResource',
+    'PrivateLinkServiceConnectionState',
     'ProxyResource',
+    'PublicIPAddressConfiguration',
     'ResizeError',
     'ResizeOperationStatus',
     'Resource',
@@ -231,9 +263,15 @@
     'ApplicationPaged',
     'OperationPaged',
     'CertificatePaged',
+    'PrivateLinkResourcePaged',
+    'PrivateEndpointConnectionPaged',
     'PoolPaged',
+    'KeySource',
     'PoolAllocationMode',
+    'PublicNetworkAccessType',
     'ProvisioningState',
+    'PrivateEndpointConnectionProvisioningState',
+    'PrivateLinkServiceConnectionStatus',
     'AccountKeyType',
     'PackageState',
     'CertificateFormat',
@@ -242,10 +280,12 @@
     'AllocationState',
     'CachingType',
     'StorageAccountType',
+    'DiskEncryptionTarget',
     'ComputeNodeDeallocationOption',
     'InterNodeCommunicationState',
     'InboundEndpointProtocol',
     'NetworkSecurityGroupRuleAccess',
+    'IPAddressProvisioningType',
     'ComputeNodeFillType',
     'ElevationLevel',
     'LoginMode',

sdk/batch/azure-mgmt-batch/azure/mgmt/batch/models/_batch_management_client_enums.py

@@ -12,12 +12,24 @@
 from enum import Enum
 
 
+class KeySource(str, Enum):
+
+    microsoft_batch = "Microsoft.Batch"  #: Batch creates and manages the encryption keys used to protect the account data.
+    microsoft_key_vault = "Microsoft.KeyVault"  #: The encryption keys used to protect the account data are stored in an external key vault.
+
+
 class PoolAllocationMode(str, Enum):
 
     batch_service = "BatchService"  #: Pools will be allocated in subscriptions owned by the Batch service.
     user_subscription = "UserSubscription"  #: Pools will be allocated in a subscription owned by the user.
 
 
+class PublicNetworkAccessType(str, Enum):
+
+    enabled = "Enabled"  #: Enables connectivity to Azure Batch through public DNS.
+    disabled = "Disabled"  #: Disables public connectivity and enables private connectivity to Azure Batch Service through private endpoint resource.
+
+
 class ProvisioningState(str, Enum):
 
     invalid = "Invalid"  #: The account is in an invalid state.
@@ -28,6 +40,21 @@ class ProvisioningState(str, Enum):
     cancelled = "Cancelled"  #: The last operation for the account is cancelled.
 
 
+class PrivateEndpointConnectionProvisioningState(str, Enum):
+
+    succeeded = "Succeeded"  #: The connection status is final and is ready for use if Status is Approved.
+    updating = "Updating"  #: The user has requested that the connection status be updated, but the update operation has not yet completed. You may not reference the connection when connecting the Batch account.
+    failed = "Failed"  #: The user requested that the connection be updated and it failed. You may retry the update operation.
+
+
+class PrivateLinkServiceConnectionStatus(str, Enum):
+
+    approved = "Approved"  #: The private endpoint connection is approved and can be used to access Batch account
+    pending = "Pending"  #: The private endpoint connection is pending and cannot be used to access Batch account
+    rejected = "Rejected"  #: The private endpoint connection is rejected and cannot be used to access Batch account
+    disconnected = "Disconnected"  #: The private endpoint connection is disconnected and cannot be used to access Batch account
+
+
 class AccountKeyType(str, Enum):
 
     primary = "Primary"  #: The primary account key.
@@ -79,6 +106,12 @@ class StorageAccountType(str, Enum):
     premium_lrs = "Premium_LRS"  #: The data disk should use premium locally redundant storage.
 
 
+class DiskEncryptionTarget(str, Enum):
+
+    os_disk = "OsDisk"  #: The OS Disk on the compute node is encrypted.
+    temporary_disk = "TemporaryDisk"  #: The temporary disk on the compute node is encrypted. On Linux this encryption applies to other partitions (such as those on mounted data disks) when encryption occurs at boot time.
+
+
 class ComputeNodeDeallocationOption(str, Enum):
 
     requeue = "Requeue"  #: Terminate running task processes and requeue the tasks. The tasks will run again when a node is available. Remove nodes as soon as tasks have been terminated.
@@ -105,6 +138,13 @@ class NetworkSecurityGroupRuleAccess(str, Enum):
     deny = "Deny"  #: Deny access.
 
 
+class IPAddressProvisioningType(str, Enum):
+
+    batch_managed = "BatchManaged"  #: A public IP will be created and managed by Batch. There may be multiple public IPs depending on the size of the Pool.
+    user_managed = "UserManaged"  #: Public IPs are provided by the user and will be used to provision the Compute Nodes.
+    no_public_ip_addresses = "NoPublicIPAddresses"  #: No public IP Address will be created for the Compute Nodes in the Pool.
+
+
 class ComputeNodeFillType(str, Enum):
 
     spread = "Spread"  #: Tasks should be assigned evenly across all nodes in the pool.