【Python】基于数据库连接实现简单注册功能_python连接数据库实现注册功能

说明

针对数据库的操作，一共分三步：
1.建立数据库连接
2.执行SQL语句
3.关闭数据库连接

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

代码实现

common.py(通用函数模块)

import pymysql
from pymysql.cursors import DictCursor
#针对数据库连接进行封装操作
def query_mysql(sql):
    conn = pymysql.connect(host='localhost', user='root', password='root', database='security', charset='utf8')
    cursor = conn.cursor()
    cursor.execute(sql)
    result=cursor.fetchall()
    conn.close()
    return result

app.py（主程序）

from exercise.common import *
def do_reg():
    username=input("请输入用户名：")
    password=input("请输入密码：")
    result=query_mysql(f"select username from users where username='{username}'")
    if len(result)==0:
        query_mysql(f"insert into users (username,password) values ('{username}','{password}')")
    else:
        print("用户名已经存在，不允许注册.")
 
if __name__ == '__main__':
    do_reg()