顾乔芝士网

持续更新的前后端开发技术栈

使用 SQLAlchemy 实现 Python 数据库操作的 ORM 完整指南

SQLAlchemy 是一个功能强大的 Python 库,支持关系型数据库的操作,并且通过 ORM(对象关系映射)方式,可以让开发者以操作 Python 对象的方式来管理数据库。


1. 安装 SQLAlchemy

首先确保安装了 SQLAlchemy:

bash



pip install sqlalchemy



如果需要支持具体数据库,比如 MySQL 或 PostgreSQL,还需要安装对应的驱动:

bash



pip install pymysql # 用于 MySQL

pip install psycopg2 # 用于 PostgreSQL




2. ORM 基本使用流程

(1)创建数据库引擎

python



from sqlalchemy import create_engine



# 替换为具体的数据库信息

DATABASE_URL = "mysql+pymysql://username:password@localhost/db_name"



# 创建数据库引擎

engine = create_engine(DATABASE_URL, echo=True) # echo=True 会输出 SQL 日志




(2)定义数据表模型

SQLAlchemy 使用 declarative_base() 来创建 ORM 模型的基类。

python



from sqlalchemy.orm import declarative_base

from sqlalchemy import Column, Integer, String



# 创建 ORM 基类

Base = declarative_base()



# 定义一个表模型

class User(Base):

__tablename__ = 'users' # 数据表名

id = Column(Integer, primary_key=True, autoincrement=True) # 主键

name = Column(String(50), nullable=False) # 用户名

email = Column(String(100), unique=True, nullable=False) # 邮箱



def __repr__(self):

return f""




(3)创建数据表

使用 Base.metadata.create_all() 方法在数据库中创建表。

python



# 创建表

Base.metadata.create_all(engine)




(4)创建会话(Session)

Session 是与数据库交互的接口。

python



from sqlalchemy.orm import sessionmaker



# 创建会话工厂

Session = sessionmaker(bind=engine)



# 创建一个会话

session = Session()




(5)执行 CRUD 操作

添加数据

python



# 创建新用户

new_user = User(name="Alice", email="alice@example.com")



# 添加到会话

session.add(new_user)



# 提交事务

session.commit()



查询数据

python



# 查询所有用户

users = session.query(User).all()

for user in users:

print(user)



# 根据条件查询

user = session.query(User).filter_by(name="Alice").first()

print(user)



更新数据

python



# 查找用户

user = session.query(User).filter_by(name="Alice").first()



# 更新用户信息

if user:

user.email = "alice_new@example.com"

session.commit()



删除数据

python



# 查找用户

user = session.query(User).filter_by(name="Alice").first()



# 删除用户

if user:

session.delete(user)

session.commit()




3. 小贴士

  1. 数据库 URL 格式:
  2. MySQL: mysql+pymysql://username:password@host:port/database
  3. PostgreSQL: postgresql+psycopg2://username:password@host:port/database
  4. 自动生成主键: Column(Integer, primary_key=True, autoincrement=True)
  5. 调试日志: 设置 echo=True 可以输出 SQL 日志,方便调试。
  6. 异常处理: 使用 try...except 捕获数据库异常,确保数据完整性。
控制面板
您好,欢迎到访网站!
  查看权限
网站分类
最新留言