Programando el apagado automático en Debian Trixie con KDE

##############################################################################
# Licensed under the GNU GPL v3 (https://www.gnu.org/licenses/gpl-3.0.html)
# Filename: auto_shutdown.py
# Version: 1.0
# Author: Jordi Fàbregas
# Contributors:
##############################################################################

'''
**WARNING**
This script is designed to run on Debian KDE workstations. It may work on other
Linux systems with systemd, but it has not been tested and is NOT RECOMMENDED 
FOR SERVERS OR MULTI-USER SYSTEMS

This script is designed to be executed with cron or systemd to automatically 
shut down the system after working hours.
Alternatively, it can be executed at startup to check if it is a working
weekday or working month. This is designed to troubleshoot some BIOS systems 
that can’t define a calendar schedule for automatic power on, by scheduling an 
automatic power off.
'''

import sys
import datetime
import subprocess

WORKING_DAYS = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday']
CLOSED_MONTHS = ['July', 'August']
SHUTDOWN_AFTER_TIME = "14:10"
KDE_PROMPT = True        # Prompt with kdialog to abort before a timeout
KDE_PROMPT_TIMEOUT = 30  # Time in seconds to abort automatic shutdown


# Get current weekday, time and month
def get_current_time():
    '''
    Get the current weekday, time and month.
    Input: (None)
    Output: Tuple with:
        - String: Weekday name (e.g. 'Monday')
        - datetime.time: Current time
        - String: Month name (e.g. 'January')
    '''
    now = datetime.datetime.now()
    day = now.strftime("%A")               # Monday, Tuesday, etc.
    current_time = now.time()              # 0-23:59:59
    month = now.strftime("%B")             # January, February, etc.

    return day, current_time, month


def shutdown_system():
    '''
    Immediately powers off the system.
    Input: (None)
    Output: (None)
    Side effects: Calls 'systemctl poweroff' to shut down the system.
    '''
    try:
        print("System shutdown...", file=sys.stdout)
        subprocess.run(["systemctl", "poweroff"], check=True)
    except Exception as e:
        print(f"Error: {e}", file=sys.stderr)


def prompt_shutdown_kde():
    '''
    Shows a kdialog prompt to confirm shutdown.
    If the user cancels, aborts shutdown.
    If the user does not respond within timeout, proceeds to shutdown.
    '''
    print("Out of working hours. Displaying shutdown prompt...")

    try:
        result = subprocess.run(
            [
                "kdialog",
                "--yesno",
                f"El sistema s'apagarà en {KDE_PROMPT_TIMEOUT} segons.\n\n"
                "Desitja cancel·lar?"
            ],
            timeout=KDE_PROMPT_TIMEOUT
        )

        if result.returncode == 0:
            print("Shutdown canceled by user.", file=sys.stdout)
        else:
            print(
                f"Scheduling shutdown in {KDE_PROMPT_TIMEOUT} seconds...",
                file=sys.stdout
            )
            shutdown_system()

    except subprocess.TimeoutExpired:
        print("User did not respond. Shutdown...", file=sys.stdout)
        shutdown_system()

    except Exception as e:
        print(f"ERROR: {e}", file=sys.stderr)


def main():
    '''
    Checks if the conditions to shut down the system are met and executes 
    the shutdown if appropriate.
    Input: (None)
    Output: (None)
    Side effects:
        - Calls prompt_shutdown_kde() to display a shutdown prompt 
          if KDE_PROMPT is True.
        - Calls shutdown_system() to power off the system if the conditions 
          are met.
    '''
    # Convert shutdown time string to datetime.time
    shutdown_hour = datetime.datetime.strptime(
        SHUTDOWN_AFTER_TIME, "%H:%M"
    ).time()

    # Get current time data
    day, current_time, month = get_current_time()

    print(f"Weekday: {day}")
    print(f"Current time: {current_time}")
    print(f"Month: {month}")

    # Check if shutdown conditions are met
    should_shutdown = (
        month in CLOSED_MONTHS or
        day not in WORKING_DAYS or
        current_time >= shutdown_hour
    )

    if should_shutdown:
        if KDE_PROMPT:
            prompt_shutdown_kde()
        else:
            shutdown_system()
    else:
        print(
            "Within working hours, no shutdown scheduled.",
            file=sys.stdout
        )


if __name__ == '__main__':
    main()

# Recarga los unit files del usuario
systemctl --user daemon-reload
# Habilita el timer al inicio de sesión
systemctl --user enable auto-shutdown.timer
# Arranca el timer ahora
systemctl --user start auto-shutdown.timer