data_main.py 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  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("kuaishou")
  15. kuaishou_start(data[5])
  16. # # 定义定时任务
  17. # def douyin_task():
  18. # data = Material.get_all_user("douyin")
  19. # # 创建一个线程池
  20. # with concurrent.futures.ThreadPoolExecutor() as executor:
  21. # futures = [executor.submit(douyin_start, user_data) for user_data in data]
  22. # # 等待所有任务执行完成
  23. # for future in concurrent.futures.as_completed(futures):
  24. # # 获取每个任务的执行结果
  25. # result = future.result()
  26. # print("处理结果:", result)
  27. # print("抖音数据抓取定时任务执行完成")
  28. #
  29. #
  30. # # 定义定时任务
  31. # def kuanshou_task():
  32. # data = Material.get_all_user("kuaishou")
  33. # # 创建一个线程池
  34. # with concurrent.futures.ThreadPoolExecutor() as executor:
  35. # futures = [executor.submit(kuaishou_start, user_data) for user_data in data]
  36. # # 等待所有任务执行完成
  37. # for future in concurrent.futures.as_completed(futures):
  38. # # 获取每个任务的执行结果
  39. # result = future.result()
  40. # print("处理结果:", result)
  41. # print("快手数据抓取定时任务执行完成.")
  42. #
  43. #
  44. # # schedule.every(2).minutes.do(douyin_task)
  45. # #
  46. # # schedule.every(2).minutes.do(kuanshou_task)
  47. # schedule.every(8).hours.do(douyin_task)
  48. #
  49. # schedule.every(8).hours.do(kuanshou_task)
  50. #
  51. # # 持续运行,直到手动终止
  52. # while True:
  53. # schedule.run_pending()
  54. # time.sleep(1)
  55. #
  56. #