From 12b38b59e73f45c542d87bd2acf553f837e20179 Mon Sep 17 00:00:00 2001
From: Mole Shang <135e2@135e2.dev>
Date: Wed, 25 Jan 2023 16:40:48 +0800
Subject: feat: support showing mails list

---
 utils/client.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'utils')

diff --git a/utils/client.py b/utils/client.py
index e0bd2f9..676a859 100644
--- a/utils/client.py
+++ b/utils/client.py
@@ -34,9 +34,17 @@ class EmailClient(object):
 
     def get_mail_by_index(self, index):
         list = self.get_mails_list()
-        msg_data = list[int(index)-1][1]
+        msg_data = list[int(index) - 1][1]
         return Email(msg_data[b"RFC822"])
 
+    def get_listed_mails(self):
+        list = self.get_mails_list()
+        listed_mails = []
+        for i in range(0, self.get_mails_count()):
+            email = Email(list[i][1][b"RFC822"])
+            listed_mails.append([email.subject, email.sender, email.date])
+        return listed_mails
+
     def __enter__(self):
         return self
 
-- 
cgit v1.2.3