01/01OS/start.py

"""
This is just for the Python package — we need a Python entrypoint.
Just starts `start.sh` with all the same command line arguments. Aliased to 01.
"""

import os
import subprocess
import sys
import psutil
from process_utils import kill_process_tree

def main():

    # Get command line arguments
    args = sys.argv[1:]

    # Get the directory of the current script
    dir_path = os.path.dirname(os.path.realpath(__file__))

    # Prepare the command
    command = [os.path.join(dir_path, 'start.sh')] + args

    try:
        # Start start.sh using psutil for better process management, and to kill all processes
        psutil.Popen(command)
    except KeyboardInterrupt:
        print("Exiting...")
        kill_process_tree()