mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2024-11-29 16:01:04 +01:00
91 lines
4.0 KiB
Python
91 lines
4.0 KiB
Python
import argparse
|
|
import subprocess
|
|
import time
|
|
import os
|
|
|
|
def service_file_exists(service_name, service_type="service"):
|
|
"""Check if a systemd service file exists."""
|
|
# Paths where service files can be stored
|
|
path = "/etc/systemd/system/"
|
|
service_file_name = service_name + "." + service_type
|
|
full_path = os.path.join(path, service_file_name)
|
|
|
|
print(f"Checking {full_path}") # Added debug output
|
|
if os.path.isfile(full_path):
|
|
return True
|
|
else:
|
|
print(f"File not found.") # Debug output if file is not found
|
|
|
|
def check_service_active(service_name):
|
|
"""Check if a service is active or activating."""
|
|
result = subprocess.run(['systemctl', 'is-active', service_name], stdout=subprocess.PIPE)
|
|
service_status = result.stdout.decode('utf-8').strip()
|
|
return service_status in ['active', 'activating']
|
|
|
|
def freeze(services_to_wait_for, ignored_services, max_attempts):
|
|
# Filter services that exist and are not in the ignored list
|
|
for service in services_to_wait_for:
|
|
print(f"\nFreezing: {service}")
|
|
if service in ignored_services:
|
|
print(f"{service} will be ignored.")
|
|
else:
|
|
attempt=0
|
|
break_time_sec=5
|
|
while check_service_active(service):
|
|
attempt += 1
|
|
print(f"({attempt}/{max_attempts}) Waiting for {break_time_sec} seconds for {service} to stop...")
|
|
time.sleep(break_time_sec)
|
|
if attempt > max_attempts:
|
|
raise Exception(f"Error: Maximum attempts ({max_attempts}) reached. Exit.")
|
|
|
|
# Stop and disable the corresponding timer, if it exists
|
|
if service_file_exists(service,"timer"):
|
|
timer_name = service + ".timer"
|
|
subprocess.run(['systemctl', 'stop', timer_name])
|
|
subprocess.run(['systemctl', 'disable', timer_name])
|
|
print(f"{timer_name} stopped and disabled.")
|
|
else:
|
|
print(f"Skipped.")
|
|
|
|
print("\nAll required services have stopped.")
|
|
|
|
def defrost(services_to_wait_for, ignored_services):
|
|
for service in services_to_wait_for:
|
|
print(f"\nUnfreezing: {service}")
|
|
if service in ignored_services:
|
|
print(f"{service} will be ignored.")
|
|
elif service_file_exists(service,"timer"):
|
|
# Start and enable the corresponding timer, if it exists
|
|
timer_name = service + ".timer"
|
|
subprocess.run(['systemctl', 'start', timer_name])
|
|
subprocess.run(['systemctl', 'enable', timer_name])
|
|
print(f"{timer_name} started and enabled.")
|
|
else:
|
|
print(f"Skipped.")
|
|
print("\nAll required services are started.")
|
|
|
|
def main(services_to_wait_for, ignored_services, action, max_attempts):
|
|
print(f"Services to wait for: {services_to_wait_for}")
|
|
print(f"Services to ignore: {ignored_services}")
|
|
if action == 'freeze':
|
|
print("Freezing services.");
|
|
freeze(services_to_wait_for, ignored_services, max_attempts)
|
|
elif action == 'defrost':
|
|
print("Unfreezing services.");
|
|
defrost(services_to_wait_for, ignored_services)
|
|
print('\nOverview:')
|
|
subprocess.run(['systemctl', 'list-timers'])
|
|
|
|
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='')
|
|
parser.add_argument('--max_attempts', type=int, default=60, help='Maximum number of attempts for freezing services')
|
|
|
|
args = parser.parse_args()
|
|
services_to_wait_for = args.services.split(',')
|
|
ignored_services = args.ignore.split(',') if args.ignore else []
|
|
max_attempts = args.max_attempts
|
|
main(services_to_wait_for, ignored_services,args.action,max_attempts)
|