diff --git a/examples/podcast_reviews/BUILD.bazel b/examples/podcast_reviews/BUILD.bazel index 169c5cd..190f857 100644 --- a/examples/podcast_reviews/BUILD.bazel +++ b/examples/podcast_reviews/BUILD.bazel @@ -184,12 +184,6 @@ databuild_job( binary = ":test_job_binary", ) -py_binary( - name = "test_job_binary", - srcs = ["unified_job.py"], - main = "unified_job.py", -) - # Test target py_binary( name = "test_jobs", diff --git a/examples/podcast_reviews/podcast_reviews.iml b/examples/podcast_reviews/podcast_reviews.iml new file mode 100644 index 0000000..ad3c0a3 --- /dev/null +++ b/examples/podcast_reviews/podcast_reviews.iml @@ -0,0 +1,9 @@ + + + + + + + + + \ No newline at end of file diff --git a/examples/podcast_reviews/unified_job.py b/examples/podcast_reviews/unified_job.py deleted file mode 100644 index f482db9..0000000 --- a/examples/podcast_reviews/unified_job.py +++ /dev/null @@ -1,47 +0,0 @@ -#!/usr/bin/env python3 - -import sys -import json - -def main(): - if len(sys.argv) < 2: - print("Usage: unified_job.py {config|exec} [args...]", file=sys.stderr) - sys.exit(1) - - command = sys.argv[1] - - if command == "config": - handle_config(sys.argv[2:]) - elif command == "exec": - handle_exec(sys.argv[2:]) - else: - print(f"Unknown command: {command}", file=sys.stderr) - print("Usage: unified_job.py {config|exec} [args...]", file=sys.stderr) - sys.exit(1) - -def handle_config(args): - if len(args) < 1: - print("Config mode requires partition ref", file=sys.stderr) - sys.exit(1) - - partition_ref = args[0] - - config = { - "configs": [{ - "outputs": [{"str": partition_ref}], - "inputs": [], - "args": ["Hello", "gorgeous", partition_ref], - "env": {"PARTITION_REF": partition_ref} - }] - } - - print(json.dumps(config)) - -def handle_exec(args): - print("What a time to be alive.") - print(f"Partition ref: {os.getenv('PARTITION_REF', 'unknown')}") - print(f"Args: {args}") - -if __name__ == "__main__": - import os - main() \ No newline at end of file