事件分析导出

This commit is contained in:
wuaho 2021-10-09 17:50:53 +08:00
parent b7462dd376
commit ed98584815

View File

@ -88,11 +88,14 @@ async def event_model_export(request: Request,
mime = mimetypes.guess_type(file_name)[0] mime = mimetypes.guess_type(file_name)[0]
excels = [] excels = []
for item in sqls: for item in sqls:
if not item.get('combination_event'): if item.get('combination_event'):
continue continue
sql = item['sql'] sql = item['sql']
event_name = item['event_name'] event_name = item['event_name']
df = await ckdb.query_dataframe(sql) df = await ckdb.query_dataframe(sql)
if 'date' in df:
df.sort_values('date',inplace=True)
df['date'] = df['date'].dt.tz_localize(None)
excels.append((df, event_name)) excels.append((df, event_name))
df_to_stream = DfToStream(*excels) df_to_stream = DfToStream(*excels)
with df_to_stream as d: with df_to_stream as d: