Skip to content

Commit e0dc3c4

Browse files
Update code_report.py (Azure#39106)
* Update code_report.py * Fix AttributeError by using `__all__` for operations
1 parent fcdfb8e commit e0dc3c4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tools/azure-sdk-tools/packaging_tools/code_report.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def create_report(module_name: str) -> Dict[str, Any]:
5757
report["client"] = []
5858

5959
# Look for models first
60-
model_names = [model_name for model_name in dir(module_to_generate.models) if model_name[0].isupper()]
60+
model_names = [model_name for model_name in module_to_generate.models.__all__ if model_name[0].isupper()]
6161
is_new_model = hasattr(module_to_generate, "_model_base")
6262
for model_name in model_names:
6363
model_cls = getattr(module_to_generate.models, model_name)
@@ -69,7 +69,7 @@ def create_report(module_name: str) -> Dict[str, Any]:
6969
report["models"]["enums"][model_name] = create_model_report(model_cls, is_new_model)
7070
# Look for operation groups
7171
try:
72-
operations_classes = [op_name for op_name in dir(module_to_generate.operations) if op_name[0].isupper()]
72+
operations_classes = [op_name for op_name in module_to_generate.operations.__all__ if op_name[0].isupper()]
7373
except AttributeError:
7474
# This guy has no "operations", this is possible (Cognitive Services). Just skip it then.
7575
operations_classes = []

0 commit comments

Comments
 (0)