29 lines
No EOL
702 B
Python
29 lines
No EOL
702 B
Python
#!/usr/bin/env python3
|
|
"""
|
|
DSL code generator that can be run as a py_binary with proper dependencies.
|
|
"""
|
|
|
|
import sys
|
|
from databuild.dsl.python.generator_lib import generate_dsl_package
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) != 4:
|
|
print("Usage: generator.py <module_path> <graph_attr> <output_dir>", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
module_path = sys.argv[1]
|
|
graph_attr = sys.argv[2]
|
|
output_dir = sys.argv[3]
|
|
|
|
try:
|
|
generate_dsl_package(module_path, graph_attr, output_dir)
|
|
except Exception as e:
|
|
print(f"ERROR: {e}", file=sys.stderr)
|
|
import traceback
|
|
traceback.print_exc()
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |