Austin/QChatGPT
1
0
Fork 0
mirror of https://github.com/RockChinQ/QChatGPT.git synced 2024-11-15 19:22:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

perf: 支持识别docker环境

Browse Source
This commit is contained in:
RockChinQ 2024-03-07 15:55:02 +08:00
parent 2471c5bf0f
commit 95784debbf
3 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Dockerfile
View File

@ -5,6 +5,7 @@ COPY . .
RUN apt update \ RUN apt update \
&& apt install gcc -y \ && apt install gcc -y \
&& python -m pip install -r requirements.txt && python -m pip install -r requirements.txt \
&& touch /.dockerenv
CMD [ "python", "main.py" ] CMD [ "python", "main.py" ]

4
pkg/core/stages/build_app.py
View File

@ -3,7 +3,7 @@ from __future__ import annotations
import sys import sys
from .. import stage, app from .. import stage, app
from ...utils import version, proxy, announce from ...utils import version, proxy, announce, platform
from ...audit.center import v2 as center_v2 from ...audit.center import v2 as center_v2
from ...audit import identifier from ...audit import identifier
from ...pipeline import pool, controller, stagemgr from ...pipeline import pool, controller, stagemgr
@ -39,7 +39,7 @@ class BuildAppStage(stage.BootingStage):
"host_id": identifier.identifier["host_id"], "host_id": identifier.identifier["host_id"],
"instance_id": identifier.identifier["instance_id"], "instance_id": identifier.identifier["instance_id"],
"semantic_version": ver_mgr.get_current_version(), "semantic_version": ver_mgr.get_current_version(),
"platform": sys.platform, "platform": platform.get_platform(),
}, },
runtime_info={ runtime_info={
"admin_id": "{}".format(ap.system_cfg.data["admin-sessions"]), "admin_id": "{}".format(ap.system_cfg.data["admin-sessions"]),

4
pkg/utils/platform.py
View File

@ -4,4 +4,8 @@ import sys
def get_platform() -> str: def get_platform() -> str:
"""获取当前平台""" """获取当前平台"""
# 检查是不是在 docker 里
if os.path.exists('/.dockerenv'):
return 'docker'
return sys.platform return sys.platform