|
|
import smtplib
|
|
|
import imaplib
|
|
|
import time
|
|
|
import sys
|
|
|
from email.mime.multipart import MIMEMultipart
|
|
|
from email.mime.text import MIMEText
|
|
|
import ntpath
|
|
|
from email.mime.base import MIMEBase
|
|
|
from email import encoders
|
|
|
|
|
|
msg = MIMEMultipart()
|
|
|
msg['From'] = "admin@mailu.io"
|
|
|
msg['To'] = "user@mailu.io"
|
|
|
msg['Subject'] = "File Test"
|
|
|
msg.attach(MIMEText(sys.argv[1], 'plain'))
|
|
|
|
|
|
if len(sys.argv) == 3:
|
|
|
part = MIMEBase('application', 'octet-stream')
|
|
|
part.set_payload((open(sys.argv[2], "rb")).read())
|
|
|
encoders.encode_base64(part)
|
|
|
part.add_header('Content-Disposition', "attachment; filename=%s" % ntpath.basename(sys.argv[2]))
|
|
|
msg.attach(part)
|
|
|
|
|
|
try:
|
|
|
smtp_server = smtplib.SMTP('localhost')
|
|
|
smtp_server.set_debuglevel(1)
|
|
|
smtp_server.connect('localhost', 587)
|
|
|
smtp_server.ehlo()
|
|
|
smtp_server.starttls()
|
|
|
smtp_server.ehlo()
|
|
|
smtp_server.login("admin@mailu.io", "password")
|
|
|
|
|
|
smtp_server.sendmail("admin@mailu.io", "user@mailu.io", msg.as_string())
|
|
|
smtp_server.quit()
|
|
|
except:
|
|
|
sys.exit(25)
|
|
|
|
|
|
time.sleep(30)
|
|
|
|
|
|
try:
|
|
|
imap_server = imaplib.IMAP4_SSL('localhost')
|
|
|
imap_server.login('user@mailu.io', 'password')
|
|
|
except:
|
|
|
sys.exit(110)
|
|
|
|
|
|
stat, count = imap_server.select('inbox')
|
|
|
try:
|
|
|
stat, data = imap_server.fetch(count[0], '(UID BODY[TEXT])')
|
|
|
except :
|
|
|
print("Couldn’t list email in imap inbox")
|
|
|
sys.exit(99)
|
|
|
|
|
|
if sys.argv[1] in str(data[0][1]):
|
|
|
print("Success sending and receiving email!")
|
|
|
else:
|
|
|
print("Failed receiving email with message %s, message not contained" % sys.argv[1])
|
|
|
sys.exit(99)
|
|
|
|
|
|
typ, data = imap_server.search(None, 'ALL')
|
|
|
for num in data[0].split():
|
|
|
imap_server.store(num, '+FLAGS', '\\Deleted')
|
|
|
imap_server.expunge()
|
|
|
|
|
|
imap_server.close()
|
|
|
imap_server.logout()
|