#!/usr/bin/env python3
import argparse
import shlex
import subprocess
def run_process(process_name):
"""
Run the given process.
"""
return subprocess.run(shlex.split(process_name))
if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog="ssh_server_control",
description="Start, stop, or show the status of the SSH server.",
)
_ = parser.add_argument(
"--start", action="store_true", default=False, help="Start the SSH server"
)
_ = parser.add_argument(
"--stop", action="store_true", default=False, help="Stop the SSH server"
)
_ = parser.add_argument(
"--status",
action="store_true",
default=False,
help="Show the status of the SSH server",
)
args = parser.parse_args()
if args.start:
run_process("sudo systemctl start ssh")
if args.stop:
run_process("sudo systemctl stop ssh")
if args.status:
run_process("sudo systemctl status ssh")