GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2026-01-29 09:01:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

handle disabled services

Browse Source
This commit is contained in:
Devaev Maxim
2019-05-11 21:36:22 +03:00
parent 447e3c2d9e
commit cf2ed326a3
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
kvmd/apps/kvmd/info.py
View File

@@ -27,6 +27,7 @@ import contextlib
from typing import Dict from typing import Dict
import dbus # pylint: disable=import-error import dbus # pylint: disable=import-error
import dbus.exceptions
from ...logging import get_logger from ...logging import get_logger
@@ -63,11 +64,22 @@ class InfoManager:
def __is_daemon_enabled(self, name: str) -> bool: def __is_daemon_enabled(self, name: str) -> bool:
if not name.startswith(".service"): if not name.startswith(".service"):
name += ".service" name += ".service"
try: try:
with contextlib.closing(dbus.SystemBus()) as bus: with contextlib.closing(dbus.SystemBus()) as bus:
systemd = bus.get_object("org.freedesktop.systemd1", "/org/freedesktop/systemd1") # pylint: disable=no-member systemd = bus.get_object("org.freedesktop.systemd1", "/org/freedesktop/systemd1") # pylint: disable=no-member
get_unit_state = systemd.get_dbus_method("GetUnitFileState", "org.freedesktop.systemd1.Manager") manager = dbus.Interface(systemd, dbus_interface="org.freedesktop.systemd1.Manager")
return (get_unit_state(name) in ["enabled", "enabled-runtime", "static", "indirect", "generated"])
try:
unit_proxy = bus.get_object("org.freedesktop.systemd1", manager.GetUnit(name)) # pylint: disable=no-member
unit_properties = dbus.Interface(unit_proxy, dbus_interface="org.freedesktop.DBus.Properties")
enabled = (unit_properties.Get("org.freedesktop.systemd1.Unit", "ActiveState") == "active")
except dbus.exceptions.DBusException as err:
if "NoSuchUnit" not in str(err):
raise
enabled = False
return (enabled or (manager.GetUnitFileState(name) in ["enabled", "enabled-runtime", "static", "indirect", "generated"]))
except Exception as err: except Exception as err:
get_logger(0).error("Can't get info about the service %r: %s: %s", name, type(err).__name__, str(err)) get_logger(0).error("Can't get info about the service %r: %s: %s", name, type(err).__name__, str(err))
return True return True