agc_main.py 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. from common import Feishu, Material
  2. from common.sql_help import sqlHelp
  3. from video_agc.agc_video_method import AgcVidoe
  4. import concurrent.futures
  5. import schedule
  6. import time
  7. # 记录今天已经返回的用户名
  8. returned_usernames_today = set()
  9. def video_start(user_data):
  10. global returned_usernames_today
  11. user_data_mark = user_data["mark"]
  12. # 开始准备执行生成视频脚本
  13. if user_data_mark is not None and user_data_mark in returned_usernames_today:
  14. print(f"视频脚本参数中的用户名 {user_data_mark} 今天已经返回过,不再启动线程。")
  15. return # 如果返回了某个用户名,并且今天已经返回过,则不启动线程
  16. elif user_data_mark is not None:
  17. print(f"视频脚本参数{user_data}")
  18. mark = AgcVidoe.video_stitching(user_data)
  19. if mark:
  20. returned_usernames_today.add(mark)
  21. data = Material.feishu_list()
  22. video_start(data[0])
  23. #
  24. # # 定义定时任务
  25. # def video_task():
  26. # print("开始执行生成视频脚.")
  27. # data = Material.feishu_list()
  28. # # 创建一个线程池
  29. # with concurrent.futures.ThreadPoolExecutor() as executor:
  30. # futures = [executor.submit(video_start, user_data) for user_data in data]
  31. # # 等待所有任务执行完成
  32. # for future in concurrent.futures.as_completed(futures):
  33. # # 获取每个任务的执行结果
  34. # result = future.result()
  35. # print("处理结果:", result)
  36. # print("执行生成视频脚结束")
  37. # schedule.every(10).minutes.do(video_task)
  38. #
  39. # # 每天0点清空集合
  40. # schedule.every().day.at("00:00").do(lambda: returned_usernames_today.clear())
  41. #
  42. #
  43. # def job_feishu_bot():
  44. # name_list = Material.feishu_name()
  45. # count_list = sqlHelp.get_count_list(name_list)
  46. # Feishu.bot('recommend', 'AGC视频', f'{"".join(count_list)}', 'all')
  47. # print("机器人通知完成")
  48. #
  49. #
  50. # # 每天下午1:30执行任务
  51. # schedule.every().day.at("15:00").do(job_feishu_bot)
  52. #
  53. # while True:
  54. # schedule.run_pending()
  55. # time.sleep(1)
  56. #
  57. # # list = Material.feishu_list()
  58. # # AgcVidoe.video_stitching(list)
  59. # # print(list)
  60. #