alg_growth_3rd_gh_reply_video_v1.py 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576
  1. # -*- coding: utf-8 -*-
  2. import pandas as pd
  3. import traceback
  4. import odps
  5. from odps import ODPS
  6. import json
  7. import time
  8. from pymysql.cursors import DictCursor
  9. from datetime import datetime, timedelta
  10. from db_helper import MysqlHelper
  11. from my_utils import check_table_partition_exits_v2, get_dataframe_from_odps, \
  12. get_odps_df_of_max_partition, get_odps_instance, get_odps_df_of_recent_partitions
  13. from my_utils import request_post, send_msg_to_feishu
  14. from my_config import set_config
  15. import numpy as np
  16. from log import Log
  17. import os
  18. from argparse import ArgumentParser
  19. from constants import AutoReplyAccountType
  20. from alg_growth_common import check_unsafe_video, filter_unsafe_video
  21. CONFIG, _ = set_config()
  22. LOGGER = Log()
  23. BASE_GROUP_NAME = '3rd-party-base'
  24. EXPLORE1_GROUP_NAME = '3rd-party-explore1'
  25. EXPLORE2_GROUP_NAME = '3rd-party-explore2'
  26. # GH_IDS will be updated by get_and_update_gh_ids
  27. GH_IDS = ('default',)
  28. account_map = {}
  29. pd.set_option('display.max_rows', None)
  30. CDN_IMG_OPERATOR = "?x-oss-process=image/resize,m_fill,w_600,h_480,limit_0/format,jpg/watermark,image_eXNoL3BpYy93YXRlcm1hcmtlci9pY29uX3BsYXlfd2hpdGUucG5nP3gtb3NzLXByb2Nlc3M9aW1hZ2UvcmVzaXplLHdfMTQ0,g_center"
  31. ODS_PROJECT = "loghubods"
  32. EXPLORE_POOL_TABLE = 'alg_growth_video_return_stats_history'
  33. GH_REPLY_STATS_TABLE = 'alg_growth_3rd_gh_reply_video_stats'
  34. ODPS_RANK_RESULT_TABLE = 'alg_3rd_gh_autoreply_video_rank_data'
  35. GH_DETAIL = 'gh_detail'
  36. RDS_RANK_RESULT_TABLE = 'alg_gh_autoreply_video_rank_data'
  37. STATS_PERIOD_DAYS = 5
  38. DEFAULT_SEND_N = 1
  39. MAX_SEND_N = 3
  40. default_video = {
  41. '泛生活': [20463342, 14095344, 13737337],
  42. '泛历史': [13586800, 12794884, 12117356],
  43. }
  44. def get_and_update_gh_ids(run_dt):
  45. db = MysqlHelper(CONFIG.MYSQL_GROWTH_INFO)
  46. gh_type = AutoReplyAccountType.EXTERNAL_GZH.value
  47. sqlstr = f"""
  48. SELECT gh_id, gh_name, category1, category2, channel,
  49. video_ids, strategy_status,
  50. autoreply_send_minigram_num as send_n
  51. FROM {GH_DETAIL}
  52. WHERE is_delete = 0 AND `type` = {gh_type}
  53. """
  54. account_data = db.get_data(sqlstr, DictCursor)
  55. account_df = pd.DataFrame(account_data)
  56. # default单独处理
  57. if 'default' not in account_df['gh_id'].values:
  58. new_row = pd.DataFrame({'gh_id': ['default'], 'gh_name': ['默认'], 'type': [2], 'category1': ['泛生活']},
  59. index=[0])
  60. account_df = pd.concat([account_df, new_row], ignore_index=True)
  61. account_df = account_df.drop_duplicates(subset=['gh_id'])
  62. global GH_IDS
  63. GH_IDS = tuple(account_df['gh_id'])
  64. global account_map
  65. account_map = { x['gh_id']: x for x in account_df.to_dict(orient='records') }
  66. for gh_id in account_map:
  67. account_info = account_map[gh_id]
  68. account_info['send_n'] = account_info.get('send_n', 1)
  69. return account_df
  70. def check_data_partition(project, table, data_dt, data_hr=None):
  71. """检查数据是否准备好"""
  72. try:
  73. partition_spec = {'dt': data_dt}
  74. if data_hr:
  75. partition_spec['hour'] = data_hr
  76. part_exist, data_count = check_table_partition_exits_v2(
  77. project, table, partition_spec)
  78. except Exception as e:
  79. data_count = 0
  80. return data_count
  81. def get_last_strategy_result(project, rank_table, dt_version, key):
  82. strategy_df = get_odps_df_of_max_partition(
  83. project, rank_table, {'ctime': dt_version}
  84. ).to_pandas()
  85. dt_version = strategy_df.iloc[0]['dt_version']
  86. sub_df = strategy_df.query(f'strategy_key == "{key}"')
  87. sub_df = sub_df[['gh_id', 'video_id', 'strategy_key', 'sort']].drop_duplicates()
  88. return sub_df, dt_version
  89. def process_reply_stats(project, table, period, run_dt):
  90. # 获取多天即转统计数据用于聚合
  91. df = get_odps_df_of_recent_partitions(project, table, period, {'dt': run_dt})
  92. df = df.to_pandas()
  93. df['video_id'] = df['video_id'].astype('int64')
  94. df = df[['gh_id', 'video_id', 'send_count', 'first_visit_uv', 'day0_return']]
  95. # 获取统计数据时统一去除不安全视频
  96. df = filter_unsafe_video(df)
  97. # 账号内聚合
  98. df = df.groupby(['video_id', 'gh_id']).agg({
  99. 'send_count': 'sum',
  100. 'first_visit_uv': 'sum',
  101. 'day0_return': 'sum'
  102. }).reset_index()
  103. # 聚合所有数据作为default
  104. default_stats_df = df.groupby('video_id').agg({
  105. 'send_count': 'sum',
  106. 'first_visit_uv': 'sum',
  107. 'day0_return': 'sum'
  108. }).reset_index()
  109. default_stats_df['gh_id'] = 'default'
  110. merged_df = pd.concat([df, default_stats_df]).reset_index(drop=True)
  111. merged_df['score'] = merged_df['day0_return'] / (merged_df['first_visit_uv'] + 100)
  112. return merged_df
  113. def rank_for_layer1(run_dt, run_hour, project, table, gh):
  114. # TODO: 加审核&退场
  115. df = get_odps_df_of_max_partition(project, table, {'dt': run_dt})
  116. df = df.to_pandas()
  117. df = filter_unsafe_video(df)
  118. # 确保重跑时可获得一致结果
  119. dt_version = f'{run_dt}{run_hour}'
  120. np.random.seed(int(dt_version) + 1)
  121. # TODO: 修改权重计算策略
  122. df['score'] = 1.0
  123. # 按照 category1 分类后进行加权随机抽样
  124. sampled_df = df.groupby('category1').apply(
  125. lambda x: x.sample(n=MAX_SEND_N, weights=x['score'], replace=False)).reset_index(drop=True)
  126. sampled_df['sort'] = sampled_df.groupby('category1')['score'].rank(method='first', ascending=False).astype(int)
  127. # 按得分排序
  128. sampled_df = sampled_df.sort_values(by=['category1', 'score'], ascending=[True, False]).reset_index(drop=True)
  129. sampled_df['strategy_key'] = EXPLORE1_GROUP_NAME
  130. sampled_df['dt_version'] = dt_version
  131. merged_dfs = []
  132. for gh_id in GH_IDS:
  133. sub_gh_df = gh.query(f'gh_id == "{gh_id}"')
  134. account_info = account_map[gh_id]
  135. send_n = account_info['send_n']
  136. sub_video_df = sampled_df.query(f'sort <= {send_n}').copy()
  137. merged_df = sub_video_df.merge(sub_gh_df, on='category1')
  138. merged_df['sort'] = send_n + 1 - merged_df['sort']
  139. merged_dfs.append(merged_df)
  140. extend_df = pd.concat(merged_dfs)
  141. result_df = extend_df[['strategy_key', 'dt_version', 'gh_id', 'sort', 'video_id', 'score']]
  142. return result_df
  143. def rank_for_layer2(run_dt, run_hour, project, stats_table, rank_table):
  144. stats_df = process_reply_stats(project, stats_table, STATS_PERIOD_DAYS, run_dt)
  145. # 确保重跑时可获得一致结果
  146. dt_version = f'{run_dt}{run_hour}'
  147. np.random.seed(int(dt_version) + 1)
  148. # TODO: 计算账号间相关性
  149. ## 账号两两组合,取有RoVn数值视频的交集,单个账号内的RoVn(平滑后)组成向量
  150. ## 求向量相关系数或cosine相似度
  151. ## 单个视频的RoVn加权求和
  152. # 当前实现基础版本:只在账号内求二级探索排序分
  153. sampled_dfs = []
  154. # 处理default逻辑(default-explore2)
  155. default_stats_df = stats_df.query('gh_id == "default"')
  156. sampled_df = default_stats_df.sample(n=DEFAULT_SEND_N, weights=default_stats_df['score'])
  157. sampled_df['sort'] = range(1, len(sampled_df) + 1)
  158. sampled_dfs.append(sampled_df)
  159. # 基础过滤for账号
  160. df = stats_df.query('send_count > 200 and score > 0')
  161. # fallback to base if necessary
  162. base_strategy_df, _ = get_last_strategy_result(
  163. project, rank_table, dt_version, BASE_GROUP_NAME)
  164. for gh_id in GH_IDS:
  165. if gh_id == 'default':
  166. continue
  167. account_info = account_map[gh_id]
  168. send_n = account_info['send_n']
  169. sub_df = df.query(f'gh_id == "{gh_id}"')
  170. if len(sub_df) < send_n:
  171. LOGGER.warning(
  172. "gh_id[{}] rows[{}] not enough for layer2, fallback to base"
  173. .format(gh_id, len(sub_df)))
  174. sub_df = base_strategy_df.query(f'gh_id == "{gh_id}"').copy()
  175. if len(sub_df) < send_n:
  176. LOGGER.warning(
  177. "gh_id[{}] rows[{}] still not enough for layer2, add backup"
  178. .format(gh_id, len(sub_df)))
  179. rows = []
  180. idx = len(sub_df)
  181. exist_video_ids = sub_df['video_id'].unique()
  182. for video_id in default_video[account_info['category1']]:
  183. if video_id in exist_video_ids:
  184. continue
  185. row = {
  186. 'gh_id': gh_id,
  187. 'sort': idx + 1,
  188. 'video_id': video_id,
  189. 'strategy_key': '' # this is not important
  190. }
  191. rows.append(row)
  192. appx_df = pd.DataFrame(rows)
  193. sub_df = pd.concat([sub_df, appx_df])
  194. sub_df['score'] = sub_df['sort']
  195. sampled_df = sub_df.sample(n=send_n, weights=sub_df['score'])
  196. sampled_df['sort'] = range(send_n, 0, -1)
  197. sampled_dfs.append(sampled_df)
  198. extend_df = pd.concat(sampled_dfs)
  199. extend_df['strategy_key'] = EXPLORE2_GROUP_NAME
  200. extend_df['dt_version'] = dt_version
  201. result_df = extend_df[['strategy_key', 'dt_version', 'gh_id', 'sort', 'video_id', 'score']]
  202. return result_df
  203. def rank_for_base(run_dt, run_hour, project, stats_table, rank_table, stg_key):
  204. stats_df = process_reply_stats(project, stats_table, STATS_PERIOD_DAYS, run_dt)
  205. # TODO: support to set base manually
  206. dt_version = f'{run_dt}{run_hour}'
  207. # 获取当前base信息, 策略表dt_version(ctime partition)采用当前时间
  208. base_strategy_df, _ = get_last_strategy_result(
  209. project, rank_table, dt_version, stg_key)
  210. default_stats_df = stats_df.query('gh_id == "default"')
  211. # 在账号内排序,决定该账号(包括default)的base利用内容
  212. # 排序过程中,确保当前base策略参与排序,因此先关联再过滤
  213. non_default_ids = list(filter(lambda x: x != 'default', GH_IDS))
  214. gh_ids_str = ','.join(f'"{x}"' for x in non_default_ids)
  215. stats_df = stats_df.query(f'gh_id in ({gh_ids_str})')
  216. stats_with_strategy_df = stats_df \
  217. .merge(
  218. base_strategy_df,
  219. on=['gh_id', 'video_id'],
  220. how='outer') \
  221. .query('strategy_key.notna() or (send_count > 500 and score > 0.05)') \
  222. .fillna({'score': 0.0})
  223. # 合并default和分账号数据
  224. grouped_stats_df = pd.concat([default_stats_df, stats_with_strategy_df]).reset_index()
  225. def set_top_n(group, n=2):
  226. group_sorted = group.sort_values(by='score', ascending=False)
  227. top_n = group_sorted.head(n)
  228. top_n['sort'] = range(1, len(top_n) + 1)
  229. return top_n
  230. ranked_df = grouped_stats_df.groupby('gh_id').apply(set_top_n, MAX_SEND_N)
  231. sampled_dfs = []
  232. for gh_id in GH_IDS:
  233. account_info = account_map[gh_id]
  234. send_n = account_info['send_n']
  235. sub_df = ranked_df.query(f'gh_id == "{gh_id}" and sort <= {send_n}').copy()
  236. if len(sub_df) < send_n:
  237. LOGGER.warning(
  238. "gh_id[{}] rows[{}] still not enough for base, add backup"
  239. .format(gh_id, len(sub_df)))
  240. rows = []
  241. idx = len(sub_df)
  242. exist_video_ids = sub_df['video_id'].unique()
  243. for video_id in default_video[account_info['category1']]:
  244. if video_id in exist_video_ids:
  245. continue
  246. row = {
  247. 'gh_id': gh_id,
  248. 'sort': idx + 1,
  249. 'video_id': video_id,
  250. 'score': 0.0,
  251. 'strategy_key': '' # this is not important
  252. }
  253. rows.append(row)
  254. if len(sub_df) + len(rows) >= send_n:
  255. break
  256. appx_df = pd.DataFrame(rows)
  257. sub_df = pd.concat([sub_df, appx_df])
  258. sub_df['sort'] = send_n + 1 - sub_df['sort']
  259. sampled_dfs.append(sub_df)
  260. ranked_df = pd.concat(sampled_dfs)
  261. ranked_df = ranked_df.reset_index(drop=True)
  262. ranked_df['strategy_key'] = stg_key
  263. ranked_df['dt_version'] = dt_version
  264. ranked_df = ranked_df[['strategy_key', 'dt_version', 'gh_id', 'sort', 'video_id', 'score']]
  265. return ranked_df
  266. def check_result_data(df):
  267. check_unsafe_video(df)
  268. for gh_id in GH_IDS:
  269. account_info = account_map[gh_id]
  270. for key in (EXPLORE1_GROUP_NAME, EXPLORE2_GROUP_NAME, BASE_GROUP_NAME):
  271. sub_df = df.query(f'gh_id == "{gh_id}" and strategy_key == "{key}"')
  272. n_records = len(sub_df)
  273. if n_records != account_info['send_n']:
  274. raise Exception(f"Unexpected record count: {gh_id},{key},{n_records}")
  275. def postprocess_override_by_config(df, gh_df, dt_version):
  276. override_config = gh_df.query('strategy_status == 0').to_dict(orient='records')
  277. override_data = {
  278. 'strategy_key': [],
  279. 'gh_id': [],
  280. 'sort': [],
  281. 'video_id': []
  282. }
  283. for row in override_config:
  284. gh_id = row['gh_id']
  285. account_info = account_map[gh_id]
  286. try:
  287. video_ids = json.loads(row['video_ids'])
  288. if not isinstance(video_ids, list):
  289. raise Exception("video_ids is not list")
  290. video_ids = video_ids[:account_info['send_n']]
  291. except Exception as e:
  292. LOGGER.error(f"json parse error: {e}. content: {row['video_ids']}")
  293. continue
  294. for idx, video_id in enumerate(video_ids):
  295. for key in (BASE_GROUP_NAME, EXPLORE1_GROUP_NAME, EXPLORE2_GROUP_NAME):
  296. df = df.drop(df.query(
  297. f'gh_id == "{gh_id}" and strategy_key == "{key}" and sort == {idx + 1}'
  298. ).index)
  299. override_data['strategy_key'].append(key)
  300. override_data['gh_id'].append(gh_id)
  301. override_data['sort'].append(idx + 1)
  302. override_data['video_id'].append(video_id)
  303. n_records = len(override_data['strategy_key'])
  304. override_data['dt_version'] = [dt_version] * n_records
  305. override_data['score'] = [0.0] * n_records
  306. df_to_append = pd.DataFrame(override_data)
  307. df = pd.concat([df, df_to_append], ignore_index=True)
  308. # 强制更换不安全视频
  309. idx = df[df['video_id'] == 14403867].index
  310. df.loc[idx, 'video_id'] = 20463342
  311. return df
  312. def build_and_transfer_base_mode(gh_df, run_dt, run_hour, dt_version, dry_run):
  313. layer1_rank = rank_for_layer1(run_dt, run_hour, ODS_PROJECT, EXPLORE_POOL_TABLE, gh_df)
  314. layer2_rank = rank_for_layer2(run_dt, run_hour, ODS_PROJECT, GH_REPLY_STATS_TABLE, ODPS_RANK_RESULT_TABLE)
  315. base_rank = rank_for_base(run_dt, run_hour, ODS_PROJECT, GH_REPLY_STATS_TABLE, ODPS_RANK_RESULT_TABLE,BASE_GROUP_NAME)
  316. final_rank_df = pd.concat([layer1_rank, layer2_rank, base_rank]).reset_index(drop=True)
  317. final_rank_df = postprocess_override_by_config(final_rank_df, gh_df, dt_version)
  318. check_result_data(final_rank_df)
  319. final_df = join_video_info(final_rank_df)
  320. if dry_run:
  321. print("==== ALL DATA ====")
  322. print(final_df[['strategy_key', 'gh_id', 'sort', 'video_id', 'score', 'title']]
  323. .sort_values(by=['strategy_key', 'gh_id', 'sort']))
  324. last_odps_df = get_odps_df_of_max_partition(
  325. ODS_PROJECT, ODPS_RANK_RESULT_TABLE, {'ctime': dt_version}
  326. ).to_pandas()
  327. merged_df = last_odps_df.merge(
  328. final_df, on=['strategy_key', 'gh_id', 'sort'], how='outer')
  329. delta_df = merged_df.query('title_x != title_y')
  330. delta_df = delta_df[['strategy_key', 'gh_id', 'sort',
  331. 'title_x', 'score_x', 'title_y', 'score_y']]
  332. delta_df.to_csv('tmp_delta_data.csv')
  333. return
  334. # save to ODPS
  335. odps_instance = get_odps_instance(ODS_PROJECT)
  336. t = odps_instance.get_table(ODPS_RANK_RESULT_TABLE)
  337. part_spec_dict = {'dt': run_dt, 'hour': run_hour, 'ctime': dt_version}
  338. part_spec = ','.join(['{}={}'.format(k, part_spec_dict[k]) for k in part_spec_dict.keys()])
  339. with t.open_writer(partition=part_spec, create_partition=True, overwrite=True) as writer:
  340. writer.write(list(final_df.itertuples(index=False)))
  341. # sync to MySQL
  342. data_to_insert = [tuple(row) for row in final_df.itertuples(index=False)]
  343. data_columns = list(final_df.columns)
  344. max_time_to_delete = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
  345. mysql = MysqlHelper(CONFIG.MYSQL_GROWTH_INFO)
  346. mysql.batch_insert(RDS_RANK_RESULT_TABLE, data_to_insert, data_columns)
  347. # remove old data of same version
  348. for key in final_df['strategy_key'].unique():
  349. sql = f"""
  350. update {RDS_RANK_RESULT_TABLE}
  351. set is_delete = 1
  352. where
  353. dt_version = '{dt_version}'
  354. and strategy_key = '{key}'
  355. and create_time < '{max_time_to_delete}'
  356. and is_delete = 0
  357. """
  358. rows = mysql.execute(sql)
  359. def build_and_transfer_delta_mode(account_df, dt_version, dry_run):
  360. # 获取最新策略信息, 策略表dt_version(ctime partition)采用当前时间
  361. last_strategy, last_dt_version = get_last_strategy_result(
  362. ODS_PROJECT, ODPS_RANK_RESULT_TABLE, dt_version, BASE_GROUP_NAME)
  363. all_accounts = account_df['gh_id'].unique()
  364. accounts_in_strategy = last_strategy['gh_id'].unique()
  365. delta_accounts = [x for x in set(all_accounts) - set(accounts_in_strategy)]
  366. if len(delta_accounts) > 0:
  367. LOGGER.info('Found {} new accounts: {}'.format(
  368. len(delta_accounts), ','.join(delta_accounts)))
  369. else:
  370. LOGGER.info('Found 0 new account, do nothing.')
  371. return
  372. # 新增账号,不存在历史,可直接忽略strategy_status字段
  373. # TODO: set default by history stats
  374. groups = (BASE_GROUP_NAME, EXPLORE1_GROUP_NAME, EXPLORE2_GROUP_NAME)
  375. rows = []
  376. for gh_id in delta_accounts:
  377. account_info = account_map[gh_id]
  378. configured_videos = account_info['video_ids']
  379. video_ids = default_video[account_info['category1']]
  380. if configured_videos:
  381. LOGGER.info(f'{gh_id} has configured video IDs: {configured_videos}')
  382. try:
  383. video_ids = [int(x) for x in configured_videos.split(',')]
  384. except Exception as e:
  385. print('invalid configured video_ids, use default instead')
  386. for group_key in groups:
  387. for idx in range(account_info['send_n']):
  388. row = {
  389. 'strategy_key': group_key,
  390. 'gh_id': gh_id,
  391. 'sort': idx + 1,
  392. 'video_id': video_ids[idx],
  393. 'dt_version': last_dt_version,
  394. 'score': 0.0
  395. }
  396. rows.append(row)
  397. df = pd.DataFrame(rows)
  398. final_df = join_video_info(df)
  399. if dry_run:
  400. print(final_df)
  401. return
  402. # 增量记录更新至MySQL
  403. data_to_insert = [tuple(row) for row in final_df.itertuples(index=False)]
  404. data_columns = list(final_df.columns)
  405. mysql = MysqlHelper(CONFIG.MYSQL_GROWTH_INFO)
  406. mysql.batch_insert(RDS_RANK_RESULT_TABLE, data_to_insert, data_columns)
  407. # 全量记录写回ODPS
  408. last_odps_df = get_odps_df_of_max_partition(
  409. ODS_PROJECT, ODPS_RANK_RESULT_TABLE, {'ctime': dt_version}
  410. ).to_pandas()
  411. updated_odps_df = pd.concat([final_df, last_odps_df], ignore_index=True)
  412. odps_instance = get_odps_instance(ODS_PROJECT)
  413. t = odps_instance.get_table(ODPS_RANK_RESULT_TABLE)
  414. target_dt = last_dt_version[0:8]
  415. target_hour = last_dt_version[8:10]
  416. part_spec_dict = {'dt': target_dt, 'hour': target_hour, 'ctime': last_dt_version}
  417. part_spec = ','.join(['{}={}'.format(k, part_spec_dict[k]) for k in part_spec_dict.keys()])
  418. with t.open_writer(partition=part_spec, create_partition=True, overwrite=True) as writer:
  419. writer.write(list(updated_odps_df.itertuples(index=False)))
  420. def join_video_info(df):
  421. db = MysqlHelper(CONFIG.MYSQL_INFO)
  422. video_ids = df['video_id'].unique().tolist()
  423. video_ids_str = ','.join([str(x) for x in video_ids])
  424. sql = f"""
  425. SELECT id as video_id, title, cover_img_path FROM wx_video
  426. WHERE id in ({video_ids_str})
  427. """
  428. rows = db.get_data(sql, DictCursor)
  429. video_df = pd.DataFrame(rows)
  430. video_df['cover_url'] = video_df['cover_img_path'] + CDN_IMG_OPERATOR
  431. final_df = df.merge(video_df, on='video_id')
  432. # odps_instance = get_odps_instance(ODS_PROJECT)
  433. # odps_df = odps.DataFrame(df)
  434. # video_df = get_dataframe_from_odps('videoods', 'wx_video')
  435. # video_df['cover_url'] = video_df['cover_img_path'] + CDN_IMG_OPERATOR
  436. # video_df = video_df['id', 'title', 'cover_url']
  437. # final_df = odps_df.join(video_df, on=('video_id', 'id'))
  438. # final_df = final_df.to_pandas()
  439. final_df = final_df[['strategy_key', 'dt_version', 'gh_id', 'sort', 'video_id', 'title', 'cover_url', 'score']]
  440. return final_df
  441. def build_and_transfer_data(run_dt, run_hour, **kwargs):
  442. dt_version = f'{run_dt}{run_hour}'
  443. dry_run = kwargs.get('dry_run', False)
  444. mode = kwargs.get('mode')
  445. gh_df = get_and_update_gh_ids(run_dt)
  446. if mode == 'delta':
  447. return build_and_transfer_delta_mode(gh_df, dt_version, dry_run)
  448. else:
  449. return build_and_transfer_base_mode(gh_df, run_dt, run_hour, dt_version, dry_run)
  450. def main():
  451. LOGGER.info("%s 开始执行" % os.path.basename(__file__))
  452. LOGGER.info(f"environment: {CONFIG.ENV_TEXT}")
  453. argparser = ArgumentParser()
  454. argparser.add_argument('-n', '--dry-run', action='store_true')
  455. argparser.add_argument('--run-at',help='assume to run at date and hour, yyyyMMddHH')
  456. argparser.add_argument('--mode', default='base', choices=['base', 'delta'], help='run mode')
  457. args = argparser.parse_args()
  458. run_date = datetime.today()
  459. if args.run_at:
  460. run_date = datetime.strptime(args.run_at, "%Y%m%d%H")
  461. LOGGER.info(f"Assume to run at {run_date.strftime('%Y-%m-%d %H:00')}")
  462. try:
  463. now_date = datetime.today()
  464. LOGGER.info(f"开始执行: {datetime.strftime(now_date, '%Y-%m-%d %H:%M')}")
  465. last_date = run_date - timedelta(1)
  466. last_dt = last_date.strftime("%Y%m%d")
  467. # 查看当前天级更新的数据是否已准备好
  468. # 当前上游统计表为天级更新,但字段设计为兼容小时级
  469. while True:
  470. h_data_count = check_data_partition(ODS_PROJECT, GH_REPLY_STATS_TABLE, last_dt, '00')
  471. if h_data_count > 0:
  472. LOGGER.info('上游数据表查询数据条数={},开始计算'.format(h_data_count))
  473. run_dt = run_date.strftime("%Y%m%d")
  474. run_hour = run_date.strftime("%H")
  475. LOGGER.info(f'run_dt: {run_dt}, run_hour: {run_hour}')
  476. build_and_transfer_data(run_dt, run_hour, dry_run=args.dry_run, mode=args.mode)
  477. LOGGER.info('数据更新完成')
  478. return
  479. else:
  480. LOGGER.info("上游数据未就绪,等待60s")
  481. time.sleep(60)
  482. except Exception as e:
  483. LOGGER.error(f"数据更新失败, exception: {e}, traceback: {traceback.format_exc()}")
  484. if CONFIG.ENV_TEXT == '开发环境' or args.dry_run:
  485. return
  486. send_msg_to_feishu(
  487. webhook=CONFIG.FEISHU_ROBOT['growth_task_robot'].get('webhook'),
  488. key_word=CONFIG.FEISHU_ROBOT['growth_task_robot'].get('key_word'),
  489. msg_text=f"rov-offline{CONFIG.ENV_TEXT} - 数据更新失败\n"
  490. f"exception: {e}\n"
  491. f"traceback: {traceback.format_exc()}"
  492. )
  493. if __name__ == '__main__':
  494. main()