Mirrors/tg
2
0
Fork 0
mirror of https://github.com/paul-nameless/tg synced 2024-11-22 11:53:08 +00:00
Code Issues Projects Releases Packages Wiki Activity

Implement messageChatAddMembers, messageChatDeleteMember, messageBasicGroupChatCreate (#160)

Browse source 
* Implement messageChatAddMembers, messageChatDeleteMember, messageBasicGroupChatCreate

* Fix sort imports
This commit is contained in:
Nameless 2020-07-23 16:48:41 +08:00 committed by GitHub
parent 50b93eb308
commit bb2d9b889f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
tg/views.py
View file

@ -216,7 +216,7 @@ class ChatView:
def _get_last_msg_data( def _get_last_msg_data(
self, chat: Dict[str, Any] self, chat: Dict[str, Any]
) -> Tuple[Optional[str], Optional[str]]: ) -> Tuple[Optional[str], Optional[str]]:
user, last_msg = get_last_msg(chat) user, last_msg = get_last_msg(chat, self.model.users)
if user: if user:
last_msg_sender = self.model.users.get_user_label(user) last_msg_sender = self.model.users.get_user_label(user)
chat_type = get_chat_type(chat) chat_type = get_chat_type(chat)
@ -528,17 +528,21 @@ class MsgView:
def _parse_msg(self, msg: MsgProxy) -> str: def _parse_msg(self, msg: MsgProxy) -> str:
if msg.is_message: if msg.is_message:
return parse_content(msg["content"]) return parse_content(msg, self.model.users)
log.debug("Unknown message type: %s", msg) log.debug("Unknown message type: %s", msg)
return "unknown msg type: " + str(msg["content"]) return "unknown msg type: " + str(msg["content"])
def get_last_msg(chat: Dict[str, Any]) -> Tuple[Optional[int], str]: def get_last_msg(
chat: Dict[str, Any], users: UserModel
) -> Tuple[Optional[int], str]:
last_msg = chat.get("last_message") last_msg = chat.get("last_message")
if not last_msg: if not last_msg:
return None, "<No messages yet>" return None, "<No messages yet>"
content = last_msg["content"] return (
return last_msg["sender_user_id"], parse_content(content) last_msg["sender_user_id"],
parse_content(MsgProxy(last_msg), users),
)
def get_date(chat: Dict[str, Any]) -> str: def get_date(chat: Dict[str, Any]) -> str:
@ -554,17 +558,31 @@ def get_date(chat: Dict[str, Any]) -> str:
return dt.strftime(date_fmt) return dt.strftime(date_fmt)
def parse_content(content: Dict[str, Any]) -> str: def parse_content(msg: MsgProxy, users: UserModel) -> str:
msg = MsgProxy({"content": content})
if msg.is_text: if msg.is_text:
return content["text"]["text"].replace("\n", " ") return msg.text_content.replace("\n", " ")
content = msg["content"]
_type = content["@type"] _type = content["@type"]
if _type == "messageBasicGroupChatCreate": if _type == "messageBasicGroupChatCreate":
return "[created the group]" return f"[created the group \"{content['title']}\"]"
if _type == "messageChatAddMembers": if _type == "messageChatAddMembers":
return "[joined the group]" user_ids = content["member_user_ids"]
if user_ids[0] == msg.sender_id:
return "[joined the group]"
users_name = ", ".join(
users.get_user_label(user_id) for user_id in user_ids
)
return f"[added {users_name}]"
if _type == "messageChatDeleteMember":
user_id = content["user_id"]
if user_id == msg.sender_id:
return "[left the group]"
user_name = users.get_user_label(user_id)
return f"[removed {user_name}]"
if _type == "messageChatChangeTitle":
return f"[changed the group name to \"{content['title']}\"]"
if not msg.content_type: if not msg.content_type:
# not implemented # not implemented