From 364949ff39daa308d9643e6bacbc8ba0cb569114 Mon Sep 17 00:00:00 2001 From: kai ru Date: Thu, 21 Nov 2024 09:43:49 +0800 Subject: [PATCH] Fix bugs for identity subcommand generation and class name with doc --- src/aaz_dev/command/controller/workspace_cfg_editor.py | 7 ++++++- src/aaz_dev/swagger/model/schema/cmd_builder.py | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/src/aaz_dev/command/controller/workspace_cfg_editor.py b/src/aaz_dev/command/controller/workspace_cfg_editor.py index 0b5f79ee..a969fac2 100644 --- a/src/aaz_dev/command/controller/workspace_cfg_editor.py +++ b/src/aaz_dev/command/controller/workspace_cfg_editor.py @@ -1067,7 +1067,12 @@ def build_identity_subresource(self, resource_id, temp_generic_update_cmd=None): update_cmd.name = ' '.join(update_cmd_names) elif update_by != "GenericOnly": return - update_op, _, identity_schema, identity_schema_idx = self.find_identity_schema_in_command(update_cmd) + + identity_schema_info = self.find_identity_schema_in_command(update_cmd) + if not identity_schema_info: + return + + update_op, _, identity_schema, identity_schema_idx = identity_schema_info subresource_idx = self.schema_idx_to_subresource_idx(identity_schema_idx) assert subresource_idx diff --git a/src/aaz_dev/swagger/model/schema/cmd_builder.py b/src/aaz_dev/swagger/model/schema/cmd_builder.py index e8306498..172bab6b 100644 --- a/src/aaz_dev/swagger/model/schema/cmd_builder.py +++ b/src/aaz_dev/swagger/model/schema/cmd_builder.py @@ -232,7 +232,7 @@ def build_schema(self, schema): def _get_cls_definition_name(self, schema): assert isinstance(schema, ReferenceSchema) - schema_cls_name = f"{to_camel_case(schema.ref.split('/')[-1])}_{self.mutability}" + schema_cls_name = f"{to_camel_case(schema.ref.split('/')[-1].replace('.', ' '))}_{self.mutability}" if self.mutability != MutabilityEnum.Read: if self.read_only: schema_cls_name += "_read"