Skip to content

Commit 26393d3

Browse files
committed
chore(wip): code generation
1 parent 5d3bc75 commit 26393d3

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

frost_sta_client/odata_codegen/generator.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1237,12 +1237,13 @@ def snake(name: str) -> str:
12371237
lines.append("")
12381238

12391239
# __all__
1240-
exported.extend(list(model.get("entity_sets", {}).keys()))
1240+
# __all__ - export only classes/complex types plus helper mapping
12411241
lines.append("__all__ = [")
1242-
for name in sorted(set(exported)):
1242+
for name in sorted(set(exported + ["ENTITY_SETS"])):
12431243
lines.append(f" '{name}',")
12441244
lines.append("]")
12451245
lines.append("")
1246+
lines.append("")
12461247

12471248
# Also write a package __init__.py so that `import <output-dir> as model` works as in README.
12481249
init_path = os.path.join(out_dir, "__init__.py")

0 commit comments

Comments
 (0)