71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
|
import subprocess
|
||
|
|
import logging
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
class ServiceManager:
|
||
|
|
def __init__(self, service_name='openvpn'):
|
||
|
|
self.service_name = service_name
|
||
|
|
self.init_system = self._detect_init_system()
|
||
|
|
|
||
|
|
def _detect_init_system(self):
|
||
|
|
"""Detect if systemd or openrc is used."""
|
||
|
|
if shutil.which('systemctl'):
|
||
|
|
return 'systemd'
|
||
|
|
elif shutil.which('rc-service'):
|
||
|
|
return 'openrc'
|
||
|
|
else:
|
||
|
|
return 'unknown'
|
||
|
|
|
||
|
|
def _run_cmd(self, cmd):
|
||
|
|
try:
|
||
|
|
subprocess.run(cmd, check=True, capture_output=True, text=True)
|
||
|
|
return True, "Success"
|
||
|
|
except subprocess.CalledProcessError as e:
|
||
|
|
return False, e.stderr.strip()
|
||
|
|
except Exception as e:
|
||
|
|
return False, str(e)
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
if self.init_system == 'systemd':
|
||
|
|
return self._run_cmd(['sudo', 'systemctl', 'start', self.service_name])
|
||
|
|
elif self.init_system == 'openrc':
|
||
|
|
return self._run_cmd(['sudo', 'rc-service', self.service_name, 'start'])
|
||
|
|
return False, "Unknown init system"
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
if self.init_system == 'systemd':
|
||
|
|
return self._run_cmd(['sudo', 'systemctl', 'stop', self.service_name])
|
||
|
|
elif self.init_system == 'openrc':
|
||
|
|
return self._run_cmd(['sudo', 'rc-service', self.service_name, 'stop'])
|
||
|
|
return False, "Unknown init system"
|
||
|
|
|
||
|
|
def restart(self):
|
||
|
|
if self.init_system == 'systemd':
|
||
|
|
return self._run_cmd(['sudo', 'systemctl', 'restart', self.service_name])
|
||
|
|
elif self.init_system == 'openrc':
|
||
|
|
return self._run_cmd(['sudo', 'rc-service', self.service_name, 'restart'])
|
||
|
|
return False, "Unknown init system"
|
||
|
|
|
||
|
|
def get_status(self):
|
||
|
|
"""Return 'active', 'inactive', or 'error'"""
|
||
|
|
if self.init_system == 'systemd':
|
||
|
|
# systemctl is-active returns 0 if active, non-zero otherwise
|
||
|
|
try:
|
||
|
|
subprocess.run(['systemctl', 'is-active', self.service_name], check=True, capture_output=True)
|
||
|
|
return 'active'
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
return 'inactive'
|
||
|
|
|
||
|
|
elif self.init_system == 'openrc':
|
||
|
|
try:
|
||
|
|
res = subprocess.run(['rc-service', self.service_name, 'status'], capture_output=True, text=True)
|
||
|
|
if 'started' in res.stdout or 'running' in res.stdout:
|
||
|
|
return 'active'
|
||
|
|
return 'inactive'
|
||
|
|
except:
|
||
|
|
return 'error'
|
||
|
|
|
||
|
|
return 'unknown'
|