python与mysql简单连接

import pymysql

def dljm_hyjm():

"""用户欢迎界面"""
print("*" * 50)
print("欢迎光临业支中心服务大厅")
print("-" * 50)
print("请选择")
print("-" * 10)
print("【1】新客户注册")
print("【2】老用户登录")
print("【3】原密码修改")
print("-" * 10)
print("【0】关闭页面")
print("-" * 50)

def dljm_xzc(yhxmsr,cursor):

"""
:param yhxmsr: 输入用户名
:return: 输入密码，只是返回注册成功
"""
sql = "SELECT * FROM userinfo where username = '%s'" % yhxmsr
count = cursor.execute(sql)

if count == 1:
    print("您输入的用户名已存在，请确认，谢谢合作")
    return

# 没有找到该用户后提示客户输入密码
while True:
    yhmmsr = input("请输入您的密码：")
    yhmmqr = input("请确认您的密码：")
    if yhmmqr == yhmmsr:

        sql = "insert into userinfo (username,passwd,createdate,state,statedate) values('%s','%s',now(),'US10',now()) " % (yhxmsr, yhmmsr)
        count = cursor.execute(sql)
        conn.commit()
        print("恭喜您注册成功， %s ，欢迎您使用其他功能" % yhxmsr)
        return
    else:
        print("您输入的密码前后不一致，请您重新输入，谢谢")

def dljm_yhdl(yhxmsr):

"""用户名登录后校验用户名是否存在，密码正确"""
sql = "SELECT * FROM userinfo where username = '%s'" % yhxmsr
count = cursor.execute(sql)
if count == 1:
    print(" %s 欢迎光临，请稍等" % yhxmsr)
else:
    print("您输入的用户名不存在，请确认，谢谢合作")
    return

# 可以找到该用户后提示客户输入密码
while True:
    yhmmsr = input("请输入您的密码：")

    sql = "SELECT passwd FROM userinfo where username = '%s'" % yhxmsr
    cursor.execute(sql)
    sjk_mm = cursor.fetchone()[0]#获取密码
    print(sjk_mm)

    if yhmmsr == sjk_mm:
        print("欢迎登录系统， %s ，欢迎您使用其他功能" % yhxmsr)
    else:
        yhxz_yn = input("您输入的密码不符，请您确认是否重新输入，【1】重新输入【其他】退出")
        if yhxz_yn == "1":
            pass
        else:
            return

def dljm_mmxg(yhxmsr):

"""用户名登录后校验用户名是否存在，密码正确"""
sql = "SELECT * FROM userinfo where username = '%s'" % yhxmsr
count = cursor.execute(sql)
if count == 1:
    print(" %s 欢迎光临，请稍等" % yhxmsr)
else:
    print("您输入的用户名不存在，请确认，谢谢合作")
    return
# 没有找到该用户后提示客户输入密码

while True:
    yhmmsr = input("请输入您的密码：")

    sql = "SELECT passwd FROM userinfo where username = '%s'" % yhxmsr
    cursor.execute(sql)
    sjk_mm = cursor.fetchone()[0]#获取密码

    if yhmmsr == sjk_mm:
        yhmmx = input_info(yhmmsr,(" %s ，请输入您的新密码_回车不修改:" % yhxmsr))
        print(yhmmx)
        sql = "update userinfo set passwd='%s' where username ='%s' " % (yhmmx,yhxmsr)
        count = cursor.execute(sql)
        conn.commit()
        return
    else:
        yhxz_yn = input("您输入的密码不符，请您确认是否重新输入，【1】重新输入【其他】退出")
        if yhxz_yn == "1":
            pass
        else:
            return

def input_info(dict_value,tip_message):
“”"

:return: 用户输入密码替换老密码，不输入就用原密码
:param dict_value: 原数据
:param tip_message: 显示
"""
result_str = input(tip_message)
if len(result_str)>0:
    return result_str
else:
    return dict_value

while True:

dljm_hyjm() # h欢迎界面

kh_dlxz = input("请输入您的选择：")
if kh_dlxz in ("1","2","3"):

    # 连接database 数据库
    conn = pymysql.connect(host='127.0.0.1',port=3306,user='szc',password='szcNSP850219',database='szc_sql',charset='utf8')
    # 得到一个可以执行SQL语句的光标对象
    cursor = conn.cursor()

    print("您的选择是【%s】，请稍候 " % kh_dlxz)
    if kh_dlxz == "1":
        print("注册界面")
        yhxmsr = input("请输入您的用户名：")
        dljm_xzc(yhxmsr,cursor)
        # 关闭光标对象
        cursor.close()
        # 关闭数据库连接
        conn.close()

    elif kh_dlxz == "2":
        print("用户登录")
        yhxmsr = input("请输入您的用户名：")
        dljm_yhdl(yhxmsr)
        # 关闭光标对象
        cursor.close()
        # 关闭数据库连接
        conn.close()


    elif kh_dlxz == "3":
        print("修改密码")
        yhxmsr = input("请输入您的用户名：")
        dljm_mmxg(yhxmsr)
        # 关闭光标对象
        cursor.close()
        # 关闭数据库连接
        conn.close()


elif kh_dlxz == "0":
    print("感谢您的光临，再见！")
    break