load.py 728 B

123456789101112131415161718192021222324252627
  1. from pathlib import Path
  2. import yaml
  3. __all__ = [
  4. 'mongo_conf', 'redis_conf', 'oss_conf',
  5. 'constants',
  6. 'headers', 'jy_proxy'
  7. ]
  8. base_path = Path(__file__).parent
  9. yaml_conf = (base_path / 'conf.yaml').resolve()
  10. yaml_constants = (base_path / 'constants.yaml').resolve()
  11. yaml_thesaurus = (base_path / 'words.yaml').resolve()
  12. yaml_areas = (base_path / 'areas.yaml').resolve()
  13. with open(yaml_conf, encoding="utf-8") as f:
  14. conf = yaml.safe_load(f)
  15. mongo_conf = conf['mongo']
  16. redis_conf = conf['redis']
  17. oss_conf: dict = conf['ali_oss']
  18. with open(yaml_constants, encoding="utf-8") as fp:
  19. constants = yaml.safe_load(fp)
  20. headers: dict = constants['headers']
  21. jy_proxy: dict = constants['proxy']