job_fj.py 1.1 KB

123456789101112131415161718192021222324252627282930313233
  1. import os
  2. import time
  3. import threading
  4. from loguru import logger
  5. from common import Material
  6. from video_rewriting.video_processor import VideoProcessor
  7. # 控制读写速度的参数
  8. MAX_BPS = 120 * 1024 * 1024 # 120MB/s
  9. MAX_WORKERS = os.cpu_count() * 2 # 线程池最大工作线程数量
  10. READ_WRITE_CHUNK_SIZE = 1024 * 1024 # 每次读写的块大小 (1MB)
  11. SLEEP_INTERVAL = READ_WRITE_CHUNK_SIZE / MAX_BPS # 控制每次读写的延迟时间
  12. # 全局锁,用于同步读写操作
  13. lock = threading.Lock()
  14. def video_task_start():
  15. data = Material.feishu_list()[1]
  16. """处理视频任务,返回用户名并根据结果决定延迟时间"""
  17. while True:
  18. try:
  19. logger.info("[+] 开始执行任务")
  20. mark = VideoProcessor.main(data)
  21. logger.info(f"[+] 返回用户名: {mark}")
  22. time.sleep(10 if mark else 120) # 根据 mark 是否为空设置延迟
  23. except Exception as e:
  24. logger.error(f"[+] 处理任务时出现异常: {e}")
  25. time.sleep(10)
  26. continue
  27. if __name__ == '__main__':
  28. video_task_start()