from lib.all import *

service_name = 'fix_diamonds'
log_dir = '/app/log'
logger = Logger(service_name, log_dir=log_dir)


class S:
    def __init__(self):
        self.mysql_db_vas = Mysql(
            "rm-bp11t1616a1kjvmx5.mysql.rds.aliyuncs.com", 3306, "vas", "root", "Wishpal2024"
        )

    def __del__(self):
        self.mysql_db_vas.close()

    # 获取mid: 钻石
    def get_mid_dias(self):
        sql = "select mid, sum(`change`) as dias  from vas_ch_income group by mid"
        docs = self.mysql_db_vas.query(sql)
        logger.Info("get_mid_dias docs: {}".format(docs))
        return docs

    # 获取mid: 已提现的钻石
    def get_mid_wdias(self):
        sql = "select mid, sum(`withdraw_dias`) as wdias from vas_withdraw_order group by mid"
        docs = self.mysql_db_vas.query(sql)
        logger.Info("get_mid_wdias docs: {}".format(docs))
        return docs

    def proc(self):
        docs_dias = self.get_mid_dias()
        docs_wdias = self.get_mid_wdias()


s = S()
s.proc()