carry_data_redis.py 918 B

1234567891011121314151617181920212223242526272829303132333435
  1. import datetime
  2. import os
  3. import schedule
  4. from loguru import logger
  5. from common.redis import insert_carry_data
  6. ENV = os.getenv('ENV', 'dev')
  7. FS_SHEET = os.getenv('FS_SHEET')
  8. NAME = os.getenv('NAME')
  9. REDIS_NAME = os.getenv('REDIS_NAME')
  10. def bot_carry_data():
  11. try:
  12. dt = int(datetime.datetime.now().strftime('%Y%m%d%H'))
  13. logger.info(f"[+] 开始获取{NAME},时区为{dt}")
  14. count = insert_carry_data(dt, REDIS_NAME,FS_SHEET, NAME)
  15. logger.info(f"[+] {NAME},时区为{dt}共获取{count}条")
  16. except Exception as e:
  17. logger.error(f"[+] 获取{NAME},时区为{dt}失败,失败信息{e}")
  18. def schedule_tasks():
  19. schedule.every().hour.at(":14").do(bot_carry_data)
  20. if __name__ == "__main__":
  21. # schedule_tasks() # 调用任务调度函数
  22. # while True:
  23. # schedule.run_pending()
  24. # time.sleep(1) # 每秒钟检查一次
  25. bot_carry_data()