We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5d3bc75 commit 26393d3Copy full SHA for 26393d3
1 file changed
frost_sta_client/odata_codegen/generator.py
@@ -1237,12 +1237,13 @@ def snake(name: str) -> str:
1237
lines.append("")
1238
1239
# __all__
1240
- exported.extend(list(model.get("entity_sets", {}).keys()))
+ # __all__ - export only classes/complex types plus helper mapping
1241
lines.append("__all__ = [")
1242
- for name in sorted(set(exported)):
+ for name in sorted(set(exported + ["ENTITY_SETS"])):
1243
lines.append(f" '{name}',")
1244
lines.append("]")
1245
1246
+ lines.append("")
1247
1248
# Also write a package __init__.py so that `import <output-dir> as model` works as in README.
1249
init_path = os.path.join(out_dir, "__init__.py")
0 commit comments