python mysql登录注册

Python操作数据库（如MySQL)，Python发送能够与数据库直接通信的数据包，并获取数据库服务器的相应结果。是一种典型的基于TCP/IP的通信过程，要求必须要满足数据库服务器的数据包规则。在Python中，要操作MySQL，需要依赖于第三方库:pymysql，先安装:pip install pymysql

代码实现

function.py(通用函数模块)

def query_mysql(sql):
    conn=pymysql.connect(host='localhost',user='root',password='123456',database='test',charset='utf8')
    cursor=conn.cursor(pymysql.cursors.DictCursor)
    cursor.execute(sql)
    res=cursor.fetchall()
    conn.close()
    return res

def update_mysql(sql):
    conn=pymysql.connect(host='localhost',user='root',password='123456',database='test',charset='utf8',autocommit=True)
    cursor=conn.cursor(pymysql.cursors.DictCursor)
    cursor.execute(sql)
    conn.close()

主程序?

from function import *
def do_reg():
    username=input("请输入用户名：")
    password=input("请输入密码：")
    result=query_mysql(f"select name from info where name='{username}'")
    if len(result)==0:
        update_mysql(f"insert into info(name,password) values('{username}','{password}')")
    else :
        print("用户名已经注册")
    
if __name__ == '__main__':
        do_reg()