cola 2 年之前
父节点
当前提交
b3b2f9c238
共有 1 个文件被更改,包括 6 次插入2 次删除
  1. 6 2
      handlers/HandlerBase.py

+ 6 - 2
handlers/HandlerBase.py

@@ -7,6 +7,7 @@ import time
 import base64
 import pandas as pd
 from model.DateUtils import DateUtils
+from pandas.io.excel import ExcelWriter
 
 log = logger()
 
@@ -49,15 +50,18 @@ class BaseHandler(RequestHandler, DateUtils):
 
     def write_download(self, filename, data):
         self.set_header('Content-Type', 'application/octet-stream')
-        self.set_header('Content-Disposition', f'attachment; filename={filename}.xlsx')
+        self.set_header('Content-Disposition', f'attachment; filename={filename}.csv')
         self.set_header("Pargam", "no-cache")
         self.set_header("Cache-Control", "no-cache")
-        df = pd.DataFrame(data).to_excel(encoding='utf-8')
+        df = pd.DataFrame(data).to_csv(encoding='utf-8')
         # print(df)
         # with open(f'./{pitcher}_{start}_{end}.csv','w',newline='') as f:
         #     f.write(df)
+        with ExcelWriter({filename}.xlsx) as ew:
+            pd.read_csv(df).to_excel(ew,sheet_name="1",index=False)
         self.write(df)
 
+
     def get_args(self):
         di = json.loads(self.request.body.decode(encoding='utf-8'))
         if isinstance(di, str):