the-littlest-jupyterhub/tljh/systemd.py

"""
Wraps systemctl to install, uninstall, start & stop systemd services.

If we use a debian package instead, we can get rid of all this code.
"""
import subprocess
import os


def reload_daemon():
    """
    Equivalent to systemctl daemon-reload.

    Makes systemd discover new units.
    """
    subprocess.run(['systemctl', 'daemon-reload'], check=True)


def install_unit(name, unit, path='/etc/systemd/system'):
    """
    Install unit with given name
    """
    with open(os.path.join(path, name), 'w') as f:
        f.write(unit)


def uninstall_unit(name, path='/etc/systemd/system'):
    """
    Uninstall unit with given name
    """
    subprocess.run(['rm', os.path.join(path, name)], check=True)


def start_service(name):
    """
    Start service with given name.
    """
    subprocess.run(['systemctl', 'start', name], check=True)


def stop_service(name):
    """
    Start service with given name.
    """
    subprocess.run(['systemctl', 'stop', name], check=True)


def restart_service(name):
    """
    Restart service with given name.
    """
    subprocess.run(['systemctl', 'restart', name], check=True)


def enable_service(name):
    """
    Enable a service with given name.

    This most likely makes the service start on bootup
    """
    subprocess.run(['systemctl', 'enable', name], check=True)


def disable_service(name):
    """
    Enable a service with given name.

    This most likely makes the service start on bootup
    """
    subprocess.run(['systemctl', 'disable', name], check=True)


def check_service_active(name):
    """
    Check if a service is currently active (running)
    """
    try:
        subprocess.run(['systemctl', 'is-active', name], check=True)
        return True
    except subprocess.CalledProcessError:
        return False


def check_service_enabled(name):
    """
    Check if a service is enabled
    """
    try:
        subprocess.run(['systemctl', 'is-enabled', name], check=True)
        return True
    except subprocess.CalledProcessError:
        return False