data_main.py 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. from common import Material, Feishu
  2. from extract_data.douyin.douyin_author import douyinAuthor
  3. from extract_data.kuaishou.kuaishou_author import kuaishouAuthor
  4. import schedule
  5. import time
  6. import concurrent.futures
  7. # 定义读取表格的函数
  8. def douyin_start(user_data):
  9. print(f"执行抖音数据抓取{user_data}")
  10. douyinAuthor.get_videoList(user_data)
  11. def kuaishou_start(user_data):
  12. print(f"执行快手数据抓取{user_data}")
  13. kuaishouAuthor.get_kuaishou_videoList(user_data)
  14. data = Material.get_all_user("douyin")
  15. douyin_start(data[5])
  16. data = Material.get_all_user("kuaishou")
  17. kuaishou_start(data[5])
  18. # # 定义定时任务
  19. # def douyin_task():
  20. # data = Material.get_all_user("douyin")
  21. # # 创建一个线程池
  22. # with concurrent.futures.ThreadPoolExecutor() as executor:
  23. # futures = [executor.submit(douyin_start, user_data) for user_data in data]
  24. # # 等待所有任务执行完成
  25. # for future in concurrent.futures.as_completed(futures):
  26. # # 获取每个任务的执行结果
  27. # result = future.result()
  28. # print("处理结果:", result)
  29. # print("抖音数据抓取定时任务执行完成")
  30. #
  31. #
  32. # # 定义定时任务
  33. # def kuanshou_task():
  34. # data = Material.get_all_user("kuaishou")
  35. # # 创建一个线程池
  36. # with concurrent.futures.ThreadPoolExecutor() as executor:
  37. # futures = [executor.submit(kuaishou_start, user_data) for user_data in data]
  38. # # 等待所有任务执行完成
  39. # for future in concurrent.futures.as_completed(futures):
  40. # # 获取每个任务的执行结果
  41. # result = future.result()
  42. # print("处理结果:", result)
  43. # print("快手数据抓取定时任务执行完成.")
  44. #
  45. #
  46. # # schedule.every(2).minutes.do(douyin_task)
  47. # #
  48. # # schedule.every(2).minutes.do(kuanshou_task)
  49. # schedule.every(8).hours.do(douyin_task)
  50. #
  51. # schedule.every(8).hours.do(kuanshou_task)
  52. #
  53. # # 持续运行,直到手动终止
  54. # while True:
  55. # schedule.run_pending()
  56. # time.sleep(1)
  57. #
  58. #