Add Batch data plane commands.

azure-cli.pyproj

@@ -119,6 +119,11 @@
     <Compile Include="command_modules\azure-cli-appservice\azure\cli\__init__.py" />
     <Compile Include="command_modules\azure-cli-appservice\azure\__init__.py" />
     <Compile Include="command_modules\azure-cli-appservice\setup.py" />
+    <Compile Include="command_modules\azure-cli-batch\azure\cli\command_modules\batch\tests\test_batch_commands.py" />
+    <Compile Include="command_modules\azure-cli-batch\azure\cli\command_modules\batch\tests\test_batch_data_plane_commands.py" />
+    <Compile Include="command_modules\azure-cli-batch\azure\cli\command_modules\batch\tests\test_batch_data_plane_command_base.py" />
+    <Compile Include="command_modules\azure-cli-batch\azure\cli\command_modules\batch\tests\test_batch_pool_commands.py" />
+    <Compile Include="command_modules\azure-cli-batch\azure\cli\command_modules\batch\_command_type.py" />
     <Compile Include="command_modules\azure-cli-cloud\azure\cli\command_modules\cloud\commands.py" />
     <Compile Include="command_modules\azure-cli-cloud\azure\cli\command_modules\cloud\custom.py" />
     <Compile Include="command_modules\azure-cli-cloud\azure\cli\command_modules\cloud\_help.py" />

src/command_modules/azure-cli-batch/azure/cli/command_modules/batch/__init__.py

@@ -3,10 +3,12 @@
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 
-import azure.cli.command_modules.batch._help # pylint: disable=unused-import
+import azure.cli.command_modules.batch._help  # pylint: disable=unused-import
+
 
 def load_params(_):
-    import azure.cli.command_modules.batch._params #pylint: disable=redefined-outer-name
+    import azure.cli.command_modules.batch._params  # pylint: disable=redefined-outer-name
+
 
 def load_commands():
-    import azure.cli.command_modules.batch.commands #pylint: disable=redefined-outer-name
+    import azure.cli.command_modules.batch.commands  # pylint: disable=redefined-outer-name

src/command_modules/azure-cli-batch/azure/cli/command_modules/batch/_client_factory.py

@@ -3,9 +3,73 @@
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 
-from azure.mgmt.batch import BatchManagementClient
 
-from azure.cli.core.commands.client_factory import get_mgmt_service_client
+def account_mgmt_client_factory(kwargs):
+    return batch_client_factory(**kwargs).batch_account
+
+
+def application_mgmt_client_factory(kwargs):
+    return batch_client_factory(**kwargs).application
+
+
+def application_package_client_factory(kwargs):
+    return batch_client_factory(**kwargs).application_package
+
+
+def location_client_factory(kwargs):
+    return batch_client_factory(**kwargs).location
+
+
+def application_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).application
+
+
+def account_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).account
+
+
+def certificate_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).certificate
+
+
+def pool_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).pool
+
+
+def job_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).job
+
+
+def job_schedule_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).job_schedule
+
+
+def task_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).task
+
+
+def file_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).file
+
+
+def compute_node_client_factory(kwargs):
+    return batch_data_service_factory(kwargs).compute_node
+
 
 def batch_client_factory(**_):
+    from azure.mgmt.batch import BatchManagementClient
+    from azure.cli.core.commands.client_factory import get_mgmt_service_client
+
     return get_mgmt_service_client(BatchManagementClient)
+
+
+def batch_data_service_factory(kwargs):
+    import azure.batch.batch_service_client as batch
+    import azure.batch.batch_auth as batchauth
+
+    account_name = kwargs.pop('account_name', None)
+    account_key = kwargs.pop('account_key', None)
+    account_endpoint = kwargs.pop('account_endpoint', None)
+    kwargs.pop('force', None)
+    credentials = batchauth.SharedKeyCredentials(account_name, account_key)
+    return batch.BatchServiceClient(credentials, base_url=account_endpoint)