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

correctly proccess sending msgs

Browse source
This commit is contained in:
Alexander Zveruk 2020-05-04 01:08:14 +03:00
parent 09f666f661
commit 8c3e17450a
2 changed files with 33 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tg/controllers/__init__.py
View file

@ -6,7 +6,7 @@ from utils import notify
log = logging.getLogger(__name__)
SUPPORTED_MSG_TYPES = "updateNewMessage", "updateChatLastMessage"
SUPPORTED_MSG_TYPES = "updateNewMessage", "updateChatLastMessage", "updateMessageSendSucceeded"
class Controller:
@ -150,7 +150,7 @@ class Controller:
if _type == 'updateNewMessage':
# with self.lock():
chat_id = update['message']['chat_id']
self.model.msgs.msgs[chat_id].append(update['message'])
self.model.msgs.add_message(chat_id, update['message'])
# msgs = self.model.get_current_chat_msg()
self.refresh_msgs()
if not update.get('disable_notification'):
@ -162,6 +162,12 @@ class Controller:
message = update['last_message']
self.model.chats.update_last_message(chat_id, message)
self.refresh_chats()
elif _type == "updateMessageSendSucceeded":
chat_id = update['message']['chat_id']
msg_id = update['old_message_id']
self.model.msgs.add_message(chat_id, update['message'])
self.model.msgs.remove_message(chat_id, msg_id)
self.refresh_msgs()
except Exception:
log.exception("Error happened in update_handler")

30
tg/models/__init__.py
View file

@ -173,6 +173,7 @@ class MsgModel:
self.tg = tg
self.msgs = defaultdict(list) # Dict[int, list]
self.current_msgs = defaultdict(int)
self.msg_ids = defaultdict(set)
def next_msg(self, chat_id, step=1):
current_msg = self.current_msgs[chat_id]
@ -200,6 +201,29 @@ class MsgModel:
return True
return False
def remove_message(self, chat_id, msg_id):
msg_set = self.msg_ids[chat_id]
if msg_id not in msg_set:
return False
log.info(f"removing msg {msg_id=}")
# FIXME: potential bottleneck, replace with constan time operation
self.msgs[chat_id] = [m for m in self.msgs[chat_id] if m['id'] != msg_id]
msg_set.remove(msg_id)
return True
def add_message(self, chat_id, message):
msg_id = message['id']
msg_set = self.msg_ids[chat_id]
if msg_id in msg_set:
return False
log.info(f"adding {msg_id=} {message}")
self.msgs[chat_id].append(message)
msg_set.add(msg_id)
return True
def add_messages(self, chat_id, messages):
return any(self.add_message(chat_id, msg) for msg in messages)
def fetch_msgs(self, chat_id, offset=0, limit=10):
if offset + limit < len(self.msgs[chat_id]):
return sorted(self.msgs[chat_id], key=lambda d: d['id'])[::-1][offset:limit]
@ -218,11 +242,7 @@ class MsgModel:
)
result.wait()
for msg in result.update['messages']:
self.msgs[chat_id].append(msg)
# TODO:
# if len(self.msgs[chat_id]) >= offset + limit:
# break
self.add_messages(chat_id, result.update['messages'])
return sorted(self.msgs[chat_id], key=lambda d: d["id"])[::-1][
offset:limit