使用python自動發送郵件
阿新 • • 發佈:2019-04-15
def lena fail from sam arr .text encode cond 最近研究郵件透明加密技術,,需要使用腳本自動發送郵件測試,於是使用python寫了一個小程序。程序可以自動選擇不定數量的附件,隨機選擇主題,隨機選擇正文,然後自由組合發送,非常適合郵件方面的測試任務。順便說一下郵件透明加密技術和市場上的其他郵件加密技術相比再部署上非常簡單。”透明“二字就是部署不改變用戶原有習慣,目前的產品有天禦雲安的隱密郵,網址是https://mail.tyyunan.com/, 各位感興趣的可以看一下。
# -*- coding: utf-8 -*- __version__ = ‘‘‘Python-based smtp Client Copyright (C) 2018 Chunyu Cao This is free software; see LICENSE file for copying conditions. There is NO warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. ‘‘‘ # py .\smtp_client.py 10.115.5.68 25 [email protected] 123456 recv_name BASE64 --send_num 1 import time import random import json import smtplib import os import argparse import base64 from urllib import request from urllib import error from email.mime.text import MIMEText from email.mime.application import MIMEApplication from email.mime.multipart import MIMEMultipart from email.mime.image import MIMEImage from email import encoders from email.message import Message from email.charset import Charset from email.charset import QP from email.charset import BASE64 from email.charset import SHORTEST subject = [‘s1‘,‘s2‘] mail_text = [‘c1‘,‘c2‘,‘c3‘,‘c4‘,‘c5‘,‘c6‘,‘c7‘,‘c8‘,‘c9‘,‘c10‘,‘c11‘] #mail_text = [‘c5‘] files_list = [‘1.jpg‘,‘2.jpg‘,‘3.jpg‘,‘1M-good.txt‘,‘4.pdf‘,‘5.pdf‘,‘6.rar‘,‘7.docx‘,‘8.docx‘,‘9.7z‘,‘10.xlsx‘] #files_list = [‘11.txt‘,‘12.txt‘,‘13.txt‘,‘14.txt‘,‘15.txt‘,‘20180911.txt‘] #files_list = [‘1.jpg‘,‘1M-good.txt‘,‘5.pdf‘,‘6.rar‘,‘7.docx‘,‘8.docx‘,‘9.7z‘,‘10.xlsx‘] #files_list = [] def save_file(file_name,data): try: with open(file_name, ‘w‘) as fp: fp.write(data) except Exception as e: print(‘save file error:[{0}]‘.format(e)) raise class SmtpClient: def __init__(self,args,file_path,subject_file_name,text_file_name): self.args = args smtplib.Debug = args.debug self.recv_name =[] data = [] self.text_file_name = text_file_name with open(file_path+os.path.sep+args.recv_name_file, ‘rb‘) as fp: data = fp.read().decode(‘utf-8‘).split(‘,‘) for i in data:self.recv_name.append(i.strip()) #self.recv_name = random.sample(self.recv_name,random.randint(0, len(self.recv_name))) #print(‘[{len}]:{name}\n‘.format(len = len(self.recv_name),name = self.recv_name)) with open(file_path+os.path.sep+‘mail_subject‘+os.path.sep+subject_file_name, ‘rb‘) as fp: self.mail_subject = fp.read() with open(file_path+os.path.sep+‘mail_text‘+os.path.sep+text_file_name, ‘rb‘) as fp: self.mail_text = fp.read() def construct_text_mail_text(self,encode_type): charset = Charset(‘utf-8‘) charset.header_encoding = SHORTEST charset.body_encoding = encode_type self.msg.add_header(‘Content-Type‘, ‘text/plain‘,name = self.text_file_name) self.msg.set_charset(charset) self.msg.set_payload(self.msg._charset.body_encode(self.mail_text)) def construct_attachment_mail_text(self,encode_type): text_msg = Message() charset = Charset(‘utf-8‘) charset.header_encoding = SHORTEST charset.body_encoding = encode_type text_msg.add_header(‘Content-Type‘, ‘text/plain‘,name = self.text_file_name) text_msg.set_charset(charset) text_msg.set_payload(text_msg._charset.body_encode(self.mail_text)) self.msg.attach(text_msg) def construct_attachment_msg(self,file_path,file_name): with open(file_path+os.path.sep+‘mail_attach‘+os.path.sep+file_name, ‘rb‘) as fp: data = fp.read() attachment_msg = MIMEApplication(data) attachment_msg.add_header(‘Content-Disposition‘,‘attachment‘,filename = file_name) self.msg.attach(attachment_msg) def construct_text_mail_header(self): self.msg = Message() self.msg[‘From‘] = self.args.user_name self.msg[‘To‘] = ";".join(self.recv_name) #self.msg[‘To‘] = r"‘xxx‘ " + ";".join(self.recv_name) #self.msg[‘X-MS-TNEF-Correlator‘] = "00000000F3B7D51C2A9243419D3CDE795E89C788C4FD2100" self.msg[‘Subject‘] = self.mail_subject.decode(‘utf-8‘) def construct_attach_mail_header(self): self.msg = MIMEMultipart(‘related‘) self.msg[‘From‘] = self.args.user_name self.msg[‘To‘] = ";".join(self.recv_name) self.msg[‘Subject‘] = self.mail_subject.decode(‘utf-8‘) def send_mail(self): try: self.smtp.sendmail(self.args.user_name,self.recv_name,self.msg.as_string()) #self.smtp.sendmail(self.args.user_name,"[email protected]",self.msg.as_string()) #print(self.msg.as_string()) save_file(os.getcwd()+os.path.sep+‘tmp_file‘+os.path.sep+‘tmp.eml‘,self.msg.as_string()) print(‘mail send Success!‘) except Exception as e: print(‘mail send fail! Error!‘) raise def login(self): try: if self.args.ssl: self.smtp = smtplib.SMTP_SSL() else: self.smtp = smtplib.SMTP() self.smtp.connect(self.args.host,self.args.port) self.smtp.login(self.args.user_name,self.args.user_pass) except Exception as e: print(‘login fail! Error:‘+str(e)) raise def logout(self): try: self.smtp.close() except Exception as e: print(‘logout fail! Error:‘+str(e)) raise def main(args): print(args) file_dir = os.getcwd() send_num = args.send_num if args.encode_type == ‘QP‘: encode_type = QP elif args.encode_type == ‘BASE64‘: encode_type = BASE64 else: encode_type = -1 while send_num > 0: random_files_list = random.sample(files_list,random.randint(0, len(files_list))) print(‘attachment num is [{0}]‘.format(len(random_files_list))) random_subject = subject[random.randint(0, len(subject)-1)] random_text = mail_text[random.randint(0, len(mail_text)-1)] mail_client = SmtpClient(args,file_dir,random_subject,random_text) if len(random_files_list) == 0: mail_client.construct_text_mail_header() mail_client.construct_text_mail_text(encode_type) else: mail_client.construct_attach_mail_header() mail_client.construct_attachment_mail_text(encode_type) for mail_attachment in random_files_list: mail_client.construct_attachment_msg(file_dir,mail_attachment) mail_client.login() mail_client.send_mail() mail_client.logout() send_num -= 1 print(‘remain mail num is [{0}]‘.format(send_num)) time.sleep(1) if __name__ == ‘__main__‘: args = argparse.ArgumentParser( __version__) args.add_argument(‘host‘, default=‘localhost‘, help=‘mail server hostanme‘) args.add_argument(‘port‘, default=25,type=int, help=‘server port‘) args.add_argument(‘user_name‘, default=‘[email protected]‘, help=‘user name‘) args.add_argument(‘user_pass‘, default=‘123456‘, help=‘user pass‘) args.add_argument(‘recv_name_file‘, default=‘recv_name‘, help=‘recv name‘) args.add_argument(‘encode_type‘, default=‘BASE64‘, help=‘input msg text encode type:BASE64|QP|7or8bit‘) args.add_argument(‘--send_num‘,default=1,type=int, help=‘mail send num‘) args.add_argument(‘--ssl‘, action=‘store_true‘, help=‘Use SSL‘) args.add_argument(‘--debug‘, type=int, help=‘Debug level 1-5‘) args.add_argument(‘--verbose‘, action=‘store_true‘, default=False, help=‘Disable verbose mode‘) main(args.parse_args())
使用python自動發送郵件