from typing import List from lib.all import * # Assume these classes exist and mimic the Go structs class ConsumeHistory: def __init__(self, ct=None, change=None, s_type=None, type_id=None, uid=None): self.ct = ct self.change = change self.s_type = s_type self.type_id = type_id self.uid = uid def GetChange(self): return self.change def GetSType(self): return self.s_type def GetTypeId(self): return self.type_id def GetCt(self): return self.ct def GetUid(self): return self.uid class CHSt: def __init__(self, ct=None, desc=None, change=None): self.ct = ct self.desc = desc self.change = change # Assume these constants are defined similarly to the Go constants CHTypeCost = 1 # 消费明细(金币) CHTypeCharge = 2 # 充值明细(金币) CHTypeIncome = 3 # 收入明细(钻石) CHTypeWithdraw = 4 # 提现明细(钻石) CHSTypeCostContact = 10001 # 消费明细，联系方式 CHSTypeCostRefundContactWechat = 10002 # 消费明细，微信联系方式退款 CHSTypeCostMembership = 10003 # 消费明细，会员资格解锁(伪金币记录，会员资格解锁中间无转金币过程) CHSTypeCostRefundMembership = 10004 # 消费明细，会员资格解锁退款(伪金币记录，会员资格解锁中间无转金币过程) CHSTypeChargeUser = 20001 # 充值明细，用户自己冲 CHSTypeChargeOp = 20002 # 充值明细，OP充值 CHSTypeChargeRefundCoins = 20003 # 充值明细，金币退款 CHSTypeChargeRefundMembership = 20004 # 充值明细，会员退款 CHSTypeChargeMembership = 20005 # 充值明细，会员充值 CHSTypeChargeRefundContactWechat = 20006 # 充值明细，微信金币退款 CHSTypeChargeZoneMoment = 20007 # 充值明细，动态解锁 CHSTypeChargeZoneAdmission = 20008 # 充值明细，空间会员 CHSTypeChargeZoneSuperfanship = 20009 # 充值明细，空间超粉 CHSTypeChargeZoneRefundAdmission = 20010 # 充值明细，空间普通会员退款 CHSTypeChargeZoneRefundMoment = 20011 # 充值明细，空间动态退款 CHSTypeChargeZoneRefundSuperfanship = 20012 # 充值明细，空间动态退款 CHSTypeIncomeContact = 30001 # 收入明细，联系方式 CHSTypeIncomeInvite = 30002 # 收入明细，邀请分成 CHSTypeIncomeRefundMembership = 30003 # 收入明细，会员退款 CHSTypeIncomeRefundContactWechat = 30004 # 收入明细，微信退款 CHSTypeIncomeThirdPartner = 30005 # 收入明细，代运营 CHSTypeIncomeCollaborator = 30006 # 收入明细，协作者 CHSTypeIncomeZoneStreamer = 30007 # 收入明细，主播空间收益 CHSTypeIncomeMembership = 30008 # 收入明细，会员 CHSTypeIncomeRefundZoneAdmission = 30009 # 收入明细，空间普通会员分成退款 CHSTypeIncomeRefundThirdPartner = 30010 # 收入明细，代运营 CHSTypeIncomeRefundCollaborator = 30011 # 收入明细，协作者 CHSTypeIncomeRefundZoneStreamer = 30012 # 收入明细，主播空间收益 CHSTypeWithdrawDiamondAuto = 40001 # 自动提现明细 CHSTypeWithdrawDiamondHvyogo = 40002 # 慧用工提现明细 CHSTypeWithdrawDiamondReversal = 40003 # 提现冲正明细 ProductIdOpCoin = "op_coin" # op充值的金币 ProductIdH5CustomCoin = "h5_custom_coin" # H5 自定义金币 ProductIdH5Coin500 = "h5_coin_500" # H5 500币 ProductIdH5Coin1000 = "h5_coin_1000" # H5 1000币 ProductIdH5Coin3000 = "h5_coin_3000" # H5 3000币 ProductIdH5Coin5000 = "h5_coin_5000" # H5 5000币 ProductIdH5Coin8000 = "h5_coin_8000" # H5 8000币 ProductIdH5Coin10000 = "h5_coin_10000" # H5 10000币 ProductIdH5Coin15000 = "h5_coin_15000" # H5 15000币 ProductIdH5Coin18000 = "h5_coin_18000" # H5 18000币 ProductIdH5Coin20000 = "h5_coin_20000" # H5 20000币 ProductIdIosCoin70 = "70_gold" ProductIdIosCoin350 = "350_gold" ProductIdIosCoin686 = "686_gold" ProductIdIosCoin2086 = "2086_gold" ProductIdIosCoin4886 = "4886_gold" ProductIdIosCoin6986 = "6986_gold" ProductIdIosCoin9086 = "9086_gold" ProductIdIosCoin13986 = "13986_gold" ProductIdIosCoin20986 = "20986_gold" ProductIdContactWechat = "contact_wechat" # 微信联系方式 ProductIdH5ContactWechat = "h5_contact_wechat" # h5的联系方式，rmb直接解锁 ProductIdSuperfanGiftContactWechat = "superfan_gift_contact_wechat" # 超粉赠送的联系方式 ProductIdMembership = "membership" # 会员 ProductIdH5Membership = "h5_membership" # 会员 ProductIdH5ZoneMoment = "h5_zone_moment" # 空间动态 ProductIdH5ZoneAdmission = "h5_zone_admission" # 空间普通会员 ProductIdH5ZoneIronfanship = "h5_zone_ironfanship" # 空间铁粉 ProductIdH5ZoneSuperfanship = "h5_zone_superfanship" # 空间超粉 ProductIdOuterRaven = "outer_raven" # 瑞文测试 def get_type_id_desc(type_id): # Placeholder function return f" (Type ID: {type_id})" def ch_list_income(ch_list: List[ConsumeHistory]) -> List[CHSt]: """ Translates the provided Go code snippet to Python. Processes a list of ConsumeHistory objects and returns a list of CHSt objects. """ list_chst = [] for ch_db in ch_list: item = CHSt(ct=ch_db.ct) change_mark = "-" if ch_db.GetChange() >= 0: change_mark = "+" if ch_db.GetSType() == CHSTypeIncomeContact: item.desc = "解锁微信" item.change = change_mark + f"{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeInvite: item.desc = "邀请收益" item.change = change_mark + f"{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeRefundMembership: item.desc = "用户会员退款" item.change = f"{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeRefundContactWechat: item.desc = "用户微信退款" item.change = f"{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeThirdPartner: item.desc = "代运营收益" + get_type_id_desc(ch_db.GetTypeId()) if ch_db.GetTypeId() == ProductIdH5ZoneAdmission: item.desc = "空间成员-代运营" elif ch_db.GetTypeId() == ProductIdH5ZoneMoment: item.desc = "空间动态-代运营" elif ch_db.GetTypeId() == ProductIdH5ZoneSuperfanship: item.desc = "空间超粉-代运营" elif ch_db.GetTypeId() == ProductIdContactWechat: if ch_db.GetCt() >= 1722497400: item.desc = f"解锁微信({ch_db.GetUid()})-代运营" elif ch_db.GetTypeId() == ProductIdSuperfanGiftContactWechat: if ch_db.GetCt() >= 1722497400: item.desc = f"超粉送微信({ch_db.GetUid()})-代运营" item.change = change_mark + f"{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeCollaborator: item.desc = "协作收益" + get_type_id_desc(ch_db.GetTypeId()) if ch_db.GetTypeId() == ProductIdH5ZoneAdmission: item.desc = "空间成员-合伙人" elif ch_db.GetTypeId() == ProductIdH5ZoneMoment: item.desc = "空间动态-合伙人" elif ch_db.GetTypeId() == ProductIdH5ZoneSuperfanship: item.desc = "空间超粉-合伙人" elif ch_db.GetTypeId() == ProductIdContactWechat: if ch_db.GetCt() >= 1722497400: item.desc = f"解锁微信({ch_db.GetUid()})-合伙人" elif ch_db.GetTypeId() == ProductIdSuperfanGiftContactWechat: if ch_db.GetCt() >= 1722497400: item.desc = f"超粉送微信({ch_db.GetUid()})-合伙人" item.change = change_mark + f"{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeZoneStreamer: item.desc = "空间收益" + get_type_id_desc(ch_db.GetTypeId()) if ch_db.GetTypeId() == ProductIdH5ZoneAdmission: item.desc = "空间成员" elif ch_db.GetTypeId() == ProductIdH5ZoneMoment: item.desc = "空间动态" elif ch_db.GetTypeId() == ProductIdH5ZoneSuperfanship: item.desc = "空间超粉" item.change = change_mark + f"{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeMembership: item.desc = "永久会员收益" + get_type_id_desc(ch_db.GetTypeId()) item.change = change_mark + f"{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeRefundZoneAdmission: item.desc = "(旧)空间退款" + get_type_id_desc(ch_db.GetTypeId()) item.change = f"{ch_db.GetChange()}钻石" if ch_db.GetChange() == 0: # This logic seems unusual, but it's from the original code item.change = f"-{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeRefundThirdPartner: item.desc = "代运营空间退款" + get_type_id_desc(ch_db.GetTypeId()) if ch_db.GetTypeId() == ProductIdH5ZoneAdmission: item.desc = "空间成员退款-代运营" elif ch_db.GetTypeId() == ProductIdH5ZoneMoment: item.desc = "空间动态退款-代运营" elif ch_db.GetTypeId() == ProductIdH5ZoneSuperfanship: item.desc = "空间超粉退款-代运营" item.change = f"{ch_db.GetChange()}钻石" if ch_db.GetChange() == 0: item.change = f"-{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeRefundCollaborator: item.desc = "合伙人空间退款" + get_type_id_desc(ch_db.GetTypeId()) if ch_db.GetTypeId() == ProductIdH5ZoneAdmission: item.desc = "空间成员退款-合伙人" elif ch_db.GetTypeId() == ProductIdH5ZoneMoment: item.desc = "空间动态退款-合伙人" elif ch_db.GetTypeId() == ProductIdH5ZoneSuperfanship: item.desc = "空间动态退款-合伙人" # This seems like a potential typo (should it be 超粉?) item.change = f"{ch_db.GetChange()}钻石" if ch_db.GetChange() == 0: item.change = f"-{ch_db.GetChange()}钻石" elif ch_db.GetSType() == CHSTypeIncomeRefundZoneStreamer: item.desc = "空间退款" + get_type_id_desc(ch_db.GetTypeId()) if ch_db.GetTypeId() == ProductIdH5ZoneAdmission: item.desc = "用户退款-空间成员" elif ch_db.GetTypeId() == ProductIdH5ZoneMoment: item.desc = "用户退款-空间动态" elif ch_db.GetTypeId() == ProductIdH5ZoneSuperfanship: item.desc = "用户退款-空间超粉" item.change = f"{ch_db.GetChange()}钻石" if ch_db.GetChange() == 0: item.change = f"-{ch_db.GetChange()}钻石" list_chst.append(item) return list_chst 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() def proc(self, mid, st, et): sql = "select * from vas_ch_income where mid=%d and ct>=%d and ct<%d" % (mid, st, et) docs = self.mysql_db_vas.query(sql) ch_list_db = list() for doc in docs: # print(doc) ch = ConsumeHistory( safe_get_int(doc, "ct"), safe_get_int(doc, "change"), safe_get_int(doc, "stype"), safe_get_str(doc, "type_id"), safe_get_int(doc, "uid") ) ch_list_db.append(ch) ch_list = ch_list_income(ch_list_db) csv_w = Csv("231638_12月_主播收益_.csv", ["时间", "钻石变化", "描述"]) for ch in ch_list: d = [ get_time_str_by_ts(ch.ct), ch.change, ch.desc, ] csv_w.append([d]) s = S() s.proc( 303906, get_ts_by_str("2024-12-01 00:00:00"), get_ts_by_str("2025-01-01 00:00:00"), )