mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2024-09-20 08:13:12 +02:00
72 lines
3.4 KiB
Python
72 lines
3.4 KiB
Python
|
import argparse
|
||
|
|
import subprocess
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
def check_service_active(service_name):
|
||
|
|
"""Check if a service is active."""
|
||
|
|
result = subprocess.run(['systemctl', 'is-active', service_name], stdout=subprocess.PIPE)
|
||
|
|
return result.stdout.decode('utf-8').strip() == 'active'
|
||
|
|
|
||
|
|
def service_exists(service_name):
|
||
|
|
"""Check if a service exists."""
|
||
|
|
result = subprocess.run(['systemctl', 'status', service_name], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||
|
|
return result.returncode == 0
|
||
|
|
|
||
|
|
def freeze(services_to_wait_for, ignored_services):
|
||
|
|
# Filter services that exist and are not in the ignored list
|
||
|
|
active_services = [service for service in services_to_wait_for if service_exists(service) and service not in ignored_services]
|
||
|
|
|
||
|
|
while active_services:
|
||
|
|
for service in active_services:
|
||
|
|
if not check_service_active(service):
|
||
|
|
print(f"{service} stopped.")
|
||
|
|
# Disable the service
|
||
|
|
subprocess.run(['systemctl', 'disable', service])
|
||
|
|
print(f"{service} disabled.")
|
||
|
|
|
||
|
|
# Stop and disable the corresponding timer, if it exists
|
||
|
|
timer_name = service + ".timer"
|
||
|
|
timer_check = subprocess.run(['systemctl', 'list-timers', '--all', timer_name], stdout=subprocess.PIPE)
|
||
|
|
if timer_name in timer_check.stdout.decode():
|
||
|
|
subprocess.run(['systemctl', 'stop', timer_name])
|
||
|
|
subprocess.run(['systemctl', 'disable', timer_name])
|
||
|
|
print(f"{timer_name} stopped and disabled.")
|
||
|
|
active_services.remove(service)
|
||
|
|
else:
|
||
|
|
print(f"Waiting for {service} to stop...")
|
||
|
|
time.sleep(5)
|
||
|
|
print("All required services have stopped.")
|
||
|
|
|
||
|
|
def defrost(services_to_wait_for, ignored_services):
|
||
|
|
for service in services_to_wait_for:
|
||
|
|
if service not in ignored_services and service_exists(service):
|
||
|
|
# Enable the service
|
||
|
|
subprocess.run(['systemctl', 'enable', service])
|
||
|
|
print(f"{service} enabled.")
|
||
|
|
|
||
|
|
# Start and enable the corresponding timer, if it exists
|
||
|
|
timer_name = service + ".timer"
|
||
|
|
timer_check = subprocess.run(['systemctl', 'list-timers', '--all', timer_name], stdout=subprocess.PIPE)
|
||
|
|
if timer_name in timer_check.stdout.decode():
|
||
|
|
subprocess.run(['systemctl', 'start', timer_name])
|
||
|
|
subprocess.run(['systemctl', 'enable', timer_name])
|
||
|
|
print(f"{timer_name} started and enabled.")
|
||
|
|
|
||
|
|
def main(services_to_wait_for, ignored_services, action):
|
||
|
|
if action == 'freeze':
|
||
|
|
# Code to handle freeze action
|
||
|
|
freeze(services_to_wait_for, ignored_services)
|
||
|
|
elif action == 'defrost':
|
||
|
|
defrost(services_to_wait_for, ignored_services)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
parser = argparse.ArgumentParser(description='freezes and defrost systemctl services and timers')
|
||
|
|
parser.add_argument('action', choices=['freeze', 'defrost'], help='Action to perform: freeze or defrost services.')
|
||
|
|
parser.add_argument('services', help='Comma-separated list of services to apply the action to')
|
||
|
|
parser.add_argument('--ignore', help='Comma-separated list of services to ignore in the action', default='')
|
||
|
|
args = parser.parse_args()
|
||
|
|
services_to_wait_for = args.services.split(',')
|
||
|
|
ignored_services = args.ignore.split(',') if args.ignore else []
|
||
|
|
main(services_to_wait_for, ignored_services,args.action)
|