python中登錄、注冊操作數據庫

# 1、注冊、登錄
# 1、注冊的時候存賬戶密碼到數據庫里面，密碼存密文，要加鹽
# 2、登錄的時候賬號、密碼從數據里面取，登錄失敗一次，錯誤次數 + 1，錯誤次數大於3不讓登錄
# 1、建表  id,username,password,error_count
# 2、賬號不存在，並且2次輸入的密碼是一致，就插入到數據庫里面
# select * from nhy_user where username='%s' % username


# 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': '***.24.3.40',
        'port': 3306,
        'password': '***',
        'user': 'jxz',
        'db': 'jxz',
        '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 password,error_count from nhy_user where username = '%s'" % username
        res = op_mysql(sql)
        print('res======', 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)


login()
# clear_error_count()