computer-playbook/roles/system-maintenance-service-freezer/files/system-maintenance-service-freezer.py

84 lines
3.6 KiB
Python
Raw Normal View History

import argparse
import subprocess
import time
2023-12-14 00:15:01 +01:00
import os
2023-12-14 00:15:01 +01:00
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):
2023-12-14 00:15:01 +01:00
"""Check if a service is active or activating."""
result = subprocess.run(['systemctl', 'is-active', service_name], stdout=subprocess.PIPE)
2023-12-14 00:15:01 +01:00
service_status = result.stdout.decode('utf-8').strip()
return service_status in ['active', 'activating', 'deactivating', 'reloading']
def freeze(services_to_wait_for, ignored_services):
# Filter services that exist and are not in the ignored list
2023-12-14 00:15:01 +01:00
for service in services_to_wait_for:
print(f"\nFreezing: {service}")
if service in ignored_services:
print(f"{service} will be ignored.")
else:
2023-12-14 00:53:17 +01:00
while check_service_active(service):
print(f"Waiting for 5 seconds for {service} to stop...")
time.sleep(5)
# 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.")
2023-12-14 00:15:01 +01:00
print("\nAll required services have stopped.")
def defrost(services_to_wait_for, ignored_services):
for service in services_to_wait_for:
2023-12-14 00:15:01 +01:00
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"
2023-12-14 00:15:01 +01:00
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):
2023-12-14 00:15:01 +01:00
print(f"Services to wait for: {services_to_wait_for}")
print(f"Services to ignore: {ignored_services}")
if action == 'freeze':
2023-12-14 00:15:01 +01:00
print("Freezing services.");
freeze(services_to_wait_for, ignored_services)
elif action == 'defrost':
2023-12-14 00:15:01 +01:00
print("Unfreezing services.");
defrost(services_to_wait_for, ignored_services)
2023-12-14 00:15:01 +01:00
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='')
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)