#!/usr/bin/python
# coding:utf-8

import smtplib
from email.mime.text import MIMEText
import os
import sys
import time

smtp_server = 'smtp.emoney.cn'
user = 'baoyifeng@emoney.cn'
password = '******'
mail_from = 'baoyifeng@emoney.cn'
display_from = 'keepalive_influx<{0}>'.format(mail_from)
mail_to = [
    "wangyuxin0623@emoney.cn",
    "lostsummer@gmail.com"
]
vip = '172.31.37.201'


def send_mail(mail_to, subject, content):
    msg = MIMEText(content, _subtype='plain', _charset='utf-8')
    msg['Subject'] = subject
    msg['From'] = display_from

    try:
        smtp = smtplib.SMTP()
        smtp.connect(smtp_server)
        smtp.login(user, password)
        smtp.sendmail(mail_from, mail_to, msg.as_string())
        smtp.close()
        print('send ok')
        return True

    except Exception as e:
        senderr = str(e)
        print(senderr)
        return False


if __name__ == "__main__":
    if (len(sys.argv) < 2):
        print("less arg!")
        sys.exit(1)
    stat = sys.argv[1]
    if stat not in ["master", "backup", "fault"]:
        print("error arg!")
        sys.exit(2)

    hostip = 'localhost'
    timestamp = time.strftime('%Y-%m-%d %X', time.localtime())
    subject = "{0} is change to be {1}, {2} floating".format(
        hostip, stat, vip)
    content = "{0} :\nvrrp transition, {1} changed to be {2}".format(
        timestamp, hostip, stat)

    sys.exit(0) if send_mail(mail_to, subject, content) else sys.exit(3)