mirror of
https://github.com/jupyterhub/the-littlest-jupyterhub.git
synced 2025-12-18 21:54:05 +08:00
118 lines
2.3 KiB
Python
118 lines
2.3 KiB
Python
"""
|
|
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 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 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_hub_ready():
|
|
"""
|
|
Check if the hub is ready
|
|
"""
|
|
|
|
try:
|
|
last_restart = subprocess.check_output([
|
|
'systemctl',
|
|
'show',
|
|
'jupyterhub',
|
|
'-p',
|
|
'ActiveEnterTimestamp'
|
|
]).decode().strip()
|
|
|
|
last_restart = " ".join(last_restart.split(" ")[-3:-1])
|
|
|
|
out = subprocess.check_output([
|
|
'journalctl',
|
|
'-u',
|
|
'jupyterhub',
|
|
'--since',
|
|
last_restart
|
|
])
|
|
|
|
if "JupyterHub is now running at" in out.decode():
|
|
return True
|
|
except subprocess.CalledProcessError:
|
|
return False
|