forked from DxvLwRYF/apiAutoTest
73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
#!/usr/bin/env/python3
|
||
# -*- coding:utf-8 -*-
|
||
"""
|
||
@project: apiAutoTest
|
||
@author: zy7y
|
||
@file: send_email.py
|
||
@ide: PyCharm
|
||
@time: 2020/8/3
|
||
"""
|
||
import yagmail
|
||
from tools import logger
|
||
import zipfile
|
||
import os
|
||
|
||
|
||
class EmailServe:
|
||
|
||
@staticmethod
|
||
def zip_report(file_path: str, out_path: str):
|
||
"""
|
||
压缩指定文件夹
|
||
:param file_path: 目标文件夹路径
|
||
:param out_path: 压缩文件保存路径+xxxx.zip
|
||
:return: 无
|
||
"""
|
||
file_path = f"{file_path}/html"
|
||
zip = zipfile.ZipFile(out_path, "w", zipfile.ZIP_DEFLATED)
|
||
for path, dirnames, filenames in os.walk(file_path):
|
||
# 去掉目标跟路径，只对目标文件夹下边的文件及文件夹进行压缩
|
||
fpath = path.replace(file_path, '')
|
||
|
||
for filename in filenames:
|
||
zip.write(
|
||
os.path.join(
|
||
path, filename), os.path.join(
|
||
fpath, filename))
|
||
zip.close()
|
||
|
||
@staticmethod
|
||
def send_email(setting: dict, file_path):
|
||
"""
|
||
入参一个字典
|
||
:param user: 发件人邮箱
|
||
:param password: 邮箱授权码
|
||
:param host: 发件人使用的邮箱服务 例如：smtp.163.com
|
||
:param contents: 内容
|
||
:param addressees: 收件人列表
|
||
:param title: 邮件标题
|
||
:param enclosures: 附件列表
|
||
:param file_path: 需要压缩的文件夹
|
||
:return:
|
||
"""
|
||
EmailServe.zip_report(
|
||
file_path=file_path,
|
||
out_path=setting['enclosures'])
|
||
yag = yagmail.SMTP(
|
||
setting['user'],
|
||
setting['password'],
|
||
setting['host'])
|
||
# 发送邮件
|
||
yag.send(
|
||
setting['addressees'],
|
||
setting['title'],
|
||
setting['contents'],
|
||
setting['enclosures'])
|
||
# 关闭服务
|
||
yag.close()
|
||
logger.info("邮件发送成功！")
|
||
|
||
|
||
# if __name__ == '__main__':
|
||
# EmailServe.zip_report('../report/html', 'report.zip')
|