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 type_id = 'h5_zone_admission' then '空间解锁'
				when type_id = 'h5_zone_moment' then '付费帖'
				when type_id = 'contact_wechat' then '个人微信'
				when 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()


s = S()
s.proc()