53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
# Define the stacks
|
||
|
|
stack1 = ["controller", "teleop", "rsp", "lidar"]
|
||
|
|
stack2 = stack1 + ["mapping", "navigation"]
|
||
|
|
stack3 = stack1 + ["amcl", "navigation"]
|
||
|
|
|
||
|
|
# Function to start a stack
|
||
|
|
def start_stack(stack):
|
||
|
|
print(f"Starting {', '.join(stack)}...")
|
||
|
|
subprocess.run(["docker-compose", "up", "-d"] + stack)
|
||
|
|
|
||
|
|
# Function to stop a stack
|
||
|
|
def stop_stack(stack):
|
||
|
|
print(f"Stopping {', '.join(stack)}...")
|
||
|
|
subprocess.run(["docker-compose", "down", "-v", "--remove-orphans"] + stack)
|
||
|
|
|
||
|
|
# Main program
|
||
|
|
if __name__ == "__main__":
|
||
|
|
while True:
|
||
|
|
print("Options:")
|
||
|
|
print("1. Start Stack 1")
|
||
|
|
print("2. Start Stack 2")
|
||
|
|
print("3. Start Stack 3")
|
||
|
|
print("4. Stop Stack 1")
|
||
|
|
print("5. Stop Stack 2")
|
||
|
|
print("6. Stop Stack 3")
|
||
|
|
print("7. Quit")
|
||
|
|
|
||
|
|
choice = input("Enter your choice: ")
|
||
|
|
|
||
|
|
if choice == "1":
|
||
|
|
stop_stack(stack2 + stack3) # Stop stack2 and stack3 to avoid interference
|
||
|
|
start_stack(stack1)
|
||
|
|
elif choice == "2":
|
||
|
|
stop_stack(stack1 + stack3) # Stop stack1 and stack3 to avoid interference
|
||
|
|
start_stack(stack2)
|
||
|
|
elif choice == "3":
|
||
|
|
stop_stack(stack1 + stack2) # Stop stack1 and stack2 to avoid interference
|
||
|
|
start_stack(stack3)
|
||
|
|
elif choice == "4":
|
||
|
|
stop_stack(stack1)
|
||
|
|
elif choice == "5":
|
||
|
|
stop_stack(stack2)
|
||
|
|
elif choice == "6":
|
||
|
|
stop_stack(stack3)
|
||
|
|
elif choice == "7":
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
print("Invalid choice. Please try again.")
|