codefuse-chatbot/configs/utils.py

15 lines
432 B
Python
Raw Normal View History

import os
def is_running_in_docker():
"""
检查当前代码是否在 Docker 容器中运行
"""
# 检查是否存在 /.dockerenv 文件
if os.path.exists('/.dockerenv'):
return True
# 检查 cgroup 文件系统是否为 /docker/ 开头
if os.path.exists("/proc/1/cgroup"):
with open('/proc/1/cgroup', 'rt') as f:
return '/docker/' in f.read()
return False