support pissmail

src/mail/mail_constants.py

@@ -19,6 +19,7 @@ DOMAIN_GMX_US = "gmx.us"
 DOMAIN_ONET = "onet.pl"
 DOMAIN_NAVER = "naver.com"
 DOMAIN_INBOX_LV = "inbox.lv"
+DOMAIN_PISS_MAIL = "pissmail.com"
 
 AOL_IMAP_SERVER = "imap.aol.com"
 IMAP_SERVER_163 = "imap.163.com"
@@ -34,15 +35,27 @@ BTVM_NE_JP = "imap.btvm.ne.jp"
 SEREVER_GMAIL = "imap.gmail.com"
 SERVER_IMAGE_ONET = "imap.poczta.onet.pl"
 SERVER_GMX = "imap.gmx.com"
+SERVER_PISS_MAIL = "mail.pissmail.com"
 INBOX_LV = "mail.inbox.lv"
 
 
 def show_folders(imap) -> list:
     folders = []
-    for i in imap.list()[1]:
-        l = i.decode().split(' "/" ')
-        folders.append(l[1])
-    return folders
+    isImapClient = isinstance(imap, IMAPClient)
+    if not isImapClient:
+        for i in imap.list()[1]:
+            l = i.decode().split(' "/" ')
+            if len(l)>1:
+                folders.append(l[1])
+        if len(folders)==0:
+            folders.append('INBOX')
+        return folders
+    else:
+        list = imap.list_folders()
+        for i in list:
+            name = i[-1]
+            folders.append(name)
+        return folders
 
 
 def create_imap(login: str):
@@ -78,6 +91,8 @@ def create_imap(login: str):
         imap = imaplib.IMAP4_SSL(NAVER_SERVER, port=993)
     elif DOMAIN_INBOX_LV in login:
         imap = imaplib.IMAP4_SSL(INBOX_LV, port=993)
+    elif DOMAIN_PISS_MAIL in login:
+        imap = imaplib.IMAP4_SSL(SERVER_PISS_MAIL, port=993)
     else:
         imap = imaplib.IMAP4_SSL(AOL_IMAP_SERVER)
     return imap