diff --git a/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/__init__.py b/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/__init__.py index edc9706441f..f2cf802b96d 100644 --- a/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/__init__.py +++ b/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/__init__.py @@ -14,5 +14,4 @@ from ._list import * from ._query_status import * from ._show import * -from ._stop import * -from ._wait import * +from ._stop import * \ No newline at end of file diff --git a/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/_create.py b/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/_create.py index 72e18165f8b..cd2a654258b 100644 --- a/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/_create.py +++ b/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/_create.py @@ -114,7 +114,6 @@ def _build_arguments_schema(cls, *args, **kwargs): options=["--time-limit-in-seconds"], arg_group="Properties", help="Maximum duration of the capture session in seconds.", - default=18000, fmt=AAZIntArgFormat( maximum=18000, minimum=0, @@ -124,7 +123,6 @@ def _build_arguments_schema(cls, *args, **kwargs): options=["--total-bytes", "--total-bytes-per-session"], arg_group="Properties", help="Maximum size of the capture output.", - default=1073741824, fmt=AAZIntArgFormat( maximum=4294967295, minimum=0, diff --git a/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/_wait.py b/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/_wait.py deleted file mode 100644 index 67fa0b0d1af..00000000000 --- a/src/azure-cli/azure/cli/command_modules/network/aaz/latest/network/network_watcher/packet_capture/_wait.py +++ /dev/null @@ -1,268 +0,0 @@ -# -------------------------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# -# Code generated by aaz-dev-tools -# -------------------------------------------------------------------------------------------- - -# pylint: skip-file -# flake8: noqa - -from azure.cli.core.aaz import * - - -@register_command( - "network network-watcher packet-capture wait", -) -class Wait(AAZWaitCommand): - """Place the CLI in a waiting state until a condition is met. - """ - - _aaz_info = { - "resources": [ - ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.network/networkwatchers/{}/packetcaptures/{}", "2024-05-01"], - ] - } - - def _handler(self, command_args): - super()._handler(command_args) - self._execute_operations() - return self._output() - - _args_schema = None - - @classmethod - def _build_arguments_schema(cls, *args, **kwargs): - if cls._args_schema is not None: - return cls._args_schema - cls._args_schema = super()._build_arguments_schema(*args, **kwargs) - - # define Arg Group "" - - _args_schema = cls._args_schema - _args_schema.network_watcher_name = AAZStrArg( - options=["--network-watcher-name"], - help="The name of the network watcher.", - required=True, - id_part="name", - ) - _args_schema.packet_capture_name = AAZStrArg( - options=["-n", "--name", "--packet-capture-name"], - help="The name of the packet capture session.", - required=True, - id_part="child_name_1", - ) - _args_schema.resource_group = AAZResourceGroupNameArg( - required=True, - ) - return cls._args_schema - - def _execute_operations(self): - self.pre_operations() - self.PacketCapturesGet(ctx=self.ctx)() - self.post_operations() - - @register_callback - def pre_operations(self): - pass - - @register_callback - def post_operations(self): - pass - - def _output(self, *args, **kwargs): - result = self.deserialize_output(self.ctx.vars.instance, client_flatten=False) - return result - - class PacketCapturesGet(AAZHttpOperation): - CLIENT_TYPE = "MgmtClient" - - def __call__(self, *args, **kwargs): - request = self.make_request() - session = self.client.send_request(request=request, stream=False, **kwargs) - if session.http_response.status_code in [200]: - return self.on_200(session) - - return self.on_error(session.http_response) - - @property - def url(self): - return self.client.format_url( - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/networkWatchers/{networkWatcherName}/packetCaptures/{packetCaptureName}", - **self.url_parameters - ) - - @property - def method(self): - return "GET" - - @property - def error_format(self): - return "ODataV4Format" - - @property - def url_parameters(self): - parameters = { - **self.serialize_url_param( - "networkWatcherName", self.ctx.args.network_watcher_name, - required=True, - ), - **self.serialize_url_param( - "packetCaptureName", self.ctx.args.packet_capture_name, - required=True, - ), - **self.serialize_url_param( - "resourceGroupName", self.ctx.args.resource_group, - required=True, - ), - **self.serialize_url_param( - "subscriptionId", self.ctx.subscription_id, - required=True, - ), - } - return parameters - - @property - def query_parameters(self): - parameters = { - **self.serialize_query_param( - "api-version", "2024-05-01", - required=True, - ), - } - return parameters - - @property - def header_parameters(self): - parameters = { - **self.serialize_header_param( - "Accept", "application/json", - ), - } - return parameters - - def on_200(self, session): - data = self.deserialize_http_content(session) - self.ctx.set_var( - "instance", - data, - schema_builder=self._build_schema_on_200 - ) - - _schema_on_200 = None - - @classmethod - def _build_schema_on_200(cls): - if cls._schema_on_200 is not None: - return cls._schema_on_200 - - cls._schema_on_200 = AAZObjectType() - - _schema_on_200 = cls._schema_on_200 - _schema_on_200.etag = AAZStrType( - flags={"read_only": True}, - ) - _schema_on_200.id = AAZStrType( - flags={"read_only": True}, - ) - _schema_on_200.name = AAZStrType( - flags={"read_only": True}, - ) - _schema_on_200.properties = AAZObjectType( - flags={"client_flatten": True}, - ) - - properties = cls._schema_on_200.properties - properties.bytes_to_capture_per_packet = AAZIntType( - serialized_name="bytesToCapturePerPacket", - ) - properties.capture_settings = AAZObjectType( - serialized_name="captureSettings", - ) - properties.continuous_capture = AAZBoolType( - serialized_name="continuousCapture", - ) - properties.filters = AAZListType() - properties.provisioning_state = AAZStrType( - serialized_name="provisioningState", - flags={"read_only": True}, - ) - properties.scope = AAZObjectType() - properties.storage_location = AAZObjectType( - serialized_name="storageLocation", - flags={"required": True}, - ) - properties.target = AAZStrType( - flags={"required": True}, - ) - properties.target_type = AAZStrType( - serialized_name="targetType", - ) - properties.time_limit_in_seconds = AAZIntType( - serialized_name="timeLimitInSeconds", - ) - properties.total_bytes_per_session = AAZIntType( - serialized_name="totalBytesPerSession", - ) - - capture_settings = cls._schema_on_200.properties.capture_settings - capture_settings.file_count = AAZIntType( - serialized_name="fileCount", - ) - capture_settings.file_size_in_bytes = AAZIntType( - serialized_name="fileSizeInBytes", - ) - capture_settings.session_time_limit_in_seconds = AAZIntType( - serialized_name="sessionTimeLimitInSeconds", - ) - - filters = cls._schema_on_200.properties.filters - filters.Element = AAZObjectType() - - _element = cls._schema_on_200.properties.filters.Element - _element.local_ip_address = AAZStrType( - serialized_name="localIPAddress", - ) - _element.local_port = AAZStrType( - serialized_name="localPort", - ) - _element.protocol = AAZStrType() - _element.remote_ip_address = AAZStrType( - serialized_name="remoteIPAddress", - ) - _element.remote_port = AAZStrType( - serialized_name="remotePort", - ) - - scope = cls._schema_on_200.properties.scope - scope.exclude = AAZListType() - scope.include = AAZListType() - - exclude = cls._schema_on_200.properties.scope.exclude - exclude.Element = AAZStrType() - - include = cls._schema_on_200.properties.scope.include - include.Element = AAZStrType() - - storage_location = cls._schema_on_200.properties.storage_location - storage_location.file_path = AAZStrType( - serialized_name="filePath", - ) - storage_location.local_path = AAZStrType( - serialized_name="localPath", - ) - storage_location.storage_id = AAZStrType( - serialized_name="storageId", - ) - storage_location.storage_path = AAZStrType( - serialized_name="storagePath", - ) - - return cls._schema_on_200 - - -class _WaitHelper: - """Helper class for Wait""" - - -__all__ = ["Wait"]