From d741c552a7dbe18515955bfc2c173367a03fdbc7 Mon Sep 17 00:00:00 2001 From: Mole Shang <135e2@135e2.dev> Date: Wed, 25 Jan 2023 11:37:45 +0800 Subject: fix(bot.py): rollback to the origin method to refresh inbox_num --- bot.py | 10 +++++----- utils/client.py | 4 ---- 2 files changed, 5 insertions(+), 9 deletions(-) diff --git a/bot.py b/bot.py index c533d51..deb8ebe 100644 --- a/bot.py +++ b/bot.py @@ -88,14 +88,14 @@ def periodic_task(context: CallbackContext) -> None: global inbox_num logger.info("entering periodic task.") with EmailClient(email_addr, email_passwd) as client: - new_unseen_num = client.get_mails_unseen_count() - if new_unseen_num: - for i in range(inbox_num, inbox_num + new_unseen_num + 1): - mail = client.get_mail_by_index(i) + new_inbox_num = client.get_mails_count() + if new_inbox_num > inbox_num: + for i in range(inbox_num, inbox_num + new_inbox_num): + mail = client.get_mail_by_index(i + 1) content = mail.__repr__() for text in handle_large_text(content): context.bot.send_message(context.job.context, text=text) - inbox_num += new_unseen_num + inbox_num = new_inbox_num def inbox(update: Update, context: CallbackContext) -> None: diff --git a/utils/client.py b/utils/client.py index 884a922..e0bd2f9 100644 --- a/utils/client.py +++ b/utils/client.py @@ -32,10 +32,6 @@ class EmailClient(object): def get_mails_count(self): return self.number - def get_mails_unseen_count(self): - unseen_messages = self.server.search("UNSEEN") - return len(unseen_messages) - def get_mail_by_index(self, index): list = self.get_mails_list() msg_data = list[int(index)-1][1] -- cgit v1.2.3