mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 03:57:30 +08:00
78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
from pathlib import Path
|
||
from typing import Tuple
|
||
|
||
import docker
|
||
|
||
from app.core.config import settings
|
||
from app.core.event import eventmanager, Event
|
||
from app.log import logger
|
||
from app.schemas import ConfigChangeEventData
|
||
from app.schemas.types import EventType
|
||
from app.utils.system import SystemUtils
|
||
|
||
|
||
class SystemHelper:
|
||
"""
|
||
系统工具类，提供系统相关的操作和判断
|
||
"""
|
||
|
||
@eventmanager.register(EventType.ConfigChanged)
|
||
def handle_config_changed(self, event: Event):
|
||
"""
|
||
处理配置变更事件，更新日志设置
|
||
:param event: 事件对象
|
||
"""
|
||
if not event:
|
||
return
|
||
event_data: ConfigChangeEventData = event.event_data
|
||
if event_data.key not in ['DEBUG', 'LOG_LEVEL', 'LOG_MAX_FILE_SIZE', 'LOG_BACKUP_COUNT',
|
||
'LOG_FILE_FORMAT', 'LOG_CONSOLE_FORMAT']:
|
||
return
|
||
logger.update_loggers()
|
||
|
||
@staticmethod
|
||
def can_restart() -> bool:
|
||
"""
|
||
判断是否可以内部重启
|
||
"""
|
||
return (
|
||
Path("/var/run/docker.sock").exists()
|
||
or settings.DOCKER_CLIENT_API != "tcp://127.0.0.1:38379"
|
||
)
|
||
|
||
@staticmethod
|
||
def restart() -> Tuple[bool, str]:
|
||
"""
|
||
执行Docker重启操作
|
||
"""
|
||
if not SystemUtils.is_docker():
|
||
return False, "非Docker环境，无法重启！"
|
||
try:
|
||
# 创建 Docker 客户端
|
||
client = docker.DockerClient(base_url=settings.DOCKER_CLIENT_API)
|
||
# 获取当前容器的 ID
|
||
container_id = None
|
||
with open("/proc/self/mountinfo", "r") as f:
|
||
data = f.read()
|
||
index_resolv_conf = data.find("resolv.conf")
|
||
if index_resolv_conf != -1:
|
||
index_second_slash = data.rfind("/", 0, index_resolv_conf)
|
||
index_first_slash = data.rfind("/", 0, index_second_slash) + 1
|
||
container_id = data[index_first_slash:index_second_slash]
|
||
if len(container_id) < 20:
|
||
index_resolv_conf = data.find("/sys/fs/cgroup/devices")
|
||
if index_resolv_conf != -1:
|
||
index_second_slash = data.rfind(" ", 0, index_resolv_conf)
|
||
index_first_slash = (
|
||
data.rfind("/", 0, index_second_slash) + 1
|
||
)
|
||
container_id = data[index_first_slash:index_second_slash]
|
||
if not container_id:
|
||
return False, "获取容器ID失败！"
|
||
# 重启当前容器
|
||
client.containers.get(container_id.strip()).restart()
|
||
return True, ""
|
||
except Exception as err:
|
||
print(str(err))
|
||
return False, f"重启时发生错误：{str(err)}"
|