ck 4 年 前
コミット
4f7e81a2a2
1 ファイル変更18 行追加2 行削除
  1. 18 2
      handlers/PitcherPanelHandler.py

+ 18 - 2
handlers/PitcherPanelHandler.py

@@ -1,8 +1,12 @@
 from handlers.HandlerBase import BaseHandler
 from model.log import logger
 from data_manage.pitcher_panel import *
+import time
+import pandas as pd
+import json
 log = logger()
 
+
 class PitcherPanelChannel(BaseHandler):
 
     def post(self):
@@ -135,8 +139,20 @@ class ChannelSummary(BaseHandler):
         start=arg.get("start")
         end=arg.get("end")
         data,total,total_data=get_channel_summary(channel,pitcher,page,page_size,order_by,order,state,location,start,end)
-        self.write_json(data=data,total=total,total_data=total_data)
-
+        # print(data)
+        if arg.get("download"):
+            # self.set_header('Content-Type', 'application/x-xls')
+            # self.set_header('Content-Disposition',
+            #                 f'attachment; filename=公众号总数据_{time.time()}.xls')
+            self.set_header("Content-Type", "text/csv")
+            self.set_header('Cache-Control', 'public, max-age=4320000000')
+
+            df = pd.DataFrame(data).to_csv(encoding='utf-8')
+            # print(df)
+            self.write(df)
+
+        else:
+            self.write_json(data=data,total=total,total_data=total_data)
 
 class PitcherTrend(BaseHandler):
     def post(self):