add start scripts

Author: naderzare

start-team.py

@@ -0,0 +1,85 @@
+import subprocess
+import os
+import signal
+import threading
+import logging
+import argparse
+import check_requirements
+
+
+# Set up logging
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+
+def run_server_script(args):
+    # Start the server.py script as a new process group
+    process = subprocess.Popen(
+        ['python3', 'server.py', '--rpc-port', args.rpc_port],
+        preexec_fn=os.setsid,  # Create a new session and set the process group ID
+        stdout=subprocess.PIPE,
+        stderr=subprocess.STDOUT  # Capture stderr and redirect it to stdout
+    )
+    return process
+
+def run_start_script(args):
+    # Start the start.sh script in its own directory as a new process group
+    process = subprocess.Popen(
+        ['bash', 'start.sh', '-t', args.team_name, '--rpc-port', args.rpc_port, '--rpc-type', 'grpc'],
+        cwd='scripts/proxy',  # Corrected directory to where start.sh is located
+        preexec_fn=os.setsid,  # Create a new session and set the process group ID
+        stdout=subprocess.PIPE,
+        stderr=subprocess.STDOUT  # Capture stderr and redirect it to stdout
+    )
+    return process
+
+def stream_output(process, prefix):
+    # Stream output from the process and log it with a prefix
+    for line in iter(process.stdout.readline, b''):
+        logging.debug(f'{prefix} {line.decode().strip()}')
+    process.stdout.close()
+
+def kill_process_group(process):
+    try:
+        os.killpg(os.getpgid(process.pid), signal.SIGTERM)  # Send SIGTERM to the process group
+    except ProcessLookupError:
+        pass  # The process might have already exited
+
+if __name__ == "__main__":
+    # Set up argument parsing
+    parser = argparse.ArgumentParser(description='Run server and team scripts.')
+    parser.add_argument('-t', '--team_name', required=False, help='The name of the team', default='CLS')
+    parser.add_argument('--rpc-port', required=False, help='The port of the server', default='50051')
+    args = parser.parse_args()
+
+    try:
+        # Check Python requirements
+        logging.debug("Checking Python requirements...")
+        check_requirements.check_requirements()
+        
+        # Run the server.py script first
+        server_process = run_server_script(args)
+        logging.debug(f"Started server.py process with PID: {server_process.pid}")
+
+        # Run the start.sh script after server.py with the given arguments
+        start_process = run_start_script(args)
+        logging.debug(f"Started start.sh process with PID: {start_process.pid} with team name {args=}")
+
+        # Monitor both processes and log their outputs
+        server_thread = threading.Thread(target=stream_output, args=(server_process, 'server:'))
+        start_thread = threading.Thread(target=stream_output, args=(start_process, 'team:'))
+
+        server_thread.start()
+        start_thread.start()
+
+        # Wait for both threads to finish
+        server_thread.join()
+        start_thread.join()
+
+    except KeyboardInterrupt:
+        logging.debug("Interrupted! Killing all processes.")
+        kill_process_group(server_process)
+        kill_process_group(start_process)
+
+    finally:
+        # Ensure all processes are killed on exit
+        kill_process_group(server_process)
+        kill_process_group(start_process)

start-team.sh

@@ -0,0 +1,76 @@
+#!/bin/bash
+
+# Ensure the script exits if any command fails
+set -e
+# check scripts/proxy directory does not exist, raise error
+if [! -d scripts/proxy ]; then
+    echo "scripts/proxy directory does not exist"
+    exit 1
+fi
+
+team_name="CLS"
+rpc_port=50051
+
+# help function
+usage() {
+    echo "Usage: $0 [options]"
+    echo "Options:"
+    echo "  -t team_name: specify team name"
+    echo "  --rpc-port PORT - specifies rpc port (default: 50051)"
+    exit 1
+}
+
+while [ $# -gt 0 ]
+do
+  case $1 in
+    -t)
+      team_name=$2
+      shift
+      ;;
+    --rpc-port)
+      rpc_port=$2
+      shift
+      ;;
+    *)
+      echo 1>&2
+      echo "invalid option \"${1}\"." 1>&2
+      echo 1>&2
+      usage
+      exit 1
+      ;;
+  esac
+
+  shift 1
+done
+
+# Check Python requirements
+echo "Checking Python requirements..."
+python3 check_requirements.py
+
+# Start server.py in the background
+echo "Starting server.py..."
+python3 server.py --rpc-port $rpc_port &
+server_pid=$!
+
+# Function to kill server and team processes on exit
+cleanup() {
+    echo "Cleaning up..."
+    kill $server_pid
+    kill $start_pid
+}
+
+# Trap the exit signal to cleanup processes
+trap cleanup EXIT
+
+# Wait a moment to ensure the server has started (optional)
+sleep 2
+
+# Start start.sh script in the correct directory with arguments
+echo "Starting start.sh with team name: $team_name and ..."
+cd scripts/proxy
+bash start.sh -t "$team_name" --rpc-port $rpc_port --rpc-type grpc &
+start_pid=$!
+
+# Wait for both background processes to finish
+wait $server_pid
+wait $start_pid