2024-11-17 20:57:16 +08:00

22 lines
605 B
Python

import os
import argparse
import subprocess
if __name__ == "__main__":
parser = argparse.ArgumentParser(escription="Script to set environment variables and run different environments.")
parser.add_argument(
"environment",
choices=["production", "development"],
nargs="?",
default="production",
help="Specify the environment to run.",
)
args = parser.parse_args()
try:
if args.environment:
os.environ["CURRENT_ENV"] = args.environment
subprocess.run(["python", "main.py"])
except KeyboardInterrupt:
pass