load.py 589 B

123456789101112131415161718192021222324
  1. from pathlib import Path
  2. import yaml
  3. __all__ = [
  4. 'mongo_conf', 'redis_conf',
  5. 'headers', 'jy_proxy',
  6. 'crawl_sites'
  7. ]
  8. base_path = Path(__file__).parent
  9. yaml_conf = (base_path / 'conf.yaml').resolve()
  10. yaml_sites = (base_path / 'sites.yaml').resolve()
  11. with open(yaml_conf, encoding="utf-8") as f:
  12. conf = yaml.safe_load(f)
  13. mongo_conf = conf['mongo']
  14. redis_conf = conf['redis']
  15. headers: dict = conf['headers']
  16. jy_proxy: dict = conf['proxy']
  17. with open(yaml_sites, encoding="utf-8") as fp:
  18. sites = yaml.safe_load(fp)
  19. crawl_sites = sites['crawl_sites']