new_contentId_task.py 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910
  1. """
  2. @author: luojunhui
  3. """
  4. import json
  5. import time
  6. from applications.config import Config, NewContentIdTaskConst
  7. from applications.log import logging
  8. from applications.functions.pqFunctions import publish_to_pq, get_pq_video_detail
  9. from applications.functions.common import shuffle_list
  10. from applications.functions.kimi import KimiServer
  11. from applications.spider import search_videos_from_web
  12. from applications.etl_function import *
  13. from applications.feishu import bot
  14. from applications.functions.aigc import record_trace_id
  15. class NewContentIdTask(object):
  16. """
  17. 不存在历史已经发布的文章的匹配流程
  18. """
  19. def __init__(self, mysql_client):
  20. self.mysql_client = mysql_client
  21. self.config = Config()
  22. self.article_match_video_table = self.config.article_match_video_table
  23. self.article_text_table = self.config.article_text_table
  24. self.article_crawler_video_table = self.config.article_crawler_video_table
  25. self.gh_id_dict = json.loads(self.config.get_config_value("testAccountLevel2"))
  26. self.account_map = json.loads(self.config.get_config_value("accountMap"))
  27. self.spider_coroutines = self.config.get_config_value("spiderCoroutines")
  28. self.const = NewContentIdTaskConst()
  29. async def get_tasks(self):
  30. """
  31. 获取 task
  32. :return:
  33. """
  34. # 处理未托管的任务
  35. await self.roll_back_unfinished_tasks(publish_flag=self.const.NEED_PUBLISH)
  36. # 处理托管任务
  37. await self.roll_back_unfinished_tasks(publish_flag=self.const.DO_NOT_NEED_PUBLISH)
  38. # 将 process_times > 3 且状态不为 4 的任务的状态修改为失败, 判断条件需要加上索引
  39. update_status_sql = f"""
  40. UPDATE
  41. {self.article_match_video_table}
  42. SET
  43. content_status = %s
  44. WHERE
  45. process_times > %s and content_status not in (%s, %s);
  46. """
  47. await self.mysql_client.async_insert(
  48. update_status_sql,
  49. params=(
  50. self.const.TASK_FAIL_STATUS,
  51. self.const.TASK_MAX_PROCESS_TIMES,
  52. self.const.TASK_ETL_COMPLETE_STATUS,
  53. self.const.TASK_PUBLISHED_STATUS
  54. )
  55. )
  56. # 获取 process_times <= 3 且 content_status = 0 的任务
  57. select_sql = f"""
  58. SELECT
  59. trace_id, content_id, flow_pool_level, gh_id, process_times, publish_flag
  60. FROM
  61. {self.article_match_video_table}
  62. WHERE
  63. content_status = {self.const.TASK_INIT_STATUS}
  64. and process_times <= {self.const.TASK_MAX_PROCESS_TIMES}
  65. ORDER BY flow_pool_level, request_timestamp
  66. LIMIT {self.spider_coroutines};
  67. """
  68. tasks = await self.mysql_client.async_select(select_sql)
  69. if tasks:
  70. return [
  71. {
  72. "trace_id": i[0],
  73. "content_id": i[1],
  74. "flow_pool_level": i[2],
  75. "gh_id": i[3],
  76. "process_times": i[4],
  77. "publish_flag": i[5]
  78. }
  79. for i in tasks
  80. ]
  81. else:
  82. return []
  83. async def roll_back_unfinished_tasks(self, publish_flag):
  84. """
  85. 将长时间处于中间状态的任务回滚
  86. """
  87. # 获取 content_status 为 处理中 的任务,判断时间, 如果超过 1h 则,则将它改为 0, process_times + 1
  88. if publish_flag == self.const.NEED_PUBLISH:
  89. processing_status_tuple = (
  90. self.const.TASK_PROCESSING_STATUS,
  91. self.const.TASK_KIMI_FINISHED_STATUS,
  92. self.const.TASK_SPIDER_FINISHED_STATUS,
  93. self.const.TASK_ETL_COMPLETE_STATUS
  94. )
  95. elif publish_flag == self.const.DO_NOT_NEED_PUBLISH:
  96. processing_status_tuple = (
  97. self.const.TASK_PROCESSING_STATUS,
  98. self.const.TASK_KIMI_FINISHED_STATUS,
  99. self.const.TASK_SPIDER_FINISHED_STATUS
  100. )
  101. else:
  102. return
  103. select_processing_sql = f"""
  104. SELECT
  105. trace_id, content_status_update_time, process_times, content_status
  106. FROM
  107. {self.article_match_video_table}
  108. WHERE
  109. content_status in {processing_status_tuple}
  110. and process_times <= {self.const.TASK_MAX_PROCESS_TIMES}
  111. and publish_flag = {publish_flag};
  112. """
  113. processing_articles = await self.mysql_client.async_select(select_processing_sql)
  114. if processing_articles:
  115. processing_list = [
  116. {
  117. "trace_id": item[0],
  118. "content_status_update_time": item[1],
  119. "process_times": item[2],
  120. "content_status": item[3]
  121. }
  122. for item in processing_articles
  123. ]
  124. for obj in processing_list:
  125. if int(time.time()) - obj['content_status_update_time'] >= self.const.TASK_PROCESSING_TIMEOUT:
  126. # 认为该任务失败
  127. await self.roll_back_content_status_when_fails(
  128. process_times=obj['process_times'] + 1,
  129. trace_id=obj['trace_id'],
  130. ori_content_status=obj['content_status']
  131. )
  132. async def get_video_list(self, content_id):
  133. """
  134. 判断该文章是否存在历史匹配视频
  135. :param content_id
  136. :return:
  137. """
  138. sql = f"""
  139. SELECT id
  140. FROM {self.article_crawler_video_table}
  141. WHERE content_id = '{content_id}' and download_status = {self.const.VIDEO_DOWNLOAD_SUCCESS_STATUS};
  142. """
  143. res_tuple = await self.mysql_client.async_select(sql)
  144. if len(res_tuple) >= self.const.MIN_MATCH_VIDEO_NUM:
  145. return True
  146. else:
  147. return False
  148. async def update_content_status(self, new_content_status, trace_id, ori_content_status):
  149. """
  150. :param new_content_status:
  151. :param trace_id:
  152. :param ori_content_status:
  153. :return:
  154. """
  155. update_sql = f"""
  156. UPDATE {self.article_match_video_table}
  157. SET content_status = %s, content_status_update_time = %s
  158. WHERE trace_id = %s and content_status = %s;
  159. """
  160. row_counts = await self.mysql_client.async_insert(
  161. sql=update_sql,
  162. params=(
  163. new_content_status,
  164. int(time.time()),
  165. trace_id,
  166. ori_content_status
  167. )
  168. )
  169. return row_counts
  170. async def roll_back_content_status_when_fails(self, process_times, trace_id, ori_content_status=None):
  171. """
  172. 处理失败,回滚至初始状态,处理次数加 1
  173. :param process_times:
  174. :param trace_id:
  175. :param ori_content_status:
  176. :return:
  177. """
  178. if not ori_content_status:
  179. ori_content_status = self.const.TASK_PROCESSING_STATUS
  180. update_article_sql = f"""
  181. UPDATE {self.article_match_video_table}
  182. SET
  183. content_status = %s,
  184. content_status_update_time = %s,
  185. process_times = %s
  186. WHERE trace_id = %s and content_status = %s;
  187. """
  188. await self.mysql_client.async_insert(
  189. sql=update_article_sql,
  190. params=(
  191. self.const.TASK_INIT_STATUS,
  192. int(time.time()),
  193. process_times + 1,
  194. trace_id,
  195. ori_content_status
  196. )
  197. )
  198. async def judge_whether_same_content_id_is_processing(self, content_id):
  199. """
  200. 同一个 content_id 只需要处理一次
  201. :param content_id:
  202. :return:
  203. success: 4
  204. init: 0
  205. fail: 99
  206. todo: 存在处理失败的content_id是否需要不再处理
  207. """
  208. select_sql = f"""
  209. SELECT distinct content_status
  210. FROM {self.article_match_video_table}
  211. WHERE content_id = '{content_id}';
  212. """
  213. result = await self.mysql_client.async_select(select_sql)
  214. if result:
  215. for item in result:
  216. content_status = item[0]
  217. # if content_status not in {self.TASK_INIT_STATUS, self.TASK_PUBLISHED_STATUS} :
  218. if content_status in {
  219. self.const.TASK_KIMI_FINISHED_STATUS,
  220. self.const.TASK_SPIDER_FINISHED_STATUS,
  221. self.const.TASK_ETL_COMPLETE_STATUS,
  222. self.const.TASK_PROCESSING_STATUS,
  223. self.const.TASK_PUBLISHED_STATUS
  224. }:
  225. return True
  226. return False
  227. else:
  228. return False
  229. async def get_downloaded_videos(self, content_id):
  230. """
  231. 获取已下载的视频
  232. :return:
  233. """
  234. sql = f"""
  235. SELECT platform, play_count, like_count, video_oss_path, cover_oss_path, user_id
  236. FROM {self.article_crawler_video_table}
  237. WHERE content_id = '{content_id}' and download_status = {self.const.VIDEO_DOWNLOAD_SUCCESS_STATUS};
  238. ORDER BY score DESC;
  239. """
  240. res_tuple = await self.mysql_client.async_select(sql)
  241. return [
  242. {
  243. "platform": i[0],
  244. "play_count": i[1],
  245. "like_count": i[2],
  246. "video_oss_path": i[3],
  247. "cover_oss_path": i[4],
  248. "uid": i[5]
  249. }
  250. for i in res_tuple
  251. ]
  252. async def get_kimi_status(self, content_id):
  253. """
  254. 通过 content_id 获取kimi info
  255. :return:
  256. """
  257. select_sql = f"""
  258. select kimi_status
  259. from {self.article_text_table}
  260. where content_id = '{content_id}';
  261. """
  262. response = await self.mysql_client.async_select(select_sql)
  263. if response:
  264. kimi_status = response[0][0]
  265. return kimi_status
  266. else:
  267. return self.const.ARTICLE_TEXT_TABLE_ERROR
  268. async def kimi_task(self, params):
  269. """
  270. 执行 kimi 任务
  271. :return:
  272. """
  273. content_id = params['content_id']
  274. trace_id = params['trace_id']
  275. process_times = params['process_times']
  276. kimi_status_code = await self.get_kimi_status(content_id=content_id)
  277. if kimi_status_code == self.const.KIMI_SUCCESS_STATUS:
  278. affected_rows = await self.update_content_status(
  279. new_content_status=self.const.TASK_KIMI_FINISHED_STATUS,
  280. trace_id=trace_id,
  281. ori_content_status=self.const.TASK_INIT_STATUS
  282. )
  283. if affected_rows == 0:
  284. logging(
  285. code="6000",
  286. info="多个进程抢占同一个任务的执行状态锁,抢占失败,return"
  287. )
  288. return
  289. get_kimi_sql = f"""
  290. SELECT article_title, kimi_title, kimi_summary, kimi_keys
  291. FROM {self.article_text_table}
  292. WHERE content_id = '{content_id}';
  293. """
  294. kimi_info = await self.mysql_client.async_select(get_kimi_sql)
  295. return {
  296. "kimi_title": kimi_info[0][1],
  297. "ori_title": kimi_info[0][0],
  298. "kimi_summary": kimi_info[0][2],
  299. "kimi_keys": json.loads(kimi_info[0][3])
  300. }
  301. elif kimi_status_code == self.const.ARTICLE_TEXT_TABLE_ERROR:
  302. logging(
  303. code="4000",
  304. info="long_articles_text表中未找到 content_id"
  305. )
  306. else:
  307. # 开始处理,讲 content_status 从 0 改为 101
  308. affected_rows = await self.update_content_status(
  309. new_content_status=self.const.TASK_PROCESSING_STATUS,
  310. trace_id=trace_id,
  311. ori_content_status=self.const.TASK_INIT_STATUS
  312. )
  313. if affected_rows == 0:
  314. logging(
  315. code="6000",
  316. info="多个进程抢占同一个任务的执行状态锁,抢占失败,return"
  317. )
  318. return
  319. K = KimiServer()
  320. try:
  321. select_sql = f"""
  322. select article_title, article_text
  323. from {self.article_text_table}
  324. where content_id = '{content_id}'
  325. """
  326. res = await self.mysql_client.async_select(select_sql)
  327. article_obj = {
  328. "article_title": res[0][0],
  329. "article_text": res[0][1],
  330. "content_id": content_id
  331. }
  332. kimi_info = await K.search_kimi_schedule(params=article_obj)
  333. kimi_title = kimi_info['k_title']
  334. content_title = kimi_info['content_title'].replace("'", "").replace('"', "")
  335. content_keys = json.dumps(kimi_info['content_keys'], ensure_ascii=False)
  336. update_kimi_sql = f"""
  337. UPDATE {self.article_text_table}
  338. SET
  339. kimi_title = %s,
  340. kimi_summary = %s,
  341. kimi_keys = %s,
  342. kimi_status = %s
  343. WHERE content_id = %s;"""
  344. await self.mysql_client.async_insert(
  345. sql=update_kimi_sql,
  346. params=(
  347. kimi_title, content_title, content_keys, self.const.KIMI_SUCCESS_STATUS, params['content_id'])
  348. )
  349. await self.update_content_status(
  350. new_content_status=self.const.TASK_KIMI_FINISHED_STATUS,
  351. trace_id=trace_id,
  352. ori_content_status=self.const.TASK_PROCESSING_STATUS
  353. )
  354. return {
  355. "kimi_title": kimi_title,
  356. "ori_title": article_obj['article_title'],
  357. "kimi_summary": content_title,
  358. "kimi_keys": kimi_info['content_keys']
  359. }
  360. except Exception as e:
  361. # kimi 任务处理失败
  362. update_kimi_sql = f"""
  363. UPDATE {self.article_text_table}
  364. SET
  365. kimi_status = %s
  366. WHERE content_id = %s
  367. """
  368. await self.mysql_client.async_insert(
  369. sql=update_kimi_sql,
  370. params=(
  371. self.const.KIMI_FAIL_STATUS,
  372. content_id
  373. )
  374. )
  375. # 将状态由 101 回退为 0
  376. await self.roll_back_content_status_when_fails(
  377. process_times=process_times,
  378. trace_id=trace_id
  379. )
  380. return {}
  381. async def spider_task(self, params, kimi_result):
  382. """
  383. 爬虫任务
  384. :return:
  385. """
  386. trace_id = params['trace_id']
  387. content_id = params['content_id']
  388. process_times = params['process_times']
  389. gh_id = params['gh_id']
  390. select_sql = f"""
  391. SELECT count(id)
  392. FROM {self.article_crawler_video_table}
  393. WHERE content_id = '{content_id}'
  394. AND download_status = {self.const.VIDEO_DOWNLOAD_SUCCESS_STATUS};
  395. """
  396. count_tuple = await self.mysql_client.async_select(select_sql)
  397. counts = count_tuple[0][0]
  398. if counts >= self.const.MIN_MATCH_VIDEO_NUM:
  399. await self.update_content_status(
  400. new_content_status=self.const.TASK_SPIDER_FINISHED_STATUS,
  401. trace_id=trace_id,
  402. ori_content_status=self.const.TASK_KIMI_FINISHED_STATUS
  403. )
  404. return True
  405. # 开始处理,将状态由 1 改成 101
  406. affected_rows = await self.update_content_status(
  407. new_content_status=self.const.TASK_PROCESSING_STATUS,
  408. ori_content_status=self.const.TASK_KIMI_FINISHED_STATUS,
  409. trace_id=trace_id
  410. )
  411. if affected_rows == 0:
  412. logging(
  413. code="6000",
  414. info="多个进程抢占同一个任务的执行状态锁,抢占失败,return"
  415. )
  416. return False
  417. try:
  418. logging(
  419. code="spider_1001",
  420. info="开始执行搜索任务",
  421. trace_id=trace_id,
  422. data=kimi_result
  423. )
  424. search_videos_count = await search_videos_from_web(
  425. info={
  426. "ori_title": kimi_result['ori_title'],
  427. "kimi_summary": kimi_result['kimi_summary'],
  428. "kimi_keys": kimi_result['kimi_keys'],
  429. "trace_id": trace_id,
  430. "gh_id": gh_id,
  431. "content_id": content_id,
  432. "crawler_video_table": self.article_crawler_video_table
  433. },
  434. gh_id_map=self.account_map,
  435. db_client=self.mysql_client
  436. )
  437. if search_videos_count >= self.const.MIN_MATCH_VIDEO_NUM:
  438. # 表示爬虫任务执行成功, 将状态从 101 改为 2
  439. logging(
  440. code="spider_1002",
  441. info="搜索成功",
  442. trace_id=trace_id,
  443. data=kimi_result
  444. )
  445. await self.update_content_status(
  446. new_content_status=self.const.TASK_SPIDER_FINISHED_STATUS,
  447. trace_id=trace_id,
  448. ori_content_status=self.const.TASK_PROCESSING_STATUS
  449. )
  450. return True
  451. else:
  452. logging(
  453. code="spider_1003",
  454. info="搜索失败",
  455. trace_id=trace_id,
  456. data=kimi_result
  457. )
  458. await self.roll_back_content_status_when_fails(
  459. process_times=process_times + 1,
  460. trace_id=trace_id
  461. )
  462. return False
  463. except Exception as e:
  464. await self.roll_back_content_status_when_fails(
  465. process_times=process_times + 1,
  466. trace_id=trace_id
  467. )
  468. print("爬虫处理失败: {}".format(e))
  469. return False
  470. async def etl_task(self, params):
  471. """
  472. download && upload videos
  473. :param params:
  474. :return:
  475. """
  476. trace_id = params['trace_id']
  477. content_id = params['content_id']
  478. process_times = params['process_times']
  479. # 判断是否有三条已经下载完成的视频
  480. select_sql = f"""
  481. select count(id)
  482. from {self.article_crawler_video_table}
  483. where content_id = '{content_id}' and download_status = {self.const.VIDEO_DOWNLOAD_SUCCESS_STATUS};
  484. """
  485. video_count_tuple = await self.mysql_client.async_select(select_sql)
  486. video_count = video_count_tuple[0][0]
  487. if video_count >= self.const.MIN_MATCH_VIDEO_NUM:
  488. affect_rows = await self.update_content_status(
  489. ori_content_status=self.const.TASK_SPIDER_FINISHED_STATUS,
  490. trace_id=trace_id,
  491. new_content_status=self.const.TASK_ETL_COMPLETE_STATUS
  492. )
  493. if affect_rows == 0:
  494. logging(
  495. code="6000",
  496. info="多个进程抢占同一个任务的执行状态锁,抢占失败,return"
  497. )
  498. return False
  499. return True
  500. else:
  501. # 开始处理, 将文章状态修改为处理状态
  502. affected_rows = await self.update_content_status(
  503. ori_content_status=self.const.TASK_SPIDER_FINISHED_STATUS,
  504. trace_id=trace_id,
  505. new_content_status=self.const.TASK_PROCESSING_STATUS
  506. )
  507. if affected_rows == 0:
  508. logging(
  509. code="6000",
  510. info="多个进程抢占同一个任务的执行状态锁,抢占失败,return"
  511. )
  512. return False
  513. select_sql = f"""
  514. SELECT id, out_video_id, platform, video_title, video_url, cover_url, user_id, trace_id
  515. FROM {self.article_crawler_video_table}
  516. WHERE content_id = '{content_id}' and download_status != {self.const.VIDEO_DOWNLOAD_SUCCESS_STATUS}
  517. ORDER BY score DESC;
  518. """
  519. videos_need_to_download_tuple = await self.mysql_client.async_select(select_sql)
  520. downloaded_count = 0
  521. for line in videos_need_to_download_tuple:
  522. params = {
  523. "id": line[0],
  524. "video_id": line[1],
  525. "platform": line[2],
  526. "video_title": line[3],
  527. "video_url": line[4],
  528. "cover_url": line[5],
  529. "user_id": line[6],
  530. "trace_id": line[7]
  531. }
  532. try:
  533. local_video_path, local_cover_path = generate_video_path(params['platform'], params['video_id'])
  534. # download videos
  535. file_path = await download_video(
  536. file_path=local_video_path,
  537. platform=params['platform'],
  538. video_url=params['video_url']
  539. )
  540. if not file_path:
  541. # 说明视频下载失败,无需上传该视频, 将该条记录设置为失败状态
  542. update_sql = f"""
  543. UPDATE {self.article_crawler_video_table}
  544. SET download_status = %s
  545. WHERE id = %s;
  546. """
  547. await self.mysql_client.async_insert(
  548. sql=update_sql,
  549. params=(self.const.VIDEO_DOWNLOAD_FAIL_STATUS, params['id'])
  550. )
  551. logging(
  552. code="etl_1001",
  553. info="etl_下载视频失败",
  554. trace_id=trace_id,
  555. function="etl_task"
  556. )
  557. else:
  558. # download cover
  559. cover_path = await download_cover(
  560. file_path=local_cover_path,
  561. platform=params['platform'],
  562. cover_url=params['cover_url']
  563. )
  564. # upload video to oss
  565. oss_video = await upload_to_oss(
  566. local_video_path=file_path,
  567. download_type="video"
  568. )
  569. # upload cover to oss
  570. if cover_path:
  571. oss_cover = await upload_to_oss(
  572. local_video_path=cover_path,
  573. download_type="image"
  574. )
  575. else:
  576. oss_cover = None
  577. # change status to success
  578. update_sql = f"""
  579. UPDATE {self.article_crawler_video_table}
  580. SET video_oss_path = %s, cover_oss_path = %s, download_status = %s
  581. WHERE id = %s;
  582. """
  583. await self.mysql_client.async_insert(
  584. sql=update_sql,
  585. params=(
  586. oss_video,
  587. oss_cover,
  588. self.const.VIDEO_DOWNLOAD_SUCCESS_STATUS,
  589. params['id']
  590. )
  591. )
  592. downloaded_count += 1
  593. logging(
  594. code="etl_1002",
  595. info="etl_视频下载成功",
  596. trace_id=trace_id,
  597. function="etl_task"
  598. )
  599. # 如果下载的视频数已经大于3, 则直接退出循环,修改状态为ETL成功状态
  600. if downloaded_count > self.const.MIN_MATCH_VIDEO_NUM:
  601. await self.update_content_status(
  602. ori_content_status=self.const.TASK_PROCESSING_STATUS,
  603. trace_id=trace_id,
  604. new_content_status=self.const.TASK_ETL_COMPLETE_STATUS
  605. )
  606. return True
  607. except Exception as e:
  608. update_sql = f"""
  609. UPDATE {self.article_crawler_video_table}
  610. SET download_status = %s
  611. WHERE id = %s;
  612. """
  613. await self.mysql_client.async_insert(
  614. sql=update_sql,
  615. params=(self.const.VIDEO_DOWNLOAD_FAIL_STATUS, params['id'])
  616. )
  617. logging(
  618. code="etl_1001",
  619. info="etl_下载视频失败",
  620. trace_id=trace_id,
  621. function="etl_task"
  622. )
  623. if downloaded_count >= 3:
  624. await self.update_content_status(
  625. ori_content_status=self.const.TASK_PROCESSING_STATUS,
  626. trace_id=trace_id,
  627. new_content_status=self.const.TASK_ETL_COMPLETE_STATUS
  628. )
  629. return True
  630. else:
  631. await self.roll_back_content_status_when_fails(
  632. process_times=process_times + 1,
  633. trace_id=trace_id
  634. )
  635. return False
  636. async def publish_task(self, params, kimi_title):
  637. """
  638. 发布任务
  639. :param kimi_title:
  640. :param params:
  641. :return:
  642. """
  643. gh_id = params['gh_id']
  644. flow_pool_level = params['flow_pool_level']
  645. content_id = params['content_id']
  646. trace_id = params['trace_id']
  647. process_times = params['process_times']
  648. # 开始处理,将状态修改为操作状态
  649. affected_rows = await self.update_content_status(
  650. ori_content_status=self.const.TASK_ETL_COMPLETE_STATUS,
  651. trace_id=trace_id,
  652. new_content_status=self.const.TASK_PROCESSING_STATUS
  653. )
  654. if affected_rows == 0:
  655. logging(
  656. code="6000",
  657. info="多个进程抢占同一个任务的执行状态锁,抢占失败,return"
  658. )
  659. return False
  660. try:
  661. download_videos = await self.get_downloaded_videos(content_id)
  662. match flow_pool_level:
  663. case "autoArticlePoolLevel4":
  664. # 冷启层, 全量做
  665. video_list = shuffle_list(download_videos)[:3]
  666. case "autoArticlePoolLevel3":
  667. if self.gh_id_dict.get(gh_id):
  668. video_list = shuffle_list(download_videos)[:3]
  669. else:
  670. video_list = download_videos[:3]
  671. case "autoArticlePoolLevel2":
  672. # 次条,只针对具体账号做
  673. video_list = []
  674. case "autoArticlePoolLevel1":
  675. # 头条,先不做
  676. video_list = download_videos[:3]
  677. case _:
  678. video_list = download_videos[:3]
  679. L = []
  680. for video_obj in video_list:
  681. params = {
  682. "videoPath": video_obj['video_oss_path'],
  683. "uid": video_obj['uid'],
  684. "title": kimi_title
  685. }
  686. publish_response = await publish_to_pq(params)
  687. video_id = publish_response['data']['id']
  688. response = await get_pq_video_detail(video_id)
  689. obj = {
  690. "uid": video_obj['uid'],
  691. "source": video_obj['platform'],
  692. "kimiTitle": kimi_title,
  693. "videoId": response['data'][0]['id'],
  694. "videoCover": response['data'][0]['shareImgPath'],
  695. "videoPath": response['data'][0]['videoPath'],
  696. "videoOss": video_obj['video_oss_path']
  697. }
  698. L.append(obj)
  699. update_sql = f"""
  700. UPDATE {self.article_match_video_table}
  701. SET content_status = %s, response = %s, process_times = %s
  702. WHERE trace_id = %s and content_status = %s;
  703. """
  704. # 从操作中状态修改为已发布状态
  705. await self.mysql_client.async_insert(
  706. sql=update_sql,
  707. params=(
  708. self.const.TASK_PUBLISHED_STATUS,
  709. json.dumps(L, ensure_ascii=False),
  710. process_times + 1,
  711. trace_id,
  712. self.const.TASK_PROCESSING_STATUS
  713. )
  714. )
  715. except Exception as e:
  716. await self.roll_back_content_status_when_fails(
  717. process_times=params['process_times'] + 1,
  718. trace_id=params['trace_id']
  719. )
  720. print(e)
  721. async def start_process(self, params):
  722. """
  723. 处理单篇文章
  724. :param params:
  725. :return:
  726. """
  727. # step1: 执行 kimi 操作
  728. # time.sleep(5) # 测试多个进程操作同一个 task 的等待时间
  729. kimi_result = await self.kimi_task(params)
  730. trace_id = params['trace_id']
  731. process_times = params['process_times']
  732. content_id = params['content_id']
  733. gh_id = params['gh_id']
  734. publish_flag = params['publish_flag']
  735. print(kimi_result)
  736. if kimi_result:
  737. # 等待 kimi 操作执行完成之后,开始执行 spider_task
  738. print("kimi success")
  739. logging(
  740. code=3001,
  741. info="kimi success",
  742. trace_id=trace_id
  743. )
  744. spider_flag = await self.spider_task(params=params, kimi_result=kimi_result)
  745. if spider_flag:
  746. # 等待爬虫执行完成后,开始执行 etl_task
  747. print("spider success")
  748. logging(
  749. code=3002,
  750. info="spider_success",
  751. trace_id=trace_id
  752. )
  753. etl_flag = await self.etl_task(params)
  754. if etl_flag:
  755. # 等待下载上传完成,执行发布任务
  756. print("etl success")
  757. logging(
  758. code="3003",
  759. info="etl_success",
  760. trace_id=trace_id
  761. )
  762. """
  763. todo 若新建计划,计划为设置托管,但接入账号又在配置账号中,仍会走托管逻辑,需考虑历史存量的处理
  764. 目前先对这两种情况都做托管操作
  765. """
  766. if publish_flag == self.const.DO_NOT_NEED_PUBLISH:
  767. logging(
  768. code="3013",
  769. info="不需要发布,长文系统托管发布",
  770. trace_id=trace_id
  771. )
  772. return
  773. else:
  774. try:
  775. await self.publish_task(params, kimi_result['kimi_title'])
  776. logging(
  777. code="3004",
  778. info="publish_success",
  779. trace_id=trace_id
  780. )
  781. await record_trace_id(
  782. trace_id=trace_id,
  783. status=self.const.RECORD_SUCCESS_TRACE_ID_CODE
  784. )
  785. except Exception as e:
  786. logging(
  787. code="6004",
  788. info="publish 失败--{}".format(e),
  789. trace_id=params['trace_id']
  790. )
  791. else:
  792. logging(
  793. code="6003",
  794. info="ETL 处理失败",
  795. trace_id=params['trace_id']
  796. )
  797. else:
  798. logging(
  799. code="6002",
  800. info="爬虫处理失败",
  801. trace_id=params['trace_id']
  802. )
  803. else:
  804. logging(
  805. code="6001",
  806. info="kimi 处理失败",
  807. trace_id=trace_id
  808. )
  809. if process_times >= self.const.TASK_MAX_PROCESS_TIMES:
  810. logging(
  811. code="6011",
  812. info="kimi处理次数达到上限, 放弃处理",
  813. trace_id=trace_id
  814. )
  815. # 将相同的content_id && content_status = 0的状态修改为kimi 失败状态
  816. update_sql = f"""
  817. UPDATE {self.article_match_video_table}
  818. SET content_status = %s
  819. WHERE content_id = %s and content_status = %s;
  820. """
  821. affected_rows = await self.mysql_client.async_insert(
  822. sql=update_sql,
  823. params=(
  824. self.const.KIMI_ILLEGAL_STATUS,
  825. content_id,
  826. self.const.TASK_INIT_STATUS
  827. )
  828. )
  829. bot(
  830. title="KIMI 处理失败",
  831. detail={
  832. "content_id": content_id,
  833. "affected_rows": affected_rows
  834. }
  835. )
  836. async def process_task(self, params):
  837. """
  838. 处理任务
  839. :return:
  840. """
  841. content_id = params['content_id']
  842. download_videos = await self.get_video_list(content_id)
  843. if not download_videos:
  844. # 开始处理, 判断是否有相同的文章 id 正在处理
  845. processing_flag = await self.judge_whether_same_content_id_is_processing(content_id)
  846. if processing_flag:
  847. logging(
  848. code="9001",
  849. info="该 content id 正在处理中, 跳过此任务--{}".format(content_id)
  850. )
  851. else:
  852. await self.start_process(params=params)
  853. else:
  854. print("存在已下载视频")
  855. async def deal(self):
  856. """
  857. function
  858. :return:
  859. """
  860. task_list = await self.get_tasks()
  861. task_dict = {}
  862. # 对 content_id去重
  863. for task in task_list:
  864. key = task['content_id']
  865. task_dict[key] = task
  866. process_list = []
  867. for item in task_dict:
  868. process_list.append(task_dict[item])
  869. logging(
  870. code="5001",
  871. info="Match Task Got {} this time".format(len(process_list)),
  872. function="Publish Task"
  873. )
  874. if task_list:
  875. total_task = len(process_list)
  876. print(process_list)
  877. a = time.time()
  878. print("开始处理,一共{}个任务".format(total_task))
  879. tasks = [self.process_task(params) for params in process_list]
  880. await asyncio.gather(*tasks)
  881. b = time.time()
  882. print("处理时间: {} s".format(b - a))
  883. else:
  884. logging(
  885. code="9008",
  886. info="没有要处理的请求"
  887. )