env.py 730 B

1234567891011121314151617181920
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import warnings
  4. # ================ 路径 ====================
  5. ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
  6. CONFIG_ROOT_PATH = os.path.join(ROOT_PATH, "configs")
  7. CONFIG_TEMPLATE_FILE = os.path.join(CONFIG_ROOT_PATH, "config_template.yaml")
  8. CONFIG_FILE = os.path.join(CONFIG_ROOT_PATH, "config.yaml")
  9. if not os.path.exists(CONFIG_FILE):
  10. CONFIG_FILE = CONFIG_TEMPLATE_FILE
  11. LOG_PATH = os.path.join(ROOT_PATH, "logs")
  12. OUTPUT_PATH = os.path.join(ROOT_PATH, "outputs")
  13. WEB_PATH = os.path.join(ROOT_PATH, "web")
  14. # Create tmp folder
  15. if not os.path.exists(OUTPUT_PATH):
  16. os.makedirs(OUTPUT_PATH)
  17. warnings.warn(f"Create output path: {OUTPUT_PATH}")