Update v0.2

onefile/fastmaillib.py

@@ -0,0 +1,97 @@
+# fastmaillib.py
+# LICENSE: MIT
+# VERSION: v0.2
+"""
+更加方便地使用 Python 收邮件
+部分基于 Liao Xuefeng 的代码, 并进行了封装操作
+此代码为 Python 3.x 而设计, 但理论上兼容 Python 2.x(2.7)
+"""
+
+import email, smtplib
+from email.mime.text import MIMEText
+from email.parser import Parser
+import poplib
+from email.header import decode_header
+from email.utils import parseaddr
+
+class Get:
+    def connect(mail, user, password, pop3_host):
+        """Connect to mail server"""
+        try:
+            obj = poplib.POP3_SSL(pop3_host)
+            obj.user(user)
+            obj.pass_(password)
+            return obj
+        except Exception as e:
+            return False
+    def num(obj):
+        """Get the number of messages on hand"""
+        try:
+            resp, mails, octets = obj.list()
+            return len(mails)
+        except Exception as e:
+            return False
+    def get(obj, num):
+        """Get mail's content"""
+        try:
+            resp, mails, octets = obj.list()
+            index = len(mails)
+            resp, lines, octets = obj.retr(index - 1) #
+            message_content = b'\r\n'.join(lines).decode('utf-8')
+            message = Parser().parsestr(message_content)
+            info = Get.print_info(message)
+            return info
+        except Exception as e:
+            return False
+    def quit(obj):
+        """Break connection"""
+        try:
+            obj.quit()
+            return True
+        except Exception as e:
+            return False
+    def print_info(msg, indent=0):
+        """Convert mail object to content"""
+        total = {}
+        if indent == 0:
+            for header in ['From', 'To', 'Subject']:
+                value = msg.get(header, '')
+                if value:
+                    if header == 'Subject':
+                        value = Get.decode_str(value)
+                        total["subject"] = value
+                    else:
+                        hdr, addr = parseaddr(value)
+                        name = Get.decode_str(hdr)
+                        value = u'%s <%s>' % (name, addr)
+                        total["name"] = name
+                        total["addr"] = addr
+                        # print(value)
+        if msg.is_multipart():
+            parts = msg.get_payload()
+            for n, part in enumerate(parts):
+                total["message"] = part
+                # print_info(part, indent + 1)
+        else:
+            content_type = msg.get_content_type()
+            if content_type == 'text/plain' or content_type == 'text/html':
+                content = msg.get_payload(decode=True)
+                charset = guess_charset(msg)
+                if charset:
+                    content = content.decode(charset)
+        return total
+    def decode_str(s):
+        """Decode mail objects"""
+        value, charset = decode_header(s)[0]
+        if charset:
+            value = value.decode(charset)
+        return value
+    def guess_charset(msg):
+        """Guess charset"""
+        charset = msg.get_charset()
+        if charset is None:
+            content_type = msg.get('Content-Type', '').lower()
+            pos = content_type.find('charset=')
+            if pos >= 0:
+                charset = content_type[pos + 8:].strip()
+        return charset