main.py 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. import os
  2. import sys
  3. import schedule
  4. import time
  5. sys.path.append(os.getcwd())
  6. from common import Common, Feishu
  7. from video_capture.kuaishou.kuaishou_author.kuaishou_author import kuaishouAuthor
  8. from common.sql_help import sql
  9. from video_stitching.video_stitching import VideoStitching
  10. from video_capture.douyin.douyin_author.douyin_author import douyinAuthor
  11. flag = True
  12. def job_video_stitching():
  13. global flag
  14. dy_yinmei_count = sql.get_dy_yinmei_account_id()
  15. koubo_count = sql.get_koubo_account_id()
  16. ks_yinmei_count = sql.get_ks_yinmei_account_id()
  17. if int(ks_yinmei_count) < 10:
  18. Common.logger().info("开始执行")
  19. video_type = "音画美文"
  20. channel_type = "kuaishou"
  21. VideoStitching.video_stitching(video_type, ks_yinmei_count, channel_type)
  22. elif int(dy_yinmei_count) < 10:
  23. Common.logger().info("开始执行")
  24. video_type = "音画美文"
  25. channel_type = "douyin"
  26. VideoStitching.video_stitching(video_type, dy_yinmei_count, channel_type)
  27. elif int(koubo_count) < 10:
  28. Common.logger().info("开始执行")
  29. video_type = "口播--美文类"
  30. channel_type = "koubo"
  31. VideoStitching.video_stitching(video_type, koubo_count, channel_type)
  32. count = int(dy_yinmei_count + koubo_count + ks_yinmei_count)
  33. if count == 30:
  34. if flag:
  35. Feishu.bot('recommend', '拼接视频', '自制视频拼接完成啦,共计30条~')
  36. flag = False
  37. if count == 0:
  38. flag = True
  39. def job_feishu_bot():
  40. dy_yinmei_count = sql.get_dy_yinmei_account_id()
  41. koubo_count = sql.get_koubo_account_id()
  42. ks_yinmei_count = sql.get_ks_yinmei_account_id()
  43. count = int(dy_yinmei_count + koubo_count + ks_yinmei_count)
  44. if count < 30:
  45. Feishu.bot('recommend', '拼接视频', f'视频生成异常,不符合预期,请检查\n目前生成数量如下:\n抖音视频拼接:{dy_yinmei_count}条 \n快手视频拼接:{ks_yinmei_count}条\n口播视频拼接:{koubo_count}条')
  46. def job_douyin_data():
  47. douyinAuthor.get_videoList()
  48. def job_kuaishou_data():
  49. kuaishouAuthor.get_kuaishou_videoList()
  50. # 每5个小时执行一次
  51. schedule.every(5).hours.do(job_douyin_data)
  52. # 每3个小时执行一次
  53. schedule.every(5).hours.do(job_kuaishou_data)
  54. # 每15分钟执行一次
  55. schedule.every(15).minutes.do(job_video_stitching)
  56. # 每天下午1:30执行任务
  57. schedule.every().day.at("13:30").do(job_feishu_bot)
  58. while True:
  59. schedule.run_pending()
  60. time.sleep(1)