SQLAlchemy选择某个字段

Posted 2023-03-31 小龙在山东

要选择某个字段，您可以使用SQLAlchemy中的select函数。例如，假设您有一个名为users的表格，其中包含id，name和age字段。要选择name字段，您可以执行以下操作：

from sqlalchemy import create_engine, Table, Column, Integer, String, MetaData

# 创建一个连接到数据库的引擎
engine = create_engine('sqlite:///example.db', echo=True)

# 定义一个表示users表的元数据对象
metadata = MetaData()

# 定义一个表示users表的对象
users = Table('users', metadata,
    Column('id', Integer, primary_key=True),
    Column('name', String),
    Column('age', Integer),
)

# 创建一个select查询，选择name字段
select_stmt = users.select().with_only_columns([users.c.name])

# 执行查询
with engine.connect() as conn:
    result = conn.execute(select_stmt).fetchall()

# 输出查询结果
print(result)

上述代码将选择users表中的name字段，并将其打印到控制台上。
如果您想选择多个字段，可以在with_only_columns方法中传入一个列表，其中包含要选择的字段。例如，如果您想选择name和age字段，可以这样做：

# 创建一个select查询，选择name和age字段
select_stmt = users.select().with_only_columns([users.c.name, users.c.age])

# 执行查询
with engine.connect() as conn:
    result = conn.execute(select_stmt).fetchall()

# 输出查询结果
print(result)

上述代码将选择users表中的name和age字段，并将它们打印到控制台上。

您还可以使用select函数的select_from方法指定要查询的表。例如，如果您有一个名为orders的表，其中包含user_id和total_price字段，您可以选择users表中的name字段和orders表中的total_price字段，如下所示：

# 定义一个表示orders表的对象
orders = Table('orders', metadata,
    Column('id', Integer, primary_key=True),
    Column('user_id', Integer),
    Column('total_price', Integer),
)

# 创建一个select查询，选择name和total_price字段，同时查询users和orders表
select_stmt = select([users.c.name, orders.c.total_price]).select_from(users.join(orders, users.c.id == orders.c.user_id))

# 执行查询
with engine.connect() as conn:
    result = conn.execute(select_stmt).fetchall()

# 输出查询结果
print(result)

上述代码将选择users表中的name字段和orders表中的total_price字段，并将它们打印到控制台上。注意，我们使用join方法将users表和orders表连接在一起，并使用select_from方法指定要查询的表。

ORM查询某个字段

要查询ORM中的某个字段，您可以使用Query对象中的with_entities方法。例如，假设您有一个名为User的ORM模型，其中包含id，name和age字段。要查询name字段，您可以执行以下操作：

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

# 创建一个连接到数据库的引擎
engine = create_engine('sqlite:///example.db', echo=True)

# 创建一个Session工厂
Session = sessionmaker(bind=engine)

# 定义一个基类
Base = declarative_base()

# 定义一个表示User模型的类
class User(Base):
    __tablename__ = 'users'

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

# 创建一个Session
session = Session()

# 查询name字段
result = session.query(User.name).all()

# 输出查询结果
print(result)

上述代码将查询User模型中的name字段，并将其打印到控制台上。注意，在query方法中传递了User.name作为参数，这意味着我们只查询name字段。

如果您想查询多个字段，可以在with_entities方法中传入一个列表，其中包含要查询的字段。例如，如果您想查询name和age字段，可以这样做：

# 查询name和age字段
result = session.query(User.name, User.age).all()

# 输出查询结果
print(result)

上述代码将查询User模型中的name和age字段，并将它们打印到控制台上。

注意，在ORM中，您还可以使用属性访问符（.）来访问模型的属性。例如，上面的示例可以简化为以下代码：

# 查询name和age字段
result = session.query(User.name, User.age)

# 输出查询结果
print(result.all())

在这种情况下，我们可以直接在query方法中使用模型的属性来查询字段。
如果您想查询多个模型的字段，可以使用join方法将多个模型连接在一起，并在with_entities方法中传递一个列表，其中包含要查询的字段。例如，假设您有一个名为Order的ORM模型，其中包含id，user_id和total_price字段，您可以选择User模型中的name字段和Order模型中的total_price字段，如下所示：

# 定义一个表示Order模型的类
class Order(Base):
    __tablename__ = 'orders'

    id = Column(Integer, primary_key=True)
    user_id = Column(Integer)
    total_price = Column(Integer)

# 查询name和total_price字段，同时查询User和Order模型
result = session.query(User.name, Order.total_price).join(Order, User.id == Order.user_id).all()

# 输出查询结果
print(result)

上述代码将选择User模型中的name字段和Order模型中的total_price字段，并将它们打印到控制台上。注意，我们使用join方法将User模型和Order模型连接在一起，并在with_entities方法中传递一个列表，其中包含要查询的字段。