Python-登录注册验证及加密

'''
需求：
1、注册的时候存账户密码到数据库里面，密码存密文，要加盐
2、登录的时候账号、密码从数据里面取，登录失败一次，错误次数+1，错误次数大于3不让登录
思路：
注册：
1、建表  id,username,password,error_count
2、账号不存在，并且2次输入的密码是一致，就插入到数据库里面
登录：
1、账号密码
2、select password,error_count from nhy_user where username = '%s' % username
3、取到error_count，判断是否大于3
4、把用户输入的密码MD5一下，和数据库里面存的密码对比
'''

import pymysql,hashlib

def op_mysql(sql:str):
    mysql_info = {
        'host': 'XXX.XXX.XXX.XXX',
        'port': XXX,
        'password': 'XXX',
        'user': 'XXX',
        'db': 'XXX',
        'charset': 'utf8',
        'autocommit': True
    }
    result = '执行完成'
    conn = pymysql.connect(**mysql_info)
    cur = conn.cursor(pymysql.cursors.DictCursor) #建立游标
    cur.execute(sql)
    if sql.strip().lower().startswith('select'):
        result  = cur.fetchone()
        # result  = cur.fetchall()
    cur.close()
    conn.close()
    return result

def md5(s,salt='$!@#$12232'):
    s = (str(s)+salt).encode()
    m = hashlib.md5(s)#加密
    return m.hexdigest()

def register(): # 注册
    for i in range(3):
        username  = input('username:').strip()
        passwd  = input('password:').strip()
        cpasswd  = input('cpassword:').strip()
        if username and passwd and cpasswd:
            if passwd == cpasswd:
                sql="select * from nhy_user where username='%s';" % username
                if op_mysql(sql):
                    print('用户已经存在！')
                else:
                    md5_password = md5(passwd)
                    sql2 = 'insert into nhy_user (username,password) value ("%s","%s");'%(username,md5_password)
                    op_mysql(sql2)
                    print('注册成功！')
            else:
                print('两次密码输入不一致')
        else:
            print('账号/密码不能为空')


def login(): # 登录
    username = input('username:').strip()
    passwd = input('passwd:').strip()
    if username and passwd:
        sql = "select passwd,error_count from app_myuser where username = '%s';"%username
        res = op_mysql(sql)
        # print('res======',type(res))
        if res:
            if res.get('error_count')>2:
                print('用户已经被锁定')
            else:
                new_passwd = md5(passwd)
                if res.get('password') == new_passwd:
                    print('登录成功！')
                else:
                    print('密码输入错误！')
                    count = res.get('error_count')+1
                    sql2 = 'update nhy_user set error_count = %s where username = "%s";'%(
                        count,username
                    )
                    op_mysql(sql2)
        else:
            print('用户不存在')


def clear_error_count():
    username = input('username:').strip()
    sql2 = 'update nhy_user set error_count = 0 where username = "%s";'%username
    op_mysql(sql2)