import asyncio
import poplib
import datetime
from email.parser import Parser
from email.header import decode_header
from email.utils import parseaddr
# QQ邮箱轮询等待tradingview报警

# indent用于缩进显示:


def parse_email(msg, indent):
    if indent == 0:
        # 邮件的From, To, Subject存在于根对象上:
        for header in ['From', 'To', 'Subject', 'Date']:
            value = msg.get(header, '')
            if value:
                if header == 'Subject':
                    # 需要解码Subject字符串:
                    value = decode_str(value)
                if header == 'Date':
                    value = value
                else:
                    # 需要解码Email地址:
                    hdr, addr = parseaddr(value)
                    name = decode_str(hdr)
                    value = u'%s <%s>' % (name, addr)
            print('%s%s: %s' % ('  ' * indent, header, value))
    if (msg.is_multipart()):
        # 如果邮件对象是一个MIMEMultipart,
        # get_payload()返回list，包含所有的子对象:
        parts = msg.get_payload()
        for n, part in enumerate(parts):
            # 递归打印每一个子对象:
            return parse_email(part, indent + 1)
    else:
        # 邮件对象不是一个MIMEMultipart,
        # 就根据content_type判断:
        content_type = msg.get_content_type()
        if content_type == 'text/plain' or content_type == 'text/html':
            # 纯文本或HTML内容:
            content = msg.get_payload(decode=True)
            # 要检测文本编码:
            charset = guess_charset(msg)
            if charset:
                global Text
                Text = content.decode(charset)
            print('%sText: %s' % ('  ' * indent, Text))
        else:
            # 不是文本，作为附件处理:
            print('%sAttachment: %s' % ('  ' * indent, content_type))

# 解码


def decode_str(s):
    value, charset = decode_header(s)[0]
    if charset:
        value = value.decode(charset)
    return value

# 猜测字符编码


def guess_charset(msg):
    # 先从msg对象获取编码:
    charset = msg.get_charset()
    if charset is None:
        # 如果获取不到，再从Content-Type字段获取:
        content_type = msg.get('Content-Type', '').lower()
        for item in content_type.split(';'):
            item = item.strip()
            if item.startswith('charset'):
                charset = item.split('=')[1]
                break
    return charset

# 此函数通过使用poplib实现接收邮件


async def qq_pop_serve():
    # 要进行邮件接收的邮箱。改成自己的邮箱
    email_address = "1348006516@qq.com"
    # 要进行邮件接收的邮箱的密码。改成自己的邮箱的密码
    # 设置 -> 账户 -> POP3/IMAP/SMTP/Exchange/CardDAV/CalDAV服务 -> 开启服务：POP3/SMTP服务
    # 设置 -> 账户 -> POP3/IMAP/SMTP/Exchange/CardDAV/CalDAV服务 -> 生成授权码
    email_password = "tfspjbixhevlihfi"
    # 邮箱对应的pop服务器，也可以直接是IP地址
    # 改成自己邮箱的pop服务器；qq邮箱不需要修改此值
    pop_server_host = "pop.qq.com"
    # 邮箱对应的pop服务器的监听端口。改成自己邮箱的pop服务器的端口；qq邮箱不需要修改此值
    pop_server_port = 995
    try:
        # 连接pop服务器。如果没有使用SSL，将POP3_SSL()改成POP3()即可其他都不需要做改动
        email_server = poplib.POP3_SSL(
            host=pop_server_host, port=pop_server_port, timeout=10)
        print("pop3----connect server success, now will check username")
    except:
        print("pop3----sorry the given email server address connect time out")
        exit(1)
    try:
        # 验证邮箱是否存在
        email_server.user(email_address)
        print("pop3----username exist, now will check password")
    except:
        print("pop3----sorry the given email address seem do not exist")
        exit(1)
    try:
        # 验证邮箱密码是否正确
        email_server.pass_(email_password)
        print("pop3----password correct,now will list email")
    except:
        print("pop3----sorry the given username seem do not correct")
        exit(1)
    while True:
        # 邮箱中其收到的邮件的数量
        email_count = len(email_server.list()[1])
        # list()返回所有邮件的编号:
        resp, mails, octets = email_server.list()
        # 遍历所有的邮件
        for i in range(1, len(mails) + 1):
            # 通过retr(index)读取第index封邮件的内容；这里读取最后一封，也即最新收到的那一封邮件
            resp, lines, octets = email_server.retr(i)
            # lines是邮件内容，列表形式使用join拼成一个byte变量
            email_content = b'\r\n'.join(lines)
            try:
                # 再将邮件内容由byte转成str类型
                email_content = email_content.decode('utf-8')
            except Exception as e:
                print(str(e))
                continue
            # # 将str类型转换成<class 'email.message.Message'>
            # msg = email.message_from_string(email_content)
            msg = Parser().parsestr(email_content)
            print('------------------------------  华丽分隔符  ------------------------------')
            # 写入邮件内容到文件
            parse_email(msg, 0)
            # 服务一直开启不用关闭
            # email_server.close()
            # 将收件时间转为标准时间

            global gtime
            gtime = msg.get('Date', '')
            gtime = gtime.replace(":", ' ')
            gtime = gtime.split()
            day = int(gtime[1])
            mon = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun',
                   'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
            month = mon.index(gtime[2]) + 1
            year = int(gtime[3])
            hour = int(gtime[4])
            minute = int(gtime[5])
            second = int(gtime[6])
            # 时区为0无需加减
            # tUTC = +0000
            totime = datetime.datetime(year, month, day, hour, minute, second)
            # 只看最新lasttime秒内收到的邮件，并且在警报触发时执行在websocket上的交易命令
            lasttime = 500000
            if msg.get("From", '') == "TradingView <noreply@tradingview.com>" and (totime-datetime.datetime.now()).seconds <= lasttime:
                # 判断警报触发时需要执行的命令:针对Text文本执行
                print('totime=', totime, 'now=', datetime.datetime.now())
                print('TEXT=', Text)
            # 删除邮件方式1：警报读取并触发交易之后删除邮件
            # email_server.dele(i)
            # print('警报读取完毕，删除邮件')
            # 删除邮件方式2：判断1天前的邮件
            # if maildate.date() < datetime.datetime.now().date() - datetime.timedelta(days=1):
            #    print("正在删除邮件 第{}封，邮件日期：{} {}".format(i+1,maildate.date(),maildate.time()))
            #    mailServer.dele(i + 1)
        await asyncio.sleep(1)


loop = asyncio.get_event_loop()
loop.run_until_complete(
    qq_pop_serve(),
)

# loop.close()、