Cambios varios

entry/email/inbox.py

@@ -1,53 +1,5 @@
 import argparse
-import imaplib
-import email
-
-
-# Email class that fetches emails from server by uid and can parse body for secretary, can also delete email by uid
-class Email:
-    def __init__(self, uid):
-        self.uid = uid
-        self.message = ''
-
-    def get(self, imap):
-        status, raw_data = imap.uid('fetch', self.uid, '(RFC822)')
-        if status != 'OK':
-            raise Exception('Could not recover message {0}'.format(self.uid))
-
-        self.message = email.message_from_bytes(raw_data[0][1])
-
-    def delete(self, imap):
-        status, result = imap.uid('STORE', self.uid, '+FLAGS', '(\\Deleted)')
-        if status != 'OK':
-            raise Exception('Could not flag message {0}'.format(self.uid))
-
-
-def connect(imap_url, port, username, password, ssl=False):
-    if ssl:
-        server = imaplib.IMAP4_SSL(imap_url, port=port)
-    else:
-        server = imaplib.IMAP4(imap_url, port=port)
-    server.login(username, password)
-    return server
-
-
-def check_inbox(server):
-    status, msg = server.select('INBOX')
-    if status != 'OK':
-        return None
-
-    status, ids = server.uid('search', None, 'All')
-    if status != 'OK':
-        return None
-
-    ids = ids[0].decode().split()
-    emails = []
-    for mid in ids:
-        em = Email(mid)
-        em.get(server)
-        emails.append(em)
-
-    return emails
+from src.email.functions import connect, check_inbox
 
 
 def main(args):