使用MySQL与Python开发简单登录功能，核心流程是先在MySQL中创建存储用户信息的表，再通过Python的pymysql库连接数据库，实现用户注册时的信息写入和登录时的信息校验。
开发前准备
首先需要确保本地已经安装MySQL数据库，并且安装了Python环境。接着需要安装Python操作MySQL的依赖库pymysql，可以通过pip命令完成安装：
pip install pymysql
数据库表设计
登录功能需要存储用户的账号和密码信息，我们可以在MySQL中创建一个名为user的用户表，表结构如下：
| 字段名 | 类型 | 说明 |
|---|---|---|
| id | int | 用户ID，自增主键 |
| username | varchar(50) | 用户账号，唯一 |
| password | varchar(50) | 用户密码 |
创建表的SQL语句如下：
CREATE TABLE user (
id INT PRIMARY KEY AUTO_INCREMENT,
username VARCHAR(50) UNIQUE NOT NULL,
password VARCHAR(50) NOT NULL
);
Python功能实现
数据库连接封装
首先封装一个获取数据库连接的函数，避免重复编写连接代码：
import pymysql
def get_db_conn():
# 建立数据库连接
conn = pymysql.connect(
host='127.0.0.1', # 数据库地址
port=3306, # 端口
user='root', # 数据库用户名
password='123456', # 数据库密码，替换为自己的实际密码
db='test_db', # 数据库名，替换为自己的实际库名
charset='utf8mb4'
)
return conn
用户注册功能
注册功能需要接收用户输入的账号和密码，先判断账号是否已存在，不存在则写入数据库：
def user_register(username, password):
conn = get_db_conn()
cursor = conn.cursor()
try:
# 先查询账号是否已存在
check_sql = "SELECT id FROM user WHERE username = %s"
cursor.execute(check_sql, (username,))
if cursor.fetchone():
return "该账号已存在，请更换账号"
# 插入新用户信息
insert_sql = "INSERT INTO user (username, password) VALUES (%s, %s)"
cursor.execute(insert_sql, (username, password))
conn.commit()
return "注册成功"
except Exception as e:
conn.rollback()
return f"注册失败：{str(e)}"
finally:
cursor.close()
conn.close()
用户登录功能
登录功能需要校验用户输入的账号和密码是否和数据库中存储的一致：
def user_login(username, password):
conn = get_db_conn()
cursor = conn.cursor()
try:
# 查询匹配的用户信息
query_sql = "SELECT id FROM user WHERE username = %s AND password = %s"
cursor.execute(query_sql, (username, password))
if cursor.fetchone():
return "登录成功"
else:
return "账号或密码错误"
except Exception as e:
return f"登录失败：{str(e)}"
finally:
cursor.close()
conn.close()
功能测试
我们可以编写简单的测试代码验证功能是否正常：
if __name__ == "__main__":
# 测试注册
print(user_register("test_user", "test123"))
# 测试登录
print(user_login("test_user", "test123"))
# 测试错误密码登录
print(user_login("test_user", "wrong_pwd"))
注意事项
- 实际开发中密码不能明文存储，建议使用hash算法对密码进行加密后再存入数据库，登录时对比加密后的字符串。
- 数据库连接的配置信息不要硬编码在代码中，建议放在配置文件里，方便后续修改和维护。
- SQL语句要使用参数化查询，避免SQL注入风险，不要直接拼接用户输入的内容到SQL语句中。
- 操作数据库后要及时关闭游标和连接，避免资源占用过高。