#! /usr/bin/env python
#coding=utf-8

import smtplib
from os.path import basename
from email.mime.application import MIMEApplication
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email.utils import COMMASPACE,formatdate
from datetime import datetime
import ResultFolder
from email.header import Header
import zipfile,os
import decode_pwd



def send_email(send_from,send_to,subject,text,files1,files2,server="smtp.163.com"):

    #assert(isinstance(send_to,list),"Send To email should be a list")

    msg = MIMEMultipart()
    msg[\'From\'] = send_from
    msg[\'To\'] = send_to
    msg[\'Date\'] = formatdate(localtime = True)
    msg[\'Subject\'] = Header(subject, \'utf-8\')
    my_password = "7A68656E676A696E6731313238"
    my_pass = decode_pwd.decode_pwd(my_password)

    msg.attach(MIMEText(text,\'html\'))

    with open(files1,"rb") as f:
        part1 = MIMEApplication(f.read())
        part1["Content-Type"] = \'application/octet-stream\'
        part1["Content-Disposition"] = \'attachment; filename="TestResult.html"\'
        msg.attach(part1)
		
    with open(files2,"rb") as g:
        part2 = MIMEApplication(g.read())
        part2["Content-Type"] = \'application/octet-stream\'
        part2["Content-Disposition"] = \'attachment; filename="Test.log"\'
        msg.attach(part2)
    smtp = smtplib.SMTP_SSL(server,465)
    smtp.login(send_from,my_pass)
    smtp.sendmail(send_from,send_to,msg.as_string())
    smtp.quit()

def zip_report(startdir,file_news):
    print startdir
    z = zipfile.ZipFile(file_news,\'w\',zipfile.ZIP_DEFLATED)
    for dirpath, dirnames, filenames in os.walk(startdir):
        fpath = dirpath.replace(startdir, \'\')  # 这一句很重要，不replace的话，就从根目录开始复制
        fpath = fpath and fpath + os.sep or \'\'  # 这句话理解我也点郁闷，实现当前文件夹以及包含的所有文件的压缩
        for filename in filenames:
            z.write(os.path.join(dirpath, filename), fpath + filename)
    print (\'压缩成功\')
    z.close()


def send_report():
    send_f = "xxxxx@163.com"
    send_t = "xxxxxx@163.com"
    subject = "Software test Report_"+ str(datetime.today())

    files1 = ResultFolder.GetRunDirectory()+"\\TestResult.html"

    print "zip log is start"
    zipdir = ResultFolder.GetRunDirectory()+"\\"
    file2_news = zipdir + \'TestLog.zip\'
    zip_report(zipdir,file2_news)
    files2 = ResultFolder.GetRunDirectory() + "\\TestLog.zip"


    with open(files1,\'r\') as f:
        text = f.read()

    send_email(send_f,send_t,subject,text,files1,files2)

if __name__==\'__main__\':
    send_report()