import hashlib
import time
from random import Random


def email_util(email, category='register'):
    """
        发送邮件的方法
        register:   注册账号
        forget:     找回密码
        change:     修改邮箱
    """
    if category == 'register':
        subject = '彬彬商场注册激活链接'
        code_str = generate_random_str(16)
        message = '请点击下面的链接激活您的账号: http://127.0.0.1:8000/active/{0}'.format(code_str)

    elif category == 'forget':
        subject = '彬彬商场忘记密码连接'
        code_str = generate_random_str(8)
        timestamp = int(time.time())
        md5 = hashlib.md5()
        md5.update((code_str + email + str(timestamp)).encode('utf8'))
        hash_str = md5.hexdigest()
        message = '请点击下面的链接修改你的密码: http://127.0.0.1:8000/reset?timestamp={0}&hash={1}&email={2}'.format(
            timestamp,
            hash_str, email)
    elif category == 'change':
        subject = '彬彬商场修改邮箱连接'
        code_str = generate_random_str(6)
        message = '你的邮箱验证码为: {0}'.format(code_str)
    else:
        return False


def generate_random_str(str_len=8):
    """ 生成长度为str_len的随机字符串 """
    custom_str = ''
    chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
    random = Random()
    for i in range(str_len):
        custom_str += chars[random.randint(0, len(chars) - 1)]
    return custom_str