feat: 添加MailboxGUI

mailbox-cli.py

@@ -160,11 +160,11 @@ Sure? [Yes/No]""")
             return
 
         if self.imap:
-            print(f"Host: {self.imap.host}:{self.imap.port}")
-            print(f"User: {self.imap.user}")
-            print(f"Passwd: {self.imap.passwd}")
-            print(f"SSL: {self.imap.ssl}")
-            print(f"MailBox: {self.imap.inbox}")
+            print(f"""Host: {self.imap.host}:{self.imap.port}
+User: {self.imap.user}
+Passwd: {self.imap.passwd}
+SSL: {self.imap.ssl}
+MailBox: {self.imap.inbox}""")
         else:
             print("Not login.")
 

mailbox-tk.py

@@ -0,0 +1,495 @@
+import tkinter as tk
+import tkinter.messagebox as msg
+import tkinter.filedialog as fd
+import tkinter.scrolledtext as st
+import json
+import imaplib
+from typing import Optional, List
+from time import strftime, localtime, strptime
+
+from mailbox.email import Mail
+from mailbox.imap import Imap
+
+# 顶层Frame
+FRAME_PADX = 5
+
+# 顶层窗口按钮
+BUTTON_WIDTH = 40
+BUTTON_WIDTH_L = 20
+BUTTON_HEIGHT = 1
+MAILBOX_WIDTH = 60
+MAILBOX_HEIGHT = 5
+
+# 登录框
+LOGIN_LABEL_WIDTH = 10
+LOGIN_LABEL_HEIGHT = 1
+LOGIN_ENTRY_WIDTH = 35
+LOGIN_BTN_WIDTH = 20
+LOGIN_RBTN_WIDTH = 6
+LOGIN_MAILBOX_WIDTH = 45
+LOGIN_MAILBOX_HEIGHT = 10
+
+# GET框
+GET_LABEL_WIDTH = 10
+GET_LABEL_HEIGHT = 1
+GET_ENTRY_WIDTH = 35
+
+# refresh框
+REFRESH_LABEL_WIDTH = 10
+REFRESH_LABEL_HEIGHT = 1
+REFRESH_ENTRY_WIDTH = 35
+
+# 邮件信息显示框
+EMAIL_INFO_BUTTON_WIDTH = 20
+EMAIL_INFO_BUTTON_HEIGHT = 1
+EMAIL_INFO_WIDTH = 100
+EMAIL_INFO_HEIGHT = 40
+
+
+class MailboxGUI(tk.Tk):
+    def __init__(self):
+        super().__init__()
+
+        self.imap: Optional[Imap] = None
+
+        self.iconbitmap('HuanMail.ico')
+        self.title("HuanMail (IMAP)")
+        self.resizable(False, False)
+
+        self.btn_frame = tk.Frame(self)
+        self.mailbox_frame = tk.Frame(self)
+        self.btn_frame.grid(column=0, row=0)
+        self.mailbox_frame.grid(column=1, row=0)
+
+        self.mailbox_num: List[str] = []
+        self.mailbox_x = tk.Scrollbar(self.mailbox_frame, orient=tk.HORIZONTAL)
+        self.mailbox_y = tk.Scrollbar(self.mailbox_frame)
+        self.mailbox = tk.Listbox(self.mailbox_frame,
+                                  xscrollcommand=self.mailbox_x.set, yscrollcommand=self.mailbox_y.set,
+                                  width=MAILBOX_WIDTH, height=MAILBOX_HEIGHT)
+        self.mailbox_x.config(command=self.mailbox.xview)
+        self.mailbox_y.config(command=self.mailbox.yview)
+
+        self.mailbox.grid(column=0, row=0, ipady=38)
+        self.mailbox_x.grid(column=0, row=1, sticky="NEWS")
+        self.mailbox_y.grid(column=1, row=0, sticky="NEWS")
+
+        self.login_frame = tk.Frame(self.btn_frame)
+        self.mailbox_frame = tk.Frame(self.btn_frame)
+        self.check_frame = tk.Frame(self.btn_frame)
+        self.config_frame = tk.Frame(self.btn_frame)
+        self.quit_frame = tk.Frame(self.btn_frame)
+
+        for i, frame in enumerate([self.login_frame, self.mailbox_frame, self.check_frame,
+                                   self.config_frame, self.quit_frame]):
+            frame.grid(column=0, row=i, padx=FRAME_PADX)
+
+        button_arg = dict(width=BUTTON_WIDTH_L, height=BUTTON_HEIGHT, border=2)
+        button_big_arg = dict(width=BUTTON_WIDTH, height=BUTTON_HEIGHT, border=2)
+
+        self.login = tk.Button(self.login_frame, text="Login", command=self.show_login, **button_arg)
+        self.info = tk.Button(self.login_frame, text="Info", command=self.show_info, **button_arg)
+
+        self.get = tk.Button(self.mailbox_frame, text="Get", command=self.show_get, **button_arg)
+        self.refresh = tk.Button(self.mailbox_frame, text="Refresh", command=self.show_refresh, **button_arg)
+
+        self.check = tk.Button(self.check_frame, text="Check", command=self.show_email, **button_big_arg)
+        self.check.pack(side="top", pady=2.5, ipadx=8)
+
+        self.load = tk.Button(self.config_frame, text="Load", command=self.load_config, **button_arg)
+        self.save = tk.Button(self.config_frame, text="Save", command=self.save_config, **button_arg)
+
+        for i in (self.login, self.info, self.get, self.refresh, self.load, self.save):
+            i.pack(side="left", pady=2.5, ipadx=2)
+
+        self.quit_btn = tk.Button(self.quit_frame, text="Quit", command=self.destroy, **button_big_arg)
+        self.quit_btn.pack(side="top", pady=2.5, ipadx=8)
+
+    def show_login(self):
+        win = tk.Toplevel(self)
+        win.title("User Login")
+        win.resizable(False, False)
+
+        label_arg = dict(width=LOGIN_LABEL_WIDTH, height=LOGIN_LABEL_HEIGHT)
+        entry_arg = dict(width=LOGIN_ENTRY_WIDTH)
+
+        host_frame = tk.Frame(win)
+        host_label = tk.Label(host_frame, text="Host: ", **label_arg)
+        host = tk.Entry(host_frame, **entry_arg)
+        host_label.pack(side="left")
+        host.pack(side="left")
+        host_frame.pack(side="top", pady=2, padx=5)
+
+        port_frame = tk.Frame(win)
+        port_label = tk.Label(port_frame, text="Port: ", **label_arg)
+        port = tk.Entry(port_frame, **entry_arg)
+        port_label.pack(side="left")
+        port.pack(side="left")
+        port_frame.pack(side="top", pady=2, padx=5)
+
+        user_frame = tk.Frame(win)
+        user_label = tk.Label(user_frame, text="User: ", **label_arg)
+        user = tk.Entry(user_frame, **entry_arg)
+        user_label.pack(side="left")
+        user.pack(side="left")
+        user_frame.pack(side="top", pady=2, padx=5)
+
+        passwd_frame = tk.Frame(win)
+        passwd_label = tk.Label(passwd_frame, text="Passwd: ", **label_arg)
+        passwd = tk.Entry(passwd_frame, **entry_arg)
+        passwd_label.pack(side="left")
+        passwd.pack(side="left")
+        passwd_frame.pack(side="top", pady=2, padx=5)
+
+        ssl_frame = tk.Frame(win)
+        ssl_label = tk.Label(ssl_frame, text="SSL: ", **label_arg)
+        ssl = tk.IntVar(ssl_frame)
+        ssl.set(1)
+        ssl_btn = (tk.Radiobutton(ssl_frame, variable=ssl, value=1, text="No", width=LOGIN_RBTN_WIDTH),
+                   tk.Radiobutton(ssl_frame, variable=ssl, value=2, text="SSL", width=LOGIN_RBTN_WIDTH),
+                   tk.Radiobutton(ssl_frame, variable=ssl, value=3, text="StartSSL", width=LOGIN_RBTN_WIDTH))
+        ssl_label.pack(side="left")
+        for i in ssl_btn:
+            i.pack(side="left")
+        ssl_frame.pack(side="top", pady=2, padx=5)
+
+        login_btn = tk.Button(win, text="Login", width=LOGIN_BTN_WIDTH,
+                              command=lambda: self.login_user(host.get(), port.get(), user.get(), passwd.get(),
+                                                              ssl.get(), mailbox_list))
+        login_btn.pack(side="top", pady=2)
+
+        mailbox_list = tk.Listbox(win, width=LOGIN_MAILBOX_WIDTH, height=LOGIN_MAILBOX_HEIGHT)
+        mailbox_list.pack(side="top", pady=(20, 2), padx=5)
+        self.refresh_mailbox(mailbox_list)
+
+        select_btn = tk.Button(win, text="Select", width=LOGIN_BTN_WIDTH,
+                               command=lambda: self.select_mailbox(mailbox=mailbox_list))
+        select_btn.pack(side="top", pady=2)
+
+    def refresh_mailbox(self, mailbox: tk.Listbox):
+        if self.imap:
+            mailbox.delete("0", tk.END)
+            for i in self.imap.list():
+                mailbox.insert(tk.END, f"{i}")
+
+    def login_user(self, host, port, user, passwd, ssl: int, listbox):
+        try:
+            port = int(port)
+        except (TypeError, ValueError):
+            msg.showerror("Login error", "Port must be int.")
+            return
+
+        if ssl == 1:
+            ssl = False
+            start_ssl = False
+        elif ssl == 2:
+            ssl = True
+            start_ssl = False
+        else:
+            ssl = False
+            start_ssl = True
+
+        try:
+            self.imap = Imap(user, passwd, host=host, port=port, ssl=ssl, start_ssl=start_ssl)
+            self.imap.connect()
+            self.imap.disconnect()
+        except imaplib.IMAP4.erro:
+            msg.showerror("Login error", "Sorry, IMAP Authentication error. Please check your user and password.")
+        except Exception:
+            msg.showerror("Login error", "Sorry, Unknown error.")
+        else:
+            self.refresh_mailbox(listbox)
+            msg.showinfo("Login", "Success!")
+
+    def select_mailbox(self, mailbox: tk.Listbox):
+        try:
+            index = mailbox.curselection()
+            if len(index) != 1:
+                msg.showwarning("Select", "Please select a mailbox.")
+                return
+
+            self.imap.inbox = mailbox.get(index[0])
+        except imaplib.IMAP4.error:
+            msg.showerror("Select error", "Bad mailbox.")
+        else:
+            msg.showinfo("Select", "Success!")
+
+    def show_info(self):
+        if self.imap:
+            msg.showinfo("Info", f"""Host: {self.imap.host}:{self.imap.port}
+User: {self.imap.user}
+Passwd: {self.imap.passwd}
+SSL: {self.imap.ssl}
+MailBox: {self.imap.inbox}""")
+        else:
+            msg.showwarning("Not login", "You should login first.")
+
+    def show_get(self):
+        win = tk.Toplevel(self)
+        win.title("User Login")
+        win.resizable(False, False)
+
+        label_arg = dict(width=GET_LABEL_WIDTH, height=GET_LABEL_HEIGHT)
+        entry_arg = dict(width=GET_ENTRY_WIDTH)
+
+        date_frame = tk.Frame(win)
+        date_label = tk.Label(date_frame, text="Date: ", **label_arg)
+        date = tk.Entry(date_frame, **entry_arg)
+        date_label.pack(side="left")
+        date.pack(side="left")
+        date_frame.pack(side="top", pady=2, padx=5)
+
+        get_btn = tk.Button(win, text="Get", width=LOGIN_BTN_WIDTH,
+                            command=lambda: self.get_mail(date.get()))
+        get_btn.pack(side="top", pady=2)
+
+    def get_mail(self, date):
+        if not self.imap:
+            msg.showwarning("Not login", "You should login first.")
+            return
+
+        if date == "ALL":
+            pass
+        elif date == "TODAY":
+            date = "ON " + strftime('%d-%b-%Y', localtime())
+        else:
+            try:
+                date = "ON " + strftime('%d-%b-%Y', strptime(date, "%Y-%m-%d"))
+            except ValueError:
+                msg.showerror("Get error", "Bad date.")
+                return
+
+        if not msg.askyesno("Sure", "Sure to load the mail from the server?\nYou should wait a moment."):
+            return
+
+        try:
+            self.imap.fetch_all(date)
+        except imaplib.IMAP4.error:
+            msg.showerror("Get error", "IMAP4 error, please check setting.")
+        else:
+            msg.showinfo("Get", "Success!")
+
+    def show_refresh(self):
+        win = tk.Toplevel(self)
+        win.title("User Login")
+        win.resizable(False, False)
+
+        label_arg = dict(width=REFRESH_LABEL_WIDTH, height=REFRESH_LABEL_HEIGHT)
+        entry_arg = dict(width=REFRESH_ENTRY_WIDTH)
+
+        start_frame = tk.Frame(win)
+        start_label = tk.Label(start_frame, text="Start: ", **label_arg)
+        start = tk.Entry(start_frame, **entry_arg)
+        start_label.pack(side="left")
+        start.pack(side="left")
+        start_frame.pack(side="top", pady=2, padx=5)
+
+        step_frame = tk.Frame(win)
+        step_label = tk.Label(step_frame, text="Step: ", **label_arg)
+        step = tk.Entry(step_frame, **entry_arg)
+        step_label.pack(side="left")
+        step.pack(side="left")
+        step_frame.pack(side="top", pady=2, padx=5)
+
+        get_btn = tk.Button(win, text="Refresh", width=LOGIN_BTN_WIDTH,
+                            command=lambda: self.refresh_mail(start.get(), step.get()))
+        get_btn.pack(side="top", pady=2)
+
+    def refresh_mail(self, start, step):
+        if not self.imap:
+            msg.showwarning("Not login", "You should login first.")
+            return
+
+        try:
+            start = int(start)
+            step = int(step)
+        except (TypeError, ValueError):
+            msg.showerror("refresh error", "Port must be int.")
+            return
+
+        try:
+            self.mailbox.delete("0", tk.END)
+            self.mailbox_num = []
+
+            mailbox = self.imap.mailbox[start:]
+            count = 0
+            for i in mailbox:
+                self.mailbox.insert(tk.END, f"{i}")
+                self.mailbox_num.append(i.num)
+                count += 1
+                if count == step:
+                    break
+        except IndexError:
+            msg.showerror("Refresh error", "Bad Index.")
+
+    def show_email(self):
+        index: tuple = self.mailbox.curselection()
+        if len(index) != 1:
+            msg.showwarning("Check", "Please select a email.")
+            return
+        else:
+            index: int = index[0]
+
+        if not self.imap:
+            msg.showwarning("Not login", "You should login first.")
+            return
+
+        email: Mail = self.imap.get_mail(self.mailbox_num[index])
+        win = tk.Toplevel(self)
+        win.title("Email info")
+        win.resizable(False, False)
+
+        arg = dict(width=EMAIL_INFO_WIDTH)
+        from_addr = tk.Label(win, text=f"From: {email.from_addr}", **arg)
+        title = tk.Label(win, text=f"Title: {email.title}", **arg)
+        info = st.ScrolledText(win, height=EMAIL_INFO_HEIGHT, state=tk.DISABLED, **arg)
+        btn_frame = tk.Frame(win)
+
+        for i in (from_addr, title, info, btn_frame):
+            i.pack(side="top", pady=2.5, ipadx=8, padx=2)
+
+        button_arg = dict(width=EMAIL_INFO_BUTTON_WIDTH, height=EMAIL_INFO_BUTTON_HEIGHT, border=2)
+        source = tk.Button(btn_frame, text="Source", command=lambda: self.save_email_source(email), **button_arg)
+        save = tk.Button(btn_frame, text="Save", command=lambda: self.save_email(email), **button_arg)
+        file = tk.Button(btn_frame, text="File", command=lambda: self.save_file(email), **button_arg)
+
+        for i in (source, save, file):
+            i.pack(side="left", ipadx=0.5)
+
+        self.show_email_text(info, email)
+
+    @staticmethod
+    def show_email_text(text: st.ScrolledText, email: Mail):
+        text.config(state=tk.NORMAL)
+        text.delete("0.0", tk.END)
+        text.insert(tk.END, email.body)
+        text.config(state=tk.DISABLED)
+
+    @staticmethod
+    def save_email_source(email: Mail):
+        file = None
+        try:
+            file = fd.asksaveasfile(mode="wb", title="Saving email source", filetypes=[("Email", ".email")],
+                                    defaultextension=".email")
+            if file is not None:
+                file.write(email.byte)
+        except IOError:
+            msg.showerror("Save source", "Sorry, IO Error.")
+        except Exception:
+            msg.showerror("Save source", "Sorry, Unknown error.")
+        finally:
+            if file:
+                file.close()
+                msg.showinfo("Saving source", "Success!")
+
+    @staticmethod
+    def save_email(email: Mail):
+        file = None
+        try:
+            file = fd.asksaveasfile(mode="w", title="Saving email text", filetypes=[("Text", ".txt")],
+                                    defaultextension=".txt")
+            if file is not None:
+                file.write(email.body)
+        except IOError:
+            msg.showerror("Save email", "Sorry, IO Error.")
+        except Exception:
+            msg.showerror("Save email", "Sorry, Unknown error.")
+        finally:
+            if file:
+                file.close()
+                msg.showinfo("Saving email", "Success!")
+
+    @staticmethod
+    def save_file(email: Mail):
+        path = fd.askdirectory(mustexist=True, title="Saving file")
+        if path is None:
+            return
+
+        try:
+            email.save_file(path)
+        except IOError:
+            msg.showerror("Save file", "Sorry, IO Error.")
+        except Exception:
+            msg.showerror("Save file", "Sorry, Unknown error.")
+        else:
+            msg.showinfo("Saving file", "Success!")
+
+    def save_config(self):
+        file = None
+        conf = {}
+
+        if self.imap:
+            conf["imap"] = {}
+            conf["imap"]["user"] = self.imap.user
+            conf["imap"]["passwd"] = self.imap.passwd
+            conf["imap"]["host"] = self.imap.host
+            conf["imap"]["port"] = self.imap.port
+            conf["imap"]["ssl"] = self.imap.ssl
+            conf["imap"]["start_ssl"] = self.imap.start_ssl
+            conf["imap"]["inbox"] = self.imap.inbox
+
+            mailbox = {}
+            for i in self.imap.mailbox:
+                byte: bytes = i.byte
+                mailbox[i.num] = byte.decode("utf-8")
+
+            conf["imap"]["mailbox"] = mailbox
+        try:
+            file = fd.asksaveasfile(mode="w", title="Saving config file", filetypes=[("JSON", ".json")],
+                                    defaultextension=".json")
+            if file is not None:
+                file.write(json.dumps(conf))
+        except IOError:
+            msg.showerror("Load config", "Sorry, IO Error.")
+        except Exception:
+            msg.showerror("Load config", "Sorry, Unknown error.")
+        finally:
+            if file:
+                file.close()
+                msg.showinfo("Saving success", "Success!")
+
+    def load_config(self):
+        file = None
+
+        try:
+            file = fd.askopenfile(mode="r", title="Loading config file", filetypes=[("JSON", ".json")])
+            if file is None:
+                return
+
+            conf = json.load(file)
+            imap = conf.get("imap")
+            if imap:
+                self.imap = Imap(
+                    user=imap["user"],
+                    passwd=imap["passwd"],
+                    host=imap.get("host", "localhost"),
+                    port=imap.get("port", 465),
+                    ssl=imap.get("ssl", True),
+                    start_ssl=imap.get("start_ssl", False)
+                )
+
+                self.imap.connect()
+                self.imap.disconnect()
+
+                self.imap.inbox = imap.get("inbox", "INBOX")
+                mailbox = imap.get("mailbox", {})
+                for num in imap.get("mailbox", {}):
+                    byte: str = mailbox[num]
+                    self.imap.add_mail(num, byte.encode("utf-8"))
+        except imaplib.IMAP4.error:
+            msg.showerror("Load config", "Sorry, IMAP Authentication error. Please check your user and password.")
+        except KeyError:
+            msg.showerror("Load config", "Sorry, Key Error.")
+        except IOError:
+            msg.showerror("Load config", "Sorry, IO Error.")
+        except Exception:
+            msg.showerror("Load config", "Sorry, Unknown error.")
+        finally:
+            if file:
+                file.close()
+                msg.showinfo("Loading success", "Success!")
+
+
+if __name__ == '__main__':
+    window = MailboxGUI()
+    window.mainloop()

mailbox/email.py

@@ -61,13 +61,22 @@ class Mail:
 
     @property
     def body(self):
-        return self.__get_body(self.__data).decode("utf-8")
+        return self.__get_body(self.__data)
 
     def __get_body(self, msg):
         if msg.is_multipart():
-            return self.__get_body(msg.get_payload(0))
+            res = ""
+            for i in msg.get_payload():
+                res += self.__get_body(i)
+            return res
         else:
-            return msg.get_payload(None, decode=True)
+            msg_type = msg.get_content_type()
+            if msg_type == "text/plain":
+                return "text/plain:\n" + msg.get_payload(decode=True).decode('utf-8') + "\n"
+            elif msg_type == "text/html":
+                return "text/html:\n" + msg.get_payload(decode=True).decode('utf-8') + "\n"
+            else:
+                return ""
 
     def save_file(self, file_dir: str):
         return self.__get_files(self.__data, file_dir)

mailbox/imap.py

@@ -103,3 +103,6 @@ class Imap:
 
     def add_mail(self, num: str, data: bytes):
         self.__mailbox[num] = Mail(num, data)
+
+    def get_mail(self, num: str):
+        return self.__mailbox.get(num, None)

sender-tk.py

@@ -170,10 +170,10 @@ class SenderGUI(tk.Tk):
             msg.showerror("Login error", "Port must be int.")
             return
 
-        if ssl == 0:
+        if ssl == 1:
             ssl = False
             start_ssl = False
-        elif ssl == 1:
+        elif ssl == 2:
             ssl = True
             start_ssl = False
         else:
@@ -183,11 +183,11 @@ class SenderGUI(tk.Tk):
         try:
             self.sender = Sender(user, passwd, host=host, port=port, debug=False, ssl=ssl, start_ssl=start_ssl)
         except smtplib.SMTPAuthenticationError:
-            msg.showinfo("Login error", "Sorry, SMTP Authentication error. Please check your user and password.")
+            msg.showerror("Login error", "Sorry, SMTP Authentication error. Please check your user and password.")
         except smtplib.SMTPException:
-            msg.showinfo("Login error", "Sorry, SMTP error.")
+            msg.showerror("Login error", "Sorry, SMTP error.")
         except Exception:
-            msg.showinfo("Login error", "Sorry, Unknown error.")
+            msg.showerror("Login error", "Sorry, Unknown error.")
         else:
             if win:
                 win.destroy()
@@ -464,7 +464,7 @@ Passwd: {self.sender.passwd}""")
             conf["sender"]["ssl"] = self.sender.ssl
             conf["sender"]["start_ssl"] = self.sender.start_ssl
         try:
-            file = fd.asksaveasfile(mode="w", title="Loading config file", filetypes=[("JSON", ".json")],
+            file = fd.asksaveasfile(mode="w", title="Saving config file", filetypes=[("JSON", ".json")],
                                     defaultextension=".json")
             if file is not None:
                 file.write(json.dumps(conf))