David
6 years ago
5 changed files with 221 additions and 48 deletions
@ -0,0 +1,100 @@ |
|||
import imaplib |
|||
import pathlib |
|||
import time |
|||
import os |
|||
from secrets import token_urlsafe |
|||
|
|||
try: |
|||
import imap_settings as settings |
|||
except: |
|||
raise ValueError("Imap settings file not found, please check the readme - miniimap") |
|||
|
|||
|
|||
def connect(): |
|||
"connect to the IMAP server Inbox" |
|||
server = imaplib.IMAP4_SSL(settings.IMAP_HOST) |
|||
server.login(settings.IMAP_USER, settings.IMAP_PWD) |
|||
tmp, data = server.select('Inbox') |
|||
|
|||
#When you connect to the inbox one of the parameters returned is the current |
|||
#number of messages in it |
|||
if os.environ.get('NUMMESSAGES') is None: |
|||
os.environ["NUMMESSAGES"] = data[0].decode("UTF-8") |
|||
|
|||
return server |
|||
|
|||
def bytesmessage_to_string(msg): |
|||
"converts bytes-like message to string" |
|||
if type(msg) is bytes: |
|||
msg = msg.decode("UTF-8").rstrip() |
|||
return msg |
|||
else: |
|||
return str(msg) |
|||
|
|||
def send(inbox, msg): |
|||
"send msg to inbox in MIME format" |
|||
|
|||
server = connect() |
|||
tmp, data = server.append(inbox, flags='', date_time=time.time(), message=str(msg).encode("UTF-8")) |
|||
server.close() |
|||
|
|||
|
|||
def recv_all(): |
|||
"""receive a list of all MIME messages from inbox newer than the last message when first connected""" |
|||
|
|||
server = connect() |
|||
r = [] |
|||
|
|||
tmp, data = server.search(None, 'ALL') |
|||
|
|||
oldermsgid = os.environ.get('NUMMESSAGES') |
|||
|
|||
for num in data[0].split(): |
|||
if int(num) >= int(oldermsgid): |
|||
tmp, data = server.fetch(num, '(RFC822)') |
|||
msg = bytesmessage_to_string(data[0][1]) |
|||
r.append((num, msg)) |
|||
os.environ["NUMMESSAGES"] = num.decode("UTF-8") |
|||
|
|||
server.close() |
|||
|
|||
return r |
|||
|
|||
|
|||
def clean_inbox(): |
|||
"""clean all messsages from IMAP inbox""" |
|||
print('cleaning IMAP...') |
|||
server = connect() |
|||
tmp, data = server.search(None, 'ALL') |
|||
for num in data[0].split(): |
|||
server.store(num, '+FLAGS', '\\Deleted') |
|||
server.expunge() |
|||
server.close() |
|||
print('IMAP inbox empty.') |
|||
|
|||
|
|||
def backup_inbox(): |
|||
"""copy all messsages from IMAP to local backup folder""" |
|||
server = connect() |
|||
tmp, data = server.search(None, 'ALL') |
|||
for num in data[0].split(): |
|||
tmp, data = server.fetch(num, '(RFC822 BODY[HEADER])') |
|||
device = str(data[0][1]).split('From: "')[1].split(' of')[0] |
|||
name = device + "_" + token_urlsafe(16) + ".eml" |
|||
msg = bytesmessage_to_string(data[0][1]) |
|||
with open(os.path.join('Backup/TestInbox',name), "wb") as f: |
|||
f.write(str(msg).encode()) |
|||
|
|||
server.close() |
|||
|
|||
def restore_inbox(): |
|||
"""copy all the messages from the Backup folder to the IMAP inbox""" |
|||
server = connect() |
|||
backups = pathlib.Path("./Backup/TestInbox") |
|||
emails = backups.glob("*.eml") |
|||
l = [ path for path in emails ] |
|||
for p in l: |
|||
with open(p, "rb") as f: |
|||
tmp, data = server.append("Inbox", flags='', date_time=p.stat().st_ctime, message=f.read(-1)) |
|||
|
|||
server.close() |
|||
Loading…
Reference in new issue