path_config.py 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. """
  2. Python 路径配置模块
  3. 用于确保在不同环境(本地、Docker)中都能正确导入模块
  4. """
  5. import sys
  6. import os
  7. def setup_python_path():
  8. """设置 Python 路径"""
  9. # 获取当前文件所在目录
  10. current_dir = os.path.dirname(os.path.abspath(__file__))
  11. # 添加项目根目录和 src 目录到 Python 路径
  12. project_root = os.path.dirname(current_dir) # LQAdminPlatform
  13. src_dir = current_dir # src
  14. paths_to_add = [
  15. project_root, # LQAdminPlatform
  16. src_dir, # LQAdminPlatform/src
  17. os.path.join(src_dir, 'app'), # LQAdminPlatform/src/app
  18. ]
  19. for path in paths_to_add:
  20. if path not in sys.path:
  21. sys.path.insert(0, path)
  22. # 设置 PYTHONPATH 环境变量
  23. pythonpath = os.environ.get('PYTHONPATH', '')
  24. new_paths = [p for p in paths_to_add if p not in pythonpath.split(os.pathsep)]
  25. if new_paths:
  26. if pythonpath:
  27. os.environ['PYTHONPATH'] = os.pathsep.join([pythonpath] + new_paths)
  28. else:
  29. os.environ['PYTHONPATH'] = os.pathsep.join(new_paths)
  30. # 自动执行路径配置
  31. setup_python_path()