mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 11:42:44 +08:00
fix: 循环引用导致的依赖自动解决失败
This commit is contained in:
parent
7812e03c9d
commit
0bca7b2247
62
main.py
62
main.py
|
@ -11,6 +11,8 @@ import traceback
|
||||||
|
|
||||||
sys.path.append(".")
|
sys.path.append(".")
|
||||||
|
|
||||||
|
from pkg.utils.log import init_runtime_log_file, reset_logging
|
||||||
|
|
||||||
try:
|
try:
|
||||||
import colorlog
|
import colorlog
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
@ -18,6 +20,7 @@ except ImportError:
|
||||||
import pkg.utils.pkgmgr as pkgmgr
|
import pkg.utils.pkgmgr as pkgmgr
|
||||||
try:
|
try:
|
||||||
pkgmgr.install_requirements("requirements.txt")
|
pkgmgr.install_requirements("requirements.txt")
|
||||||
|
pkgmgr.install_upgrade("websockets")
|
||||||
import colorlog
|
import colorlog
|
||||||
except ImportError:
|
except ImportError:
|
||||||
print("依赖不满足,请查看 https://github.com/RockChinQ/qcg-installer/issues/15")
|
print("依赖不满足,请查看 https://github.com/RockChinQ/qcg-installer/issues/15")
|
||||||
|
@ -30,15 +33,6 @@ from urllib3.exceptions import InsecureRequestWarning
|
||||||
import pkg.utils.context
|
import pkg.utils.context
|
||||||
|
|
||||||
|
|
||||||
log_colors_config = {
|
|
||||||
'DEBUG': 'green', # cyan white
|
|
||||||
'INFO': 'white',
|
|
||||||
'WARNING': 'yellow',
|
|
||||||
'ERROR': 'red',
|
|
||||||
'CRITICAL': 'cyan',
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# 是否使用override.json覆盖配置
|
# 是否使用override.json覆盖配置
|
||||||
# 仅在启动时提供 --override 或 -r 参数时生效
|
# 仅在启动时提供 --override 或 -r 参数时生效
|
||||||
use_override = False
|
use_override = False
|
||||||
|
@ -60,56 +54,6 @@ def ensure_dependencies():
|
||||||
|
|
||||||
known_exception_caught = False
|
known_exception_caught = False
|
||||||
|
|
||||||
log_file_name = "qchatgpt.log"
|
|
||||||
|
|
||||||
|
|
||||||
def init_runtime_log_file():
|
|
||||||
"""为此次运行生成日志文件
|
|
||||||
格式: qchatgpt-yyyy-MM-dd-HH-mm-ss.log
|
|
||||||
"""
|
|
||||||
global log_file_name
|
|
||||||
|
|
||||||
# 检查logs目录是否存在
|
|
||||||
if not os.path.exists("logs"):
|
|
||||||
os.mkdir("logs")
|
|
||||||
|
|
||||||
# 检查本目录是否有qchatgpt.log,若有,移动到logs目录
|
|
||||||
if os.path.exists("qchatgpt.log"):
|
|
||||||
shutil.move("qchatgpt.log", "logs/qchatgpt.legacy.log")
|
|
||||||
|
|
||||||
log_file_name = "logs/qchatgpt-%s.log" % time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime())
|
|
||||||
|
|
||||||
|
|
||||||
def reset_logging():
|
|
||||||
global log_file_name
|
|
||||||
|
|
||||||
import config
|
|
||||||
|
|
||||||
if pkg.utils.context.context['logger_handler'] is not None:
|
|
||||||
logging.getLogger().removeHandler(pkg.utils.context.context['logger_handler'])
|
|
||||||
|
|
||||||
for handler in logging.getLogger().handlers:
|
|
||||||
logging.getLogger().removeHandler(handler)
|
|
||||||
|
|
||||||
logging.basicConfig(level=config.logging_level, # 设置日志输出格式
|
|
||||||
filename=log_file_name, # log日志输出的文件位置和文件名
|
|
||||||
format="[%(asctime)s.%(msecs)03d] %(filename)s (%(lineno)d) - [%(levelname)s] : %(message)s",
|
|
||||||
# 日志输出的格式
|
|
||||||
# -8表示占位符,让输出左对齐,输出长度都为8位
|
|
||||||
datefmt="%Y-%m-%d %H:%M:%S" # 时间输出的格式
|
|
||||||
)
|
|
||||||
sh = logging.StreamHandler()
|
|
||||||
sh.setLevel(config.logging_level)
|
|
||||||
sh.setFormatter(colorlog.ColoredFormatter(
|
|
||||||
fmt="%(log_color)s[%(asctime)s.%(msecs)03d] %(filename)s (%(lineno)d) - [%(levelname)s] : "
|
|
||||||
"%(message)s",
|
|
||||||
datefmt="%Y-%m-%d %H:%M:%S",
|
|
||||||
log_colors=log_colors_config
|
|
||||||
))
|
|
||||||
logging.getLogger().addHandler(sh)
|
|
||||||
pkg.utils.context.context['logger_handler'] = sh
|
|
||||||
return sh
|
|
||||||
|
|
||||||
|
|
||||||
def override_config():
|
def override_config():
|
||||||
import config
|
import config
|
||||||
|
|
|
@ -156,8 +156,8 @@ def install_plugin(repo_url: str):
|
||||||
import pkg.utils.pkgmgr
|
import pkg.utils.pkgmgr
|
||||||
pkg.utils.pkgmgr.install_requirements("plugins/"+repo_url.split(".git")[0].split("/")[-1]+"/requirements.txt")
|
pkg.utils.pkgmgr.install_requirements("plugins/"+repo_url.split(".git")[0].split("/")[-1]+"/requirements.txt")
|
||||||
|
|
||||||
import main
|
import pkg.utils.log as log
|
||||||
main.reset_logging()
|
log.reset_logging()
|
||||||
|
|
||||||
|
|
||||||
def uninstall_plugin(plugin_name: str) -> str:
|
def uninstall_plugin(plugin_name: str) -> str:
|
||||||
|
|
|
@ -117,8 +117,8 @@ class PluginUpdateCommand(AbstractCommandNode):
|
||||||
import pkg.utils.pkgmgr
|
import pkg.utils.pkgmgr
|
||||||
pkg.utils.pkgmgr.install_requirements("/".join(plugin['path'].split('/')[:-1])+"/requirements.txt")
|
pkg.utils.pkgmgr.install_requirements("/".join(plugin['path'].split('/')[:-1])+"/requirements.txt")
|
||||||
|
|
||||||
import main
|
import pkg.utils.log as log
|
||||||
main.reset_logging()
|
log.reset_logging()
|
||||||
|
|
||||||
pkg.utils.context.get_qqbot_manager().notify_admin("已更新插件: {}".format(", ".join(updated)))
|
pkg.utils.context.get_qqbot_manager().notify_admin("已更新插件: {}".format(", ".join(updated)))
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|
66
pkg/utils/log.py
Normal file
66
pkg/utils/log.py
Normal file
|
@ -0,0 +1,66 @@
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
import logging
|
||||||
|
import shutil
|
||||||
|
|
||||||
|
|
||||||
|
log_file_name = "qchatgpt.log"
|
||||||
|
|
||||||
|
|
||||||
|
log_colors_config = {
|
||||||
|
'DEBUG': 'green', # cyan white
|
||||||
|
'INFO': 'white',
|
||||||
|
'WARNING': 'yellow',
|
||||||
|
'ERROR': 'red',
|
||||||
|
'CRITICAL': 'cyan',
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def init_runtime_log_file():
|
||||||
|
"""为此次运行生成日志文件
|
||||||
|
格式: qchatgpt-yyyy-MM-dd-HH-mm-ss.log
|
||||||
|
"""
|
||||||
|
global log_file_name
|
||||||
|
|
||||||
|
# 检查logs目录是否存在
|
||||||
|
if not os.path.exists("logs"):
|
||||||
|
os.mkdir("logs")
|
||||||
|
|
||||||
|
# 检查本目录是否有qchatgpt.log,若有,移动到logs目录
|
||||||
|
if os.path.exists("qchatgpt.log"):
|
||||||
|
shutil.move("qchatgpt.log", "logs/qchatgpt.legacy.log")
|
||||||
|
|
||||||
|
log_file_name = "logs/qchatgpt-%s.log" % time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime())
|
||||||
|
|
||||||
|
|
||||||
|
def reset_logging():
|
||||||
|
global log_file_name
|
||||||
|
|
||||||
|
import config
|
||||||
|
import pkg.utils.context
|
||||||
|
import colorlog
|
||||||
|
|
||||||
|
if pkg.utils.context.context['logger_handler'] is not None:
|
||||||
|
logging.getLogger().removeHandler(pkg.utils.context.context['logger_handler'])
|
||||||
|
|
||||||
|
for handler in logging.getLogger().handlers:
|
||||||
|
logging.getLogger().removeHandler(handler)
|
||||||
|
|
||||||
|
logging.basicConfig(level=config.logging_level, # 设置日志输出格式
|
||||||
|
filename=log_file_name, # log日志输出的文件位置和文件名
|
||||||
|
format="[%(asctime)s.%(msecs)03d] %(filename)s (%(lineno)d) - [%(levelname)s] : %(message)s",
|
||||||
|
# 日志输出的格式
|
||||||
|
# -8表示占位符,让输出左对齐,输出长度都为8位
|
||||||
|
datefmt="%Y-%m-%d %H:%M:%S" # 时间输出的格式
|
||||||
|
)
|
||||||
|
sh = logging.StreamHandler()
|
||||||
|
sh.setLevel(config.logging_level)
|
||||||
|
sh.setFormatter(colorlog.ColoredFormatter(
|
||||||
|
fmt="%(log_color)s[%(asctime)s.%(msecs)03d] %(filename)s (%(lineno)d) - [%(levelname)s] : "
|
||||||
|
"%(message)s",
|
||||||
|
datefmt="%Y-%m-%d %H:%M:%S",
|
||||||
|
log_colors=log_colors_config
|
||||||
|
))
|
||||||
|
logging.getLogger().addHandler(sh)
|
||||||
|
pkg.utils.context.context['logger_handler'] = sh
|
||||||
|
return sh
|
|
@ -1,21 +1,25 @@
|
||||||
from pip._internal import main as pipmain
|
from pip._internal import main as pipmain
|
||||||
|
|
||||||
import main
|
import pkg.utils.log as log
|
||||||
|
|
||||||
|
|
||||||
def install(package):
|
def install(package):
|
||||||
pipmain(['install', package])
|
pipmain(['install', package])
|
||||||
main.reset_logging()
|
log.reset_logging()
|
||||||
|
|
||||||
|
def install_upgrade(package):
|
||||||
|
pipmain(['install', '--upgrade', package])
|
||||||
|
log.reset_logging()
|
||||||
|
|
||||||
|
|
||||||
def run_pip(params: list):
|
def run_pip(params: list):
|
||||||
pipmain(params)
|
pipmain(params)
|
||||||
main.reset_logging()
|
log.reset_logging()
|
||||||
|
|
||||||
|
|
||||||
def install_requirements(file):
|
def install_requirements(file):
|
||||||
pipmain(['install', '-r', file, "--upgrade"])
|
pipmain(['install', '-r', file, "--upgrade"])
|
||||||
main.reset_logging()
|
log.reset_logging()
|
||||||
|
|
||||||
|
|
||||||
def ensure_dulwich():
|
def ensure_dulwich():
|
||||||
|
|
Loading…
Reference in New Issue
Block a user