From 4b1298d0c28b25c473ef96fee92dc1803e6880d5 Mon Sep 17 00:00:00 2001
From: lwl0608 <lwl0608@foxmail.com>
Date: Mon, 1 Jul 2024 14:37:05 +0800
Subject: [PATCH] fix blockrev

---
 cronjob/vas/user_income_calc.py | 77 +++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)
 create mode 100644 cronjob/vas/user_income_calc.py

diff --git a/cronjob/vas/user_income_calc.py b/cronjob/vas/user_income_calc.py
new file mode 100644
index 0000000..2bbfe31
--- /dev/null
+++ b/cronjob/vas/user_income_calc.py
@@ -0,0 +1,77 @@
+from lib.all import *
+
+
+class S:
+    def __init__(self):
+        self.mysql_db_vas = Mysql(
+            "rm-bp11t1616a1kjvmx5.mysql.rds.aliyuncs.com", 3306, "vas", "root", "Wishpal2024"
+        )
+        self.col_user_income = MongoDB(
+            host="mongodb://root:Wishpal2024@dds-bp1da1ddd62bede41.mongodb.rds.aliyuncs.com:3717,dds-bp1da1ddd62bede42.mongodb.rds.aliyuncs.com:3717/admin?replicaSet=mgset-77304659",
+            port=3717,
+            db="vas",
+            collection="user_income"
+        )
+
+    def __del__(self):
+        self.mysql_db_vas.close()
+        self.col_user_income.close()
+
+    # 获取每日收益
+    def get_week_dashboard(self):
+        et = get_today_zero_time()
+        st = et - 7 * 86400
+        sql = '''
+select t1.mid,t1.ymd,sum(t1.`change`) income
+from
+(
+	select mid,`change`,DATE_FORMAT(FROM_UNIXTIME(ct), '%Y%m%d') as ymd from vas_ch_income where ct>={} and ct<{} and mid>0
+) t1
+group by t1.mid,t1.ymd
+        '''.format(st, et)
+        docs = self.mysql_db_vas.query(sql)
+        print("week, len(docs): {}".format(len(docs)))
+
+        mid_map = dict()
+        for doc in docs:
+            mid = safe_get_int(doc, "mid")
+            mid_map[mid] = 1
+        mids = list(mid_map.keys())
+        print("week, len(mids): {}".format(len(mids)))
+
+    # 获取收益来源
+    def get_income_from_dashboard(self):
+        et = get_today_zero_time()
+        st = et - 7 * 86400
+        sql = '''
+select  t1.mid,
+		t1.type_id,
+		sum(t1.`change`) income
+from
+(
+	select 	mid,
+			case
+				when t1.type_id = 'h5_zone_admission' then '空间解锁'
+				when t1.type_id = 'h5_zone_moment' then '付费帖'
+				when t1.type_id = 'contact_wechat' then '个人微信'
+				when t1.type_id = 'h5_zone_superfanship' then '超粉解锁'
+				ELSE '其他' 
+			END AS type_id,
+			`change` 
+	from vas_ch_income where mid=182308 and ct>=1719158400 and ct<1719763200
+) t1
+group by t1.mid,t1.type_id
+        '''.format(st, et)
+        docs = self.mysql_db_vas.query(sql)
+        print("from, len(docs): {}".format(len(docs)))
+
+        mid_map = dict()
+        for doc in docs:
+            mid = safe_get_int(doc, "mid")
+            mid_map[mid] = 1
+        mids = list(mid_map.keys())
+        print("from, len(mids): {}".format(len(mids)))
+
+    def proc(self):
+        self.get_week_dashboard()
+        self.get_income_from_dashboard()