From 07aa025947a366bee0202f2751673ada718d6059 Mon Sep 17 00:00:00 2001
From: wuaho <wuaho@legu.cc>
Date: Fri, 30 Jul 2021 16:25:39 +0800
Subject: [PATCH] =?UTF-8?q?ltv=20=E5=A4=9A=E7=BB=B4=E5=BA=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/api_v1/endpoints/xquery.py | 5 +++++
 models/x_analysis.py           | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/api/api_v1/endpoints/xquery.py b/api/api_v1/endpoints/xquery.py
index 2bfb6ad..e6414ac 100644
--- a/api/api_v1/endpoints/xquery.py
+++ b/api/api_v1/endpoints/xquery.py
@@ -46,9 +46,14 @@ async def ltv_model_sql(
 
     res = analysis.ltv_model_sql()
     sql = res['sql']
+    quota = res['quota']
     df = await ckdb.query_dataframe(sql)
     df.fillna(0, inplace=True)
     df.rename(columns={'date': '注册日期', 'cnt1': '角色数'}, inplace=True)
+    if quota == '#account_id':
+        df.rename(columns={'cnt1': '角色数'}, inplace=True)
+    elif quota in ('#account_id'):
+        df.rename(columns={'cnt1': '设备数'}, inplace=True)
     data = {
         'title': df.columns.tolist(),
         'rows': df.values.tolist()
diff --git a/models/x_analysis.py b/models/x_analysis.py
index c82fd9d..373c905 100644
--- a/models/x_analysis.py
+++ b/models/x_analysis.py
@@ -126,4 +126,4 @@ class XAnalysis:
         order by date
         """
         print(sql)
-        return {'sql': sql}
+        return {'sql': sql,'quota':quota}