52 lines
1.1 KiB
Python
52 lines
1.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import sys, os, subprocess
|
||
|
|
|
||
|
|
help_text = f"""
|
||
|
|
Usage: {os.path.basename(__file__)} $1 $2
|
||
|
|
|
||
|
|
required:
|
||
|
|
$1 - name of service
|
||
|
|
$2 - calendar ()
|
||
|
|
"""
|
||
|
|
|
||
|
|
number_of_required_arguments = 2
|
||
|
|
number_of_optional_arguments = 0
|
||
|
|
|
||
|
|
if len(sys.argv) == 1:
|
||
|
|
print(help_text)
|
||
|
|
sys.exit()
|
||
|
|
elif sys.argv[1] in ["help", "-h"]:
|
||
|
|
print(help_text)
|
||
|
|
elif len(sys.argv) - 1 < number_of_required_arguments:
|
||
|
|
print(f"You provided not enough arguments")
|
||
|
|
print(help_text)
|
||
|
|
sys.exit(1)
|
||
|
|
elif len(sys.argv) > number_of_required_arguments + number_of_optional_arguments + 1:
|
||
|
|
print(f"You provided extra arguments")
|
||
|
|
print(help_text)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
name_of_service = sys.argv[1]
|
||
|
|
calendar = sys.argv[2]
|
||
|
|
|
||
|
|
timer_file = f"""
|
||
|
|
[Unit]
|
||
|
|
Description={name_of_service} timer
|
||
|
|
|
||
|
|
[Timer]
|
||
|
|
Unit={name_of_service}.service
|
||
|
|
OnCalendar={calendar}
|
||
|
|
|
||
|
|
[Install]
|
||
|
|
WantedBy=timers.target
|
||
|
|
"""
|
||
|
|
|
||
|
|
try:
|
||
|
|
with open(f"/lib/systemd/system/{name_of_service}.timer", "w") as f:
|
||
|
|
f.write(timer_file)
|
||
|
|
os.system(f"systemctl enable --now {name_of_service}.timer && echo Success!!!")
|
||
|
|
except:
|
||
|
|
print("Something went wrong...")
|
||
|
|
sys.exit(1)
|
||
|
|
|