pq_video_reids.py 1013 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. for i in range(3):
  13. try:
  14. dt = int(datetime.datetime.now().strftime('%Y%m%d%H'))
  15. logger.info(f"[+] 开始获取{NAME},时区为{dt}")
  16. count = insert_carry_data(dt, REDIS_NAME,FS_SHEET, NAME)
  17. logger.info(f"[+] {NAME},时区为{dt}共获取{count}条")
  18. return
  19. except Exception as e:
  20. logger.error(f"[+] 获取{NAME},时区为{dt}失败,失败信息{e}")
  21. continue
  22. def schedule_tasks():
  23. schedule.every().hour.at(":05").do(bot_carry_data)
  24. if __name__ == "__main__":
  25. schedule_tasks() # 调用任务调度函数
  26. while True:
  27. schedule.run_pending()
  28. time.sleep(1) # 每秒钟检查一次
  29. # bot_carry_data()