136 lines
4.4 KiB
Python
Executable file
136 lines
4.4 KiB
Python
Executable file
#! /usr/bin/python3
|
|
|
|
import os
|
|
import subprocess
|
|
import yaml
|
|
import argparse
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
prog="podman-tool",
|
|
description="Utility for managing podman containers",
|
|
epilog=""" commands: create: create services specified in ptconfig.yml without starting;
|
|
start: start one specific service or all services at once;
|
|
stop: stop one specific service or all services at once;
|
|
shell: run a shell in an existing service;
|
|
karaf: run a karaf shell in an existing service (only useful for openHAB);
|
|
down: stop all services and delete their pods""",
|
|
)
|
|
|
|
if "REPO" in os.environ:
|
|
repo = os.environ["REPO"] + "/"
|
|
print(
|
|
f"Your current repository path is {repo}. To change it, change the REPO environment variable."
|
|
)
|
|
else:
|
|
repo = ""
|
|
print(
|
|
"The tool will use the config file from the directory it is running in. If you want to change this,"
|
|
"set the REPO environment variable."
|
|
)
|
|
|
|
config = repo + "ptconfig.yml"
|
|
with open(config, "r") as ptcfg:
|
|
read_config = yaml.safe_load(ptcfg)
|
|
|
|
compose_config = repo + "podman-compose.yml"
|
|
|
|
services = read_config["services"]
|
|
|
|
parser.add_argument(
|
|
"command", help="command to run (create, start, stop, shell, karaf, down)"
|
|
)
|
|
parser.add_argument(
|
|
"service",
|
|
help="name of service if applicable, or leave empty for all services",
|
|
nargs="?",
|
|
default="",
|
|
)
|
|
parser.add_argument(
|
|
"-y", action="store_true", help="automatically confirm any queries"
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
def create():
|
|
subprocess.run(["rm", "-rf", compose_config])
|
|
invalid_options = ["addons", "bindings", "commands"]
|
|
with open(config) as oldfile, open(compose_config, "w") as newfile:
|
|
for line in oldfile:
|
|
if not any(
|
|
invalid_option in line for invalid_option in invalid_options
|
|
):
|
|
newfile.write(line)
|
|
subprocess.run(["podman-compose", "-f", compose_config, "up", "--no-start"])
|
|
for service in list(read_config["services"]):
|
|
for index in list(
|
|
read_config["services"][service]["addons"]["bindings"]["commands"]
|
|
):
|
|
subprocess.run([index])
|
|
|
|
def start():
|
|
start_service = args.service
|
|
if start_service == "":
|
|
start_service = list(services)
|
|
for i in range(len(start_service)):
|
|
subprocess.run(["podman-compose", "start", start_service[i]])
|
|
exit(0)
|
|
else:
|
|
subprocess.run(["podman-compose", "start", start_service])
|
|
exit(0)
|
|
|
|
def stop():
|
|
stop_service = args.service
|
|
if stop_service == "":
|
|
if args.y:
|
|
confirm = "y"
|
|
else:
|
|
confirm = input("Stop all services? (y/N): ")
|
|
if confirm.lower() == "y" or confirm.lower() == "yes":
|
|
stop_service = list(services)
|
|
for i in range(len(stop_service)):
|
|
subprocess.run(["podman-compose", "stop", stop_service[i]])
|
|
exit(0)
|
|
else:
|
|
print("aborting...")
|
|
exit(0)
|
|
else:
|
|
subprocess.run(["podman-compose", "stop", stop_service])
|
|
exit(0)
|
|
|
|
def shell():
|
|
subprocess.run(["podman-compose", "exec", args.service, "sh"])
|
|
|
|
def karaf():
|
|
subprocess.run(
|
|
["podman-compose", "exec", args.service, "/openhab/runtime/bin/client"]
|
|
)
|
|
|
|
def down():
|
|
if args.y:
|
|
confirm = "y"
|
|
else:
|
|
confirm = input("Stop all services and delete all pods? (y/N): ")
|
|
if confirm.lower() == "y" or confirm.lower() == "yes":
|
|
subprocess.run(["podman-compose", "-f", compose_config, "down"])
|
|
exit(0)
|
|
else:
|
|
print("aborting...")
|
|
exit(0)
|
|
|
|
if args.command == "create":
|
|
create()
|
|
elif args.command == "start":
|
|
start()
|
|
elif args.command == "stop":
|
|
stop()
|
|
elif args.command == "shell":
|
|
shell()
|
|
elif args.command == "karaf":
|
|
karaf()
|
|
elif args.command == "down":
|
|
down()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|