# encoding=utf-8
import pymysql as pms
import openpyxl
import datetime
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.header import Header
import smtplib

def get_datas(sql):
  # 一個傳入sql導(dǎo)出數(shù)據(jù)的函數(shù)
  # 跟數(shù)據(jù)庫建立連接
  conn = pms.connect(host='實例地址', user='用戶',
            passwd='密碼', database='庫名', port=3306, charset="utf8")
  # 使用 cursor() 方法創(chuàng)建一個游標(biāo)對象 cursor
  cur = conn.cursor()
  # 使用 execute() 方法執(zhí)行 SQL
  cur.execute(sql)
  # 獲取所需要的數(shù)據(jù)
  datas = cur.fetchall()
  #關(guān)閉連接
  cur.close()
  #返回所需的數(shù)據(jù)
  return datas

def get_fields(sql):
  # 一個傳入sql導(dǎo)出字段的函數(shù)
  conn = pms.connect(host='rm-rj91p2yhl9dm2xmbixo.mysql.rds.aliyuncs.com', user='bi-analyzer',
            passwd='pcNzcKPnn', database='kikuu', port=3306, charset="utf8")
  cur = conn.cursor()
  cur.execute(sql)
  # 獲取所需要的字段名稱
  fields = cur.description
  cur.close()
  return fields

def get_excel(data, field, file):
  # 將數(shù)據(jù)和字段名寫入excel的函數(shù)
  #新建一個工作薄對象
  new = openpyxl.Workbook()
  #激活一個新的sheet
  sheet = new.active
  #給sheet命名
  sheet.title = '數(shù)據(jù)展示'
  #將字段名稱循環(huán)寫入excel第一行，因為字段格式列表里包含列表，每個列表的第一元素才是字段名稱
  for col in range(len(field)):
    #row代表行數(shù)，column代表列數(shù)，value代表單元格輸入的值，行數(shù)和列數(shù)都是從1開始，這點于python不同要注意
    _ = sheet.cell(row=1, column=col+1, value=u'%s' % field[col][0])
   #將數(shù)據(jù)循環(huán)寫入excel的每個單元格中  
  for row in range(len(data)):
    for col in range(len(field)):
      #因為第一行寫了字段名稱，所以要從第二行開始寫入
      _ = sheet.cell(row=row+2, column=col + 1, value=u'%s' % data[row][col])
      #將生成的excel保存，這步是必不可少的
  newworkbook = new.save(file)
  #返回生成的excel
  return newworkbook

def getYesterday():
  # 獲取昨天日期的字符串格式的函數(shù)
  #獲取今天的日期
  today = datetime.date.today()
  #獲取一天的日期格式數(shù)據(jù)
  oneday = datetime.timedelta(days=1)
  #昨天等于今天減去一天
  yesterday = today - oneday
  #獲取昨天日期的格式化字符串
  yesterdaystr = yesterday.strftime('%Y-%m-%d')
  #返回昨天的字符串
  return yesterdaystr

def create_email(email_from, email_to, email_Subject, email_text, annex_path, annex_name):
  # 輸入發(fā)件人昵稱、收件人昵稱、主題，正文，附件地址,附件名稱生成一封郵件
  #生成一個空的帶附件的郵件實例
  message = MIMEMultipart()
  #將正文以text的形式插入郵件中
  message.attach(MIMEText(email_text, 'plain', 'utf-8'))
  #生成發(fā)件人名稱（這個跟發(fā)送的郵件沒有關(guān)系）
  message['From'] = Header(email_from, 'utf-8')
  #生成收件人名稱（這個跟接收的郵件也沒有關(guān)系）
  message['To'] = Header(email_to, 'utf-8')
  #生成郵件主題
  message['Subject'] = Header(email_Subject, 'utf-8')
  #讀取附件的內(nèi)容
  att1 = MIMEText(open(annex_path, 'rb').read(), 'base64', 'utf-8')
  att1["Content-Type"] = 'application/octet-stream'
  #生成附件的名稱
  att1["Content-Disposition"] = 'attachment; filename=' + annex_name
  #將附件內(nèi)容插入郵件中
  message.attach(att1)
  #返回郵件
  return message

def send_email(sender, password, receiver, msg):
  # 一個輸入郵箱、密碼、收件人、郵件內(nèi)容發(fā)送郵件的函數(shù)
  try:
    #找到你的發(fā)送郵箱的服務(wù)器地址，已加密的形式發(fā)送
    server = smtplib.SMTP_SSL("smtp.mxhichina.com", 465) # 發(fā)件人郵箱中的SMTP服務(wù)器
    server.ehlo()
    #登錄你的賬號
    server.login(sender, password) # 括號中對應(yīng)的是發(fā)件人郵箱賬號、郵箱密碼
    #發(fā)送郵件
    server.sendmail(sender, receiver, msg.as_string()) # 括號中對應(yīng)的是發(fā)件人郵箱賬號、收件人郵箱賬號（是一個列表）、郵件內(nèi)容
    print("郵件發(fā)送成功")
    server.quit() # 關(guān)閉連接
  except Exception:
    print(traceback.print_exc())
    print("郵件發(fā)送失敗")

def main():
  print(datetime.datetime.now())
  my_sql = sql = "SELECT a.id '用戶ID',\
      a.gmtCreate '用戶注冊時間',\
      af.lastLoginTime '最后登錄時間',\
      af.totalBuyCount '歷史付款子單數(shù)',\
      af.paidmountUSD '歷史付款金額',\
      af.lastPayTime '用戶最后支付時間'\
     FROM table a\
   LEFT JOIN tableb af ON a.id= af.accountId ;"
  # 生成數(shù)據(jù)
  my_data = get_datas(my_sql)
  # 生成字段名稱
  my_field = get_fields(my_sql)
  # 得到昨天的日期
  yesterdaystr = getYesterday()
  # 文件名稱
  my_file_name = 'user attribute' + yesterdaystr + '.xlsx'
  # 文件路徑
  file_path = 'D:/work/report/' + my_file_name
  # 生成excel
  get_excel(my_data, my_field, file_path)

  my_email_from = 'BI部門自動報表機器人'
  my_email_to = '運營部'
  # 郵件標(biāo)題
  my_email_Subject = 'user' + yesterdaystr
  # 郵件正文
  my_email_text = "Dear all,\n\t附件為每周數(shù)據(jù)，請查收！\n\nBI團隊 "
  #附件地址
  my_annex_path = file_path
  #附件名稱
  my_annex_name = my_file_name
  # 生成郵件
  my_msg = create_email(my_email_from, my_email_to, my_email_Subject,
             my_email_text, my_annex_path, my_annex_name)
  my_sender = '阿里云郵箱'
  my_password = '我的密碼'
  my_receiver = [10001@qq.com']#接收人郵箱列表
  # 發(fā)送郵件
  send_email(my_sender, my_password, my_receiver, my_msg)
  print(datetime.datetime.now())

if __name__ == "__main__":
  main();