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("352039_11月_主播收益_.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(
31476,
get_ts_by_str("2024-11-01 00:00:00"),
get_ts_by_str("2024-12-01 00:00:00"),
)