在 Flask 中使用 SQLAlchemy（一个流行的 ORM 工具）来操作数据库时，创建和更新数据是非常常见的任务。以下是如何使用 SQLAlchemy 来创建新数据和更新已有数据的详细说明。 ### 1. 创建新的数据 首先需要定义一个模型类，并使用 SQLAlchemy 的 `session` 来添加新记录到数据库中。 ```python from flask import Flask from flask_sqlalchemy import SQLAlchemy app = Flask(__name__) # 配置数据库连接 app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///example.db' db = SQLAlchemy(app) # 定义模型 class User(db.Model): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(80), unique=True, nullable=False) email = db.Column(db.String(120), unique=True, nullable=False) def __repr__(self): return f'<User {self.username}>' # 创建数据库表 with app.app_context(): db.create_all() # 添加新数据 def create_user(username, email): with app.app_context(): new_user = User(username=username, email=email) db.session.add(new_user) db.session.commit() ``` **解释：** - `User` 是我们定义的一个模型类，对应数据库中的 `user` 表。 - `create_user` 函数用于向数据库中插入一条新的用户记录。 - `db.session.add(new_user)` 将新对象添加到会话中。 - `db.session.commit()` 提交事务，将更改保存到数据库。 --- ### 2. 更新已有的数据 要更新数据库中的记录，可以通过查询获取对象并修改其属性，然后提交更改。 ```python # 更新数据 def update_user(user_id, new_username=None, new_email=None): with app.app_context(): user = User.query.get(user_id) if user: if new_username: user.username = new_username if new_email: user.email = new_email db.session.commit() else: print(f"User with id {user_id} not found.") ``` **解释：** - `User.query.get(user_id)` 使用主键查询指定的用户。 - 修改 `user.username` 或 `user.email` 属性以更新用户信息。 - 再次调用 `db.session.commit()` 提交更改。 --- ### 示例代码整合 以下是一个完整的示例，包含创建和更新功能： ```python from flask import Flask from flask_sqlalchemy import SQLAlchemy app = Flask(__name__) app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///example.db' db = SQLAlchemy(app) class User(db.Model): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(80), unique=True, nullable=False) email = db.Column(db.String(120), unique=True, nullable=False) def __repr__(self): return f'<User {self.username}>' # 初始化数据库 with app.app_context(): db.create_all() # 创建新用户 def create_user(username, email): with app.app_context(): new_user = User(username=username, email=email) db.session.add(new_user) db.session.commit() print(f"User {username} created.") # 更新用户 def update_user(user_id, new_username=None, new_email=None): with app.app_context(): user = User.query.get(user_id) if user: if new_username: user.username = new_username if new_email: user.email = new_email db.session.commit() print(f"User {user.id} updated.") else: print(f"User with id {user_id} not found.") if __name__ == '__main__': # 测试创建和更新 create_user('alice', 'alice@example.com') update_user(1, new_username='AliceWonderland', new_email='alice.wonder@example.com') ``` --- ###