mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-13 09:40:30 +08:00
87 lines
4.0 KiB
Python
87 lines
4.0 KiB
Python
# ========================================================================== #
|
|
# #
|
|
# KVMD - The main Pi-KVM daemon. #
|
|
# #
|
|
# Copyright (C) 2018 Maxim Devaev <mdevaev@gmail.com> #
|
|
# #
|
|
# This program is free software: you can redistribute it and/or modify #
|
|
# it under the terms of the GNU General Public License as published by #
|
|
# the Free Software Foundation, either version 3 of the License, or #
|
|
# (at your option) any later version. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, #
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
|
# GNU General Public License for more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License #
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>. #
|
|
# #
|
|
# ========================================================================== #
|
|
|
|
|
|
import os
|
|
import contextlib
|
|
|
|
from typing import Dict
|
|
|
|
import dbus # pylint: disable=import-error
|
|
import dbus.exceptions
|
|
|
|
from ...logging import get_logger
|
|
|
|
from ...yamlconf.loader import load_yaml_file
|
|
|
|
from ... import aiotools
|
|
|
|
|
|
# =====
|
|
class InfoManager:
|
|
def __init__(
|
|
self,
|
|
meta_path: str,
|
|
extras_path: str,
|
|
) -> None:
|
|
|
|
self.__meta_path = meta_path
|
|
self.__extras_path = extras_path
|
|
|
|
async def get_meta(self) -> Dict:
|
|
return (await aiotools.run_async(load_yaml_file, self.__meta_path))
|
|
|
|
async def get_extras(self) -> Dict:
|
|
return (await aiotools.run_async(self.__inner_get_extras))
|
|
|
|
def __inner_get_extras(self) -> Dict:
|
|
extras: Dict[str, Dict] = {}
|
|
for app in os.listdir(self.__extras_path):
|
|
if app[0] != "." and os.path.isdir(os.path.join(self.__extras_path, app)):
|
|
extras[app] = load_yaml_file(os.path.join(self.__extras_path, app, "manifest.yaml"))
|
|
daemon = extras[app].get("daemon", "")
|
|
if isinstance(daemon, str) and daemon.strip():
|
|
extras[app]["enabled"] = self.__is_daemon_enabled(daemon)
|
|
return extras
|
|
|
|
def __is_daemon_enabled(self, name: str) -> bool:
|
|
if not name.startswith(".service"):
|
|
name += ".service"
|
|
|
|
try:
|
|
with contextlib.closing(dbus.SystemBus()) as bus:
|
|
systemd = bus.get_object("org.freedesktop.systemd1", "/org/freedesktop/systemd1") # pylint: disable=no-member
|
|
manager = dbus.Interface(systemd, dbus_interface="org.freedesktop.systemd1.Manager")
|
|
|
|
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:
|
|
get_logger(0).error("Can't get info about the service %r: %s: %s", name, type(err).__name__, str(err))
|
|
return True
|