carry_data_redis.py 931 B

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