Skip to content

{Storage Actions} Migrate to TypeSpec based module #9579

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
necusjz merged 3 commits into from
Feb 9, 2026
Merged

{Storage Actions} Migrate to TypeSpec based module #9579

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
65a11ce
chore: add tsp generation
necusjz Feb 5, 2026
fd08140
test: rerun test cases
necusjz Feb 5, 2026
805d3e3
docs: update changelog
necusjz Feb 5, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/storage-actions/HISTORY.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,10 @@

Release History
===============
1.1.0
++++++
* Migrated to TypeSpec based module.

1.0.0
++++++
* GA Storage Actions with api-version 2023-01-01
Expand Down
86 changes: 43 additions & 43 deletions src/storage-actions/azext_storage_actions/aaz/latest/storage_actions/task/_create.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ def _build_arguments_schema(cls, *args, **kwargs):

operations = cls._args_schema.action.else_.operations
operations.Element = AAZObjectArg()
cls._build_args_storage_task_operation_create(operations.Element)
cls._build_args_storagetaskoperation_create_or_update_create(operations.Element)

if_ = cls._args_schema.action.if_
if_.condition = AAZStrArg(
Expand All @@ -174,51 +174,51 @@ def _build_arguments_schema(cls, *args, **kwargs):

operations = cls._args_schema.action.if_.operations
operations.Element = AAZObjectArg()
cls._build_args_storage_task_operation_create(operations.Element)
cls._build_args_storagetaskoperation_create_or_update_create(operations.Element)
return cls._args_schema

_args_storage_task_operation_create = None
_args_storagetaskoperation_create_or_update_create = None

@classmethod
def _build_args_storage_task_operation_create(cls, _schema):
if cls._args_storage_task_operation_create is not None:
_schema.name = cls._args_storage_task_operation_create.name
_schema.on_failure = cls._args_storage_task_operation_create.on_failure
_schema.on_success = cls._args_storage_task_operation_create.on_success
_schema.parameters = cls._args_storage_task_operation_create.parameters
def _build_args_storagetaskoperation_create_or_update_create(cls, _schema):
if cls._args_storagetaskoperation_create_or_update_create is not None:
_schema.name = cls._args_storagetaskoperation_create_or_update_create.name
_schema.on_failure = cls._args_storagetaskoperation_create_or_update_create.on_failure
_schema.on_success = cls._args_storagetaskoperation_create_or_update_create.on_success
_schema.parameters = cls._args_storagetaskoperation_create_or_update_create.parameters
return

cls._args_storage_task_operation_create = AAZObjectArg()
cls._args_storagetaskoperation_create_or_update_create = AAZObjectArg()

storage_task_operation_create = cls._args_storage_task_operation_create
storage_task_operation_create.name = AAZStrArg(
storagetaskoperation_create_or_update_create = cls._args_storagetaskoperation_create_or_update_create
storagetaskoperation_create_or_update_create.name = AAZStrArg(
options=["name"],
help="The operation to be performed on the object.",
required=True,
enum={"DeleteBlob": "DeleteBlob", "SetBlobExpiry": "SetBlobExpiry", "SetBlobImmutabilityPolicy": "SetBlobImmutabilityPolicy", "SetBlobLegalHold": "SetBlobLegalHold", "SetBlobTags": "SetBlobTags", "SetBlobTier": "SetBlobTier", "UndeleteBlob": "UndeleteBlob"},
)
storage_task_operation_create.on_failure = AAZStrArg(
storagetaskoperation_create_or_update_create.on_failure = AAZStrArg(
options=["on-failure"],
help="Action to be taken when the operation fails for a object.",
enum={"break": "break"},
)
storage_task_operation_create.on_success = AAZStrArg(
storagetaskoperation_create_or_update_create.on_success = AAZStrArg(
options=["on-success"],
help="Action to be taken when the operation is successful for a object.",
enum={"continue": "continue"},
)
storage_task_operation_create.parameters = AAZDictArg(
storagetaskoperation_create_or_update_create.parameters = AAZDictArg(
options=["parameters"],
help="Key-value parameters for the operation.",
)

parameters = cls._args_storage_task_operation_create.parameters
parameters = cls._args_storagetaskoperation_create_or_update_create.parameters
parameters.Element = AAZStrArg()

_schema.name = cls._args_storage_task_operation_create.name
_schema.on_failure = cls._args_storage_task_operation_create.on_failure
_schema.on_success = cls._args_storage_task_operation_create.on_success
_schema.parameters = cls._args_storage_task_operation_create.parameters
_schema.name = cls._args_storagetaskoperation_create_or_update_create.name
_schema.on_failure = cls._args_storagetaskoperation_create_or_update_create.on_failure
_schema.on_success = cls._args_storagetaskoperation_create_or_update_create.on_success
_schema.parameters = cls._args_storagetaskoperation_create_or_update_create.parameters

def _execute_operations(self):
self.pre_operations()
Expand Down Expand Up @@ -363,7 +363,7 @@ def content(self):

operations = _builder.get(".properties.action.else.operations")
if operations is not None:
_CreateHelper._build_schema_storage_task_operation_create(operations.set_elements(AAZObjectType, "."))
_CreateHelper._build_schema_storagetaskoperation_create_or_update_create(operations.set_elements(AAZObjectType, "."))

if_ = _builder.get(".properties.action.if")
if if_ is not None:
Expand All @@ -372,7 +372,7 @@ def content(self):

operations = _builder.get(".properties.action.if.operations")
if operations is not None:
_CreateHelper._build_schema_storage_task_operation_create(operations.set_elements(AAZObjectType, "."))
_CreateHelper._build_schema_storagetaskoperation_create_or_update_create(operations.set_elements(AAZObjectType, "."))

tags = _builder.get(".tags")
if tags is not None:
Expand Down Expand Up @@ -489,7 +489,7 @@ def _build_schema_on_200_201(cls):

operations = cls._schema_on_200_201.properties.action["else"].operations
operations.Element = AAZObjectType()
_CreateHelper._build_schema_storage_task_operation_read(operations.Element)
_CreateHelper._build_schema_storagetaskoperation_read(operations.Element)

if_ = cls._schema_on_200_201.properties.action["if"]
if_.condition = AAZStrType(
Expand All @@ -501,7 +501,7 @@ def _build_schema_on_200_201(cls):

operations = cls._schema_on_200_201.properties.action["if"].operations
operations.Element = AAZObjectType()
_CreateHelper._build_schema_storage_task_operation_read(operations.Element)
_CreateHelper._build_schema_storagetaskoperation_read(operations.Element)

system_data = cls._schema_on_200_201.system_data
system_data.created_at = AAZStrType(
Expand Down Expand Up @@ -533,7 +533,7 @@ class _CreateHelper:
"""Helper class for Create"""

@classmethod
def _build_schema_storage_task_operation_create(cls, _builder):
def _build_schema_storagetaskoperation_create_or_update_create(cls, _builder):
if _builder is None:
return
_builder.set_prop("name", AAZStrType, ".name", typ_kwargs={"flags": {"required": True}})
Expand All @@ -545,38 +545,38 @@ def _build_schema_storage_task_operation_create(cls, _builder):
if parameters is not None:
parameters.set_elements(AAZStrType, ".")

_schema_storage_task_operation_read = None
_schema_storagetaskoperation_read = None

@classmethod
def _build_schema_storage_task_operation_read(cls, _schema):
if cls._schema_storage_task_operation_read is not None:
_schema.name = cls._schema_storage_task_operation_read.name
_schema.on_failure = cls._schema_storage_task_operation_read.on_failure
_schema.on_success = cls._schema_storage_task_operation_read.on_success
_schema.parameters = cls._schema_storage_task_operation_read.parameters
def _build_schema_storagetaskoperation_read(cls, _schema):
if cls._schema_storagetaskoperation_read is not None:
_schema.name = cls._schema_storagetaskoperation_read.name
_schema.on_failure = cls._schema_storagetaskoperation_read.on_failure
_schema.on_success = cls._schema_storagetaskoperation_read.on_success
_schema.parameters = cls._schema_storagetaskoperation_read.parameters
return

cls._schema_storage_task_operation_read = _schema_storage_task_operation_read = AAZObjectType()
cls._schema_storagetaskoperation_read = _schema_storagetaskoperation_read = AAZObjectType()

storage_task_operation_read = _schema_storage_task_operation_read
storage_task_operation_read.name = AAZStrType(
storagetaskoperation_read = _schema_storagetaskoperation_read
storagetaskoperation_read.name = AAZStrType(
flags={"required": True},
)
storage_task_operation_read.on_failure = AAZStrType(
storagetaskoperation_read.on_failure = AAZStrType(
serialized_name="onFailure",
)
storage_task_operation_read.on_success = AAZStrType(
storagetaskoperation_read.on_success = AAZStrType(
serialized_name="onSuccess",
)
storage_task_operation_read.parameters = AAZDictType()
storagetaskoperation_read.parameters = AAZDictType()

parameters = _schema_storage_task_operation_read.parameters
parameters = _schema_storagetaskoperation_read.parameters
parameters.Element = AAZStrType()

_schema.name = cls._schema_storage_task_operation_read.name
_schema.on_failure = cls._schema_storage_task_operation_read.on_failure
_schema.on_success = cls._schema_storage_task_operation_read.on_success
_schema.parameters = cls._schema_storage_task_operation_read.parameters
_schema.name = cls._schema_storagetaskoperation_read.name
_schema.on_failure = cls._schema_storagetaskoperation_read.on_failure
_schema.on_success = cls._schema_storagetaskoperation_read.on_success
_schema.parameters = cls._schema_storagetaskoperation_read.parameters


__all__ = ["Create"]
44 changes: 22 additions & 22 deletions src/storage-actions/azext_storage_actions/aaz/latest/storage_actions/task/_list.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -248,7 +248,7 @@ def _build_schema_on_200(cls):

operations = cls._schema_on_200.value.Element.properties.action["else"].operations
operations.Element = AAZObjectType()
_ListHelper._build_schema_storage_task_operation_read(operations.Element)
_ListHelper._build_schema_storagetaskoperation_read(operations.Element)

if_ = cls._schema_on_200.value.Element.properties.action["if"]
if_.condition = AAZStrType(
Expand All @@ -260,7 +260,7 @@ def _build_schema_on_200(cls):

operations = cls._schema_on_200.value.Element.properties.action["if"].operations
operations.Element = AAZObjectType()
_ListHelper._build_schema_storage_task_operation_read(operations.Element)
_ListHelper._build_schema_storagetaskoperation_read(operations.Element)

system_data = cls._schema_on_200.value.Element.system_data
system_data.created_at = AAZStrType(
Expand Down Expand Up @@ -467,7 +467,7 @@ def _build_schema_on_200(cls):

operations = cls._schema_on_200.value.Element.properties.action["else"].operations
operations.Element = AAZObjectType()
_ListHelper._build_schema_storage_task_operation_read(operations.Element)
_ListHelper._build_schema_storagetaskoperation_read(operations.Element)

if_ = cls._schema_on_200.value.Element.properties.action["if"]
if_.condition = AAZStrType(
Expand All @@ -479,7 +479,7 @@ def _build_schema_on_200(cls):

operations = cls._schema_on_200.value.Element.properties.action["if"].operations
operations.Element = AAZObjectType()
_ListHelper._build_schema_storage_task_operation_read(operations.Element)
_ListHelper._build_schema_storagetaskoperation_read(operations.Element)

system_data = cls._schema_on_200.value.Element.system_data
system_data.created_at = AAZStrType(
Expand Down Expand Up @@ -510,38 +510,38 @@ def _build_schema_on_200(cls):
class _ListHelper:
"""Helper class for List"""

_schema_storage_task_operation_read = None
_schema_storagetaskoperation_read = None

@classmethod
def _build_schema_storage_task_operation_read(cls, _schema):
if cls._schema_storage_task_operation_read is not None:
_schema.name = cls._schema_storage_task_operation_read.name
_schema.on_failure = cls._schema_storage_task_operation_read.on_failure
_schema.on_success = cls._schema_storage_task_operation_read.on_success
_schema.parameters = cls._schema_storage_task_operation_read.parameters
def _build_schema_storagetaskoperation_read(cls, _schema):
if cls._schema_storagetaskoperation_read is not None:
_schema.name = cls._schema_storagetaskoperation_read.name
_schema.on_failure = cls._schema_storagetaskoperation_read.on_failure
_schema.on_success = cls._schema_storagetaskoperation_read.on_success
_schema.parameters = cls._schema_storagetaskoperation_read.parameters
return

cls._schema_storage_task_operation_read = _schema_storage_task_operation_read = AAZObjectType()
cls._schema_storagetaskoperation_read = _schema_storagetaskoperation_read = AAZObjectType()

storage_task_operation_read = _schema_storage_task_operation_read
storage_task_operation_read.name = AAZStrType(
storagetaskoperation_read = _schema_storagetaskoperation_read
storagetaskoperation_read.name = AAZStrType(
flags={"required": True},
)
storage_task_operation_read.on_failure = AAZStrType(
storagetaskoperation_read.on_failure = AAZStrType(
serialized_name="onFailure",
)
storage_task_operation_read.on_success = AAZStrType(
storagetaskoperation_read.on_success = AAZStrType(
serialized_name="onSuccess",
)
storage_task_operation_read.parameters = AAZDictType()
storagetaskoperation_read.parameters = AAZDictType()

parameters = _schema_storage_task_operation_read.parameters
parameters = _schema_storagetaskoperation_read.parameters
parameters.Element = AAZStrType()

_schema.name = cls._schema_storage_task_operation_read.name
_schema.on_failure = cls._schema_storage_task_operation_read.on_failure
_schema.on_success = cls._schema_storage_task_operation_read.on_success
_schema.parameters = cls._schema_storage_task_operation_read.parameters
_schema.name = cls._schema_storagetaskoperation_read.name
_schema.on_failure = cls._schema_storagetaskoperation_read.on_failure
_schema.on_success = cls._schema_storagetaskoperation_read.on_success
_schema.parameters = cls._schema_storagetaskoperation_read.parameters


__all__ = ["List"]
4 changes: 2 additions & 2 deletions ...storage-actions/azext_storage_actions/aaz/latest/storage_actions/task/_list_assignment.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def _build_arguments_schema(cls, *args, **kwargs):

def _execute_operations(self):
self.pre_operations()
self.StorageTaskAssignmentList(ctx=self.ctx)()
self.StorageTasksList(ctx=self.ctx)()
self.post_operations()

@register_callback
Expand All @@ -82,7 +82,7 @@ def _output(self, *args, **kwargs):
next_link = self.deserialize_output(self.ctx.vars.instance.next_link)
return result, next_link

class StorageTaskAssignmentList(AAZHttpOperation):
class StorageTasksList(AAZHttpOperation):
CLIENT_TYPE = "MgmtClient"

def __call__(self, *args, **kwargs):
Expand Down
4 changes: 2 additions & 2 deletions src/storage-actions/azext_storage_actions/aaz/latest/storage_actions/task/_list_report.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ def _build_arguments_schema(cls, *args, **kwargs):

def _execute_operations(self):
self.pre_operations()
self.StorageTasksReportList(ctx=self.ctx)()
self.StorageTasksList(ctx=self.ctx)()
self.post_operations()

@register_callback
Expand All @@ -86,7 +86,7 @@ def _output(self, *args, **kwargs):
next_link = self.deserialize_output(self.ctx.vars.instance.next_link)
return result, next_link

class StorageTasksReportList(AAZHttpOperation):
class StorageTasksList(AAZHttpOperation):
CLIENT_TYPE = "MgmtClient"

def __call__(self, *args, **kwargs):
Expand Down
Loading
Loading