SQLAlchemy 数据库查询操作详解|极客教程

SQLAlchemy 数据库查询操作详解

在开发中，经常会涉及到数据库操作，其中查询操作是使用频率比较高的一种。SQLAlchemy是一个流行的Python ORM库，它提供了丰富的API来操作数据库。本文将详细介绍使用SQLAlchemy进行数据库查询操作的各种方法。

创建数据库连接

首先，我们需要创建一个数据库连接，使用SQLAlchemy的create_engine函数来实现。在创建连接时，需要指定数据库的连接信息，如数据库类型、用户名、密码、主机地址和端口号等。

from sqlalchemy import create_engine

# 创建数据库连接
engine = create_engine('数据库连接字符串')

定义数据表映射类

在进行数据库查询操作之前，我们需要先定义数据表和数据表中的字段映射类。这些映射类可以继承自SQLAlchemy的Base类，并使用declarative_base函数进行声明。

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String)
    age = Column(Integer)

创建会话对象

接下来，我们需要创建一个会话对象来执行数据库查询操作。会话对象可以通过sessionmaker函数创建，同时需要使用bind参数绑定到数据库连接上。

from sqlalchemy.orm import sessionmaker

Session = sessionmaker(bind=engine)
session = Session()

执行简单查询

查询所有数据

我们可以使用query函数来执行简单的查询操作，通过调用all方法可以查询所有数据。

from sqlalchemy import select

# 查询所有数据
results = session.query(User).all()
for user in results:
    print(user.id, user.name, user.age)

查询指定条件数据

除了查询所有数据外，我们还可以根据指定条件来进行查询。通过使用filter函数可以添加查询条件。

# 查询年龄大于20的用户
results = session.query(User).filter(User.age > 20).all()
for user in results:
    print(user.id, user.name, user.age)

使用原始SQL查询

有时候，我们可能需要执行一些复杂的SQL查询操作，这时可以使用原始SQL语句来执行查询。可以通过execute函数来执行原始SQL查询。

# 原始SQL查询
result = session.execute('SELECT * FROM users')
for row in result:
    print(row)

关联查询

在数据库中，有时候数据表之间会存在关联关系，我们可以通过relationship函数来进行关联查询。

from sqlalchemy import ForeignKey
from sqlalchemy.orm import relationship

class Order(Base):
    __tablename__ = 'orders'

    id = Column(Integer, primary_key=True)
    user_id = Column(Integer, ForeignKey('users.id'))
    amount = Column(Integer)

    # 定义与User表的关联关系
    user = relationship("User")

分页查询

当数据量比较大时，我们可能需要对查询结果进行分页显示。可以使用limit和offset函数来进行分页查询。

# 分页查询
results = session.query(User).limit(10).offset(0).all()
for user in results:
    print(user.id, user.name, user.age)

聚合查询

有时候，我们需要对查询结果进行一些统计操作，比如求和、平均值等。可以使用func函数来进行聚合查询。

from sqlalchemy import func

# 聚合查询
result = session.query(func.count(User.id)).scalar()
print(result)

排序查询

查询结果可能并不是按照我们期望的顺序排列，这时可以使用order_by函数来进行排序查询。

# 排序查询
results = session.query(User).order_by(User.age.desc()).all()
for user in results:
    print(user.id, user.name, user.age)

总结

通过本文的介绍，我们了解了如何使用SQLAlchemy进行数据库查询操作。SQLAlchemy提供了丰富的API来满足不同查询需求，同时也支持原始SQL查询操作。在实际开发中，通过合理使用SQLAlchemy的查询方法，我们可以轻松地实现各种数据库查询操作。

SQLAlchemy 数据库查询操作详解