mirror of
https://github.com/LC044/WeChatMsg
synced 2024-11-15 06:11:19 +08:00
103 lines
2.8 KiB
Python
103 lines
2.8 KiB
Python
import os.path
|
|
from typing import Dict
|
|
|
|
from PyQt5.QtCore import Qt
|
|
from PyQt5.QtGui import QPixmap
|
|
|
|
from app.DataBase import data
|
|
from app.Ui.Icon import Icon
|
|
|
|
|
|
# from app.Ui.Icon import Icon
|
|
|
|
|
|
class Person:
|
|
def __init__(self, wxid: str):
|
|
|
|
self.wxid = wxid
|
|
self.conRemark = data.get_conRemark(wxid)
|
|
self.nickname, self.alias = data.get_nickname(wxid)
|
|
self.avatar_path = data.get_avator(wxid)
|
|
if os.path.exists(self.avatar_path):
|
|
self.avatar = QPixmap(self.avatar_path).scaled(60, 60)
|
|
else:
|
|
self.avatar_path = './app/data/icons/default_avatar.svg'
|
|
# self.avatar_path = Icon.Default_avatar_path
|
|
self.avatar = QPixmap(self.avatar_path).scaled(60, 60)
|
|
|
|
|
|
class Me(Person):
|
|
def __init__(self, wxid: str):
|
|
super(Me, self).__init__(wxid)
|
|
self.city = None
|
|
self.province = None
|
|
|
|
|
|
class Contact(Person):
|
|
def __init__(self, wxid: str):
|
|
super(Contact, self).__init__(wxid)
|
|
self.smallHeadImgUrl = ''
|
|
self.bigHeadImgUrl = ''
|
|
|
|
|
|
def singleton(cls):
|
|
_instance = {}
|
|
|
|
def inner():
|
|
if cls not in _instance:
|
|
_instance[cls] = cls()
|
|
return _instance[cls]
|
|
|
|
return inner
|
|
|
|
|
|
@singleton
|
|
class MePC:
|
|
def __init__(self):
|
|
self.avatar = QPixmap(Icon.Default_avatar_path)
|
|
|
|
def set_avatar(self, img_bytes):
|
|
if not img_bytes:
|
|
self.avatar.load(Icon.Default_avatar_path)
|
|
return
|
|
if img_bytes[:4] == b'\x89PNG':
|
|
self.avatar.loadFromData(img_bytes, format='PNG')
|
|
else:
|
|
self.avatar.loadFromData(img_bytes, format='jfif')
|
|
self.avatar = QPixmap()
|
|
|
|
|
|
class ContactPC:
|
|
def __init__(self, contact_info: Dict):
|
|
self.wxid = contact_info.get('UserName')
|
|
self.remark = contact_info.get('Remark')
|
|
# Alias,Type,Remark,NickName,PYInitial,RemarkPYInitial,ContactHeadImgUrl.smallHeadImgUrl,ContactHeadImgUrl,bigHeadImgUrl
|
|
self.alias = contact_info.get('Alias')
|
|
self.nickName = contact_info.get('NickName')
|
|
if not self.remark:
|
|
self.remark = self.nickName
|
|
self.smallHeadImgUrl = contact_info.get('smallHeadImgUrl')
|
|
self.smallHeadImgBLOG = b''
|
|
self.avatar = QPixmap()
|
|
|
|
def set_avatar(self, img_bytes):
|
|
if not img_bytes:
|
|
self.avatar.load(Icon.Default_avatar_path)
|
|
return
|
|
if img_bytes[:4] == b'\x89PNG':
|
|
self.avatar.loadFromData(img_bytes, format='PNG')
|
|
else:
|
|
self.avatar.loadFromData(img_bytes, format='jfif')
|
|
self.avatar.scaled(60, 60, Qt.IgnoreAspectRatio, Qt.SmoothTransformation)
|
|
|
|
|
|
class Group(Person):
|
|
def __init__(self, wxid: str):
|
|
super(Group, self).__init__(wxid)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
p1 = MePC()
|
|
p2 = MePC()
|
|
print(p1 == p2)
|