Преглед изворни кода

导出的excel文档新增video_url

liyuyi@c-top.com.cn пре 4 година
родитељ
комит
3e86f6c7cc
1 измењених фајлова са 8 додато и 6 уклоњено
  1. 8 6
      get_high_quality_script.py

+ 8 - 6
get_high_quality_script.py

@@ -97,15 +97,15 @@ if __name__ == '__main__':
                 charge_script_df = script_df.merge(merge_df, on='signature', how='inner')
                 charge_script_df.rename(columns={'charge': '消耗', 'script': '脚本'}, inplace=True)
 
-                charge_script_df[charge_script_df.activation >= 50][['消耗', '脚本']]. \
+                charge_script_df[charge_script_df.activation >= 50][['消耗', '脚本', 'video_url']]. \
                     to_excel(writer, sheet_name='汇创思拓_%s项目近一周高质量素材脚本' % project_name, index=False, header=True)
-                charge_script_df[charge_script_df.activation <= 10][['消耗', '脚本']]. \
+                charge_script_df[charge_script_df.activation <= 10][['消耗', '脚本', 'video_url']]. \
                     to_excel(writer, sheet_name='汇创思拓_%s项目近一周低质量素材脚本' % project_name, index=False, header=True)
 
             if 0 in channel:
                 logger.info("项目:%s, 巨量引擎外部素材,开始执行!" % project_name)
-                sql = """select signature  from ctop_ai_video_info_from_ocean_engine where project_name = '%s' and stat_date >= '%s'""" \
-                      % (project_name, one_week_date)
+                sql = """select signature, video_url  from ctop_ai_video_info_from_ocean_engine 
+                where project_name = '%s' and stat_date >= '%s'""" % (project_name, one_week_date)
                 df = pd.read_sql(sql, test_read_engine)
                 logger.info("项目:%s, 巨量引擎优质素材个数为 %s!" % (project_name, len(df)))
 
@@ -115,8 +115,10 @@ if __name__ == '__main__':
                 script_df.drop_duplicates('signature', keep='first', inplace=True)
                 script_df['script'] = script_df['task_result'].apply(lambda x: eval(x)['Data']['Result'].split(']')[1].strip())
                 script_df.rename(columns={'script': '脚本'}, inplace=True)
-
-                script_df[['脚本']].to_excel(writer, sheet_name='巨量引擎_%s项目近一周高质量素材脚本' % project_name, index=False, header=True)
+                merge_df = script_df.merge(df, on='signature', how='inner')
+                merge_df[['脚本', 'video_url']].to_excel(writer, sheet_name='巨量引擎_%s项目近一周高质量素材脚本' % project_name,
+                                                       index=False,
+                                                       header=True)
             writer.save()
             logger.info("项目: %s,文件导出完成!" % project_name)
         except: