74 lines
2.8 KiB
Python
74 lines
2.8 KiB
Python
from XiguaUser_pb2 import User as UserPb
|
|
|
|
|
|
class User:
|
|
def __init__(self, json=None):
|
|
self.ID = 0
|
|
self.name = ""
|
|
self.brand = ""
|
|
self.level = 0
|
|
self.type = 0
|
|
self.block = False
|
|
self.mute = False
|
|
if json:
|
|
if type(json) == bytes:
|
|
self.parsePb(json)
|
|
elif type(json) == UserPb:
|
|
self.parseUserPb(json)
|
|
else:
|
|
self.parse(json)
|
|
|
|
def parseUserPb(self, _user):
|
|
self.ID = _user.id
|
|
self.name = _user.nickname
|
|
self.brand = _user.fansClub.fansClub.title
|
|
self.level = _user.fansClub.fansClub.level
|
|
|
|
def parsePb(self, raw):
|
|
_user = UserPb()
|
|
_user.ParseFromString(raw)
|
|
self.parseUserPb(_user)
|
|
|
|
def parse(self, json):
|
|
if "extra" in json:
|
|
if "user" in json["extra"] and json["extra"]["user"] is not None:
|
|
self.ID = json["extra"]['user']['user_id']
|
|
self.name = json["extra"]['user']['name']
|
|
if "im_discipulus_info" in json["extra"] and json["extra"]["im_discipulus_info"] is not None:
|
|
self.level = json["extra"]["im_discipulus_info"]["level"]
|
|
self.brand = json["extra"]["im_discipulus_info"]["discipulus_group_title"]
|
|
if "user_room_auth_status" in json["extra"] and json["extra"]["user_room_auth_status"] is not None:
|
|
self.type = json["extra"]["user_room_auth_status"]["user_type"]
|
|
self.block = json["extra"]["user_room_auth_status"]["is_block"]
|
|
self.mute = json["extra"]["user_room_auth_status"]["is_silence"]
|
|
elif "user_info" in json and json["user_info"] is not None:
|
|
self.ID = json['user_info']['user_id']
|
|
self.name = json['user_info']['name']
|
|
elif "anchor" in json and json["anchor"] is not None:
|
|
if "user_info" in json["anchor"] and json["anchor"]['user_info'] is not None:
|
|
self.ID = json["anchor"]['user_info']['user_id']
|
|
self.name = json["anchor"]['user_info']['name']
|
|
if self.type is None:
|
|
self.type = 0
|
|
if isinstance(self.level, str):
|
|
self.level = int(self.level)
|
|
|
|
def __str__(self):
|
|
if self.level == 0:
|
|
if self.type == 1:
|
|
return "[房管]{}".format(self.name)
|
|
elif self.type == 3:
|
|
return "[主播]{}".format(self.name)
|
|
else:
|
|
return "{}".format(self.name)
|
|
else:
|
|
if self.type != 0:
|
|
return "[{}{}]{}".format(self.brand, self.level, self.name)
|
|
return "<{}{}>{}".format(self.brand, self.level, self.name)
|
|
|
|
def __unicode__(self):
|
|
return self.__str__()
|
|
|
|
def __repr__(self):
|
|
return "西瓜用户【{}(ID:{})】".format(self.name, self.ID)
|