Início / coding / Perguntas / 79496102
Accepted
robertspierre
Asked: 2025-03-09 22:59:55 +0800 CST

SQLAlchemy usa "IN" para selecionar correspondência em pares

  • 772

Considere o seguinte BD:

from sqlalchemy import String, select, create_engine
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, Session

class Base(DeclarativeBase):
    pass

class User(Base):
    __tablename__ = "user_account"
    name: Mapped[str] = mapped_column(String(30))
    surname: Mapped[str] = mapped_column(String(30))
    def __repr__(self):
        return f"User(name={self.name!r}, surname={self.surname!r})"

engine = create_engine("sqlite+pysqlite:///test3.sqlite3", echo=True)

Base.metadata.create_all(engine)

with Session(engine) as session:
    user1 = User(
        name="Mario",
        surname="Rossi"
    )
    user2 = User(
        name="Mario",
        surname="Bianchi",
    )
    user3 = User(
        name="Giovanni",
        surname="Bianchi",
    )
    session.add_all([user1, user2, user3])
    session.commit()

Agora suponha que eu tenha uma lista de usuários que desejo encontrar:

users = [("Mario", "Rossi"), ("Giovanni", "Bianchi")]

Então eu correria:

names = [name for name, _ in users]
surnames = [surname for _, surname in users]
with Session(engine) as session:
    stmt = select(User).where(User.name.in_(names)).where(User.surname.in_(surnames))
    print(session.execute(stmt).scalars().all())

que retorna:

[User(name='Mario', surname='Rossi'), User(name='Mario', surname='Bianchi'), User(name='Giovanni', surname='Bianchi')]

mas "Mario Bianchi" não estava na lista de usuários de entrada que eu tinha.

Como posso concatenar instruções IN para selecionar apenas correspondência em pares?

Ou seja, se eu tiver varA in (el1, el2) AND varB IN (el3, el4), não quero selecionar entradas comvarA==el1 AND varB==el4

python

1 respostas

  1. Best Answer

    Você precisa usar ORuma condição booleana mais explícita sqlalchemy.or_(conditions)que garanta que cada par (nome, sobrenome) seja verificado explicitamente.

    from sqlalchemy import or_

with Session(engine) as session:
    stmt = select(User).where(
        or_(*[(User.name == name) & (User.surname == surname) for name, surname in users])
    )
    print(session.execute(stmt).scalars().all())

    Isso funciona em todos os bancos de dados, incluindo SQLite.

    Otimização específica do PostgreSQL

    Se você estiver usando PostgreSQL, podemos usar tuple_para uma consulta mais elegante e otimizada:

    from sqlalchemy.sql.expression import tuple_

with Session(engine) as session:
    stmt = select(User).where(tuple_(User.name, User.surname).in_(users))
    print(session.execute(stmt).scalars().all())
    • 4

relate perguntas