python-mysql數據庫登錄與注冊練習

import hashlib,pymysql,datetime
def my_db(sql):
    import pymysql
    coon = pymysql.connect(
        host = '118.24.3.40',user = 'jxz',passwd = '123456',
        port = 3306,db = 'jxz',charset = 'utf8'
    )
    cur = coon.cursor()#建立游標
    cur.execute(sql)#執行sql
    if sql.strip()[:6].upper() == 'SELECT':
        res = cur.fetchall()
    else:
        coon.commit()
        res = 'ok'
    cur.close()
    coon.close()
    return res

def my_md5(str):
    import hashlib
    new_str = str.encode()#吧字符串轉成bytes類型
    m = hashlib.md5()#實例化md5對象
    m.update(new_str)#加密
    return m.hexdigest()#獲取返回結果

def reg():
    username = input("username:").strip()
    pwd = input("pwd:").strip()
    cpwd = input('cpwd:').strip()
    if username and pwd and cpwd:
        sql = 'select * from nhy where name = "%s";'%username
        res = my_db(sql)
        if res:
            print("該用戶已經存在！")
        else:
            if pwd == cpwd:
                md5_pwd = my_md5(pwd)
                insert_sql = 'insert into nhy (name,pwd) values ("%s","%s");'%(username,md5_pwd)
                my_db(insert_sql)
                print("注冊成功！")
            else:
                print('兩次輸入的密碼不一致！')
    else:
        print('必填項不能為空！')

def login():
    username = input('username:').strip()
    pwd = input('pwd:').strip()
    if username and pwd:
        md5_pwd = my_md5(pwd)
        sql = 'select * from nhy where name = "%s" and pwd = "%s";'%(username,md5_pwd)
        res = my_db(sql)
        if res:
            print("歡迎，登陸成功！今天是%s"%datetime.date.today())
        else:
            print('賬號或密碼錯誤')
    else:
        print("必填項不能為空！")
#login()
reg()