SQLAlchemy: filtro por la pertenencia a al menos uno-muchos-a-muchos mesa

Question

Usando SQLAlchemy 0.7.1 y una base de datos MySQL 5.1 relacionado, Tengo una relación muchos-a-muchos relación establecido de la siguiente manera:

user_groups = Table('user_groups', Base.metadata, 
    Column('user_id', String(128), ForeignKey('users.username')), 
    Column('group_id', Integer, ForeignKey('groups.id')) 
) 

class ZKUser(Base, ZKTableAudit): 
    __tablename__ = 'users' 

    username = Column(String(128), primary_key=True) 
    first_name = Column(String(512)) 
    last_name = Column(String(512)) 

    groups = relationship(ZKGroup, secondary=user_groups, backref='users') 

class ZKGroup(Base, ZKTableAudit): 
    __tablename__ = 'groups' 

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

Los usuarios pueden pertenecer a múltiples grupos, y los grupos pueden contener varios usuarios.

Lo que intento hacer es crear una consulta SQLAlchemy que devuelva solo a los Usuarios que pertenecen a al menos un Grupo de una lista de Grupos.

Jugué con la función in_, pero eso solo parece funcionar para probar valores escalares para la membresía en una lista. No soy muy escritor de SQL, así que ni siquiera sé qué tipo de declaración SELECT requeriría.

Answer 1

bien, después de mucha investigación, me di cuenta de que era mi propia ignorancia de la terminología de SQL que me estaba frenando. Mi búsqueda de una solución para encontrar usuarios que pertenezcan a "al menos uno de" la lista de grupos debería haber sido encontrar usuarios que pertenezcan a "cualquiera" de la lista de grupos. La función any ORM de SQLAlchemy hace exactamente lo que necesitaba, así:

session.query(ZKUser).filter(ZKUser.groups.any(ZKGroup.id.in_([1,2,3]))) 

Ese código emite este SQL (en MySQL 5.1):

SELECT * FROM users 
WHERE EXISTS (
    SELECT 1 FROM user_groups, groups 
    WHERE users.id = user_groups.contact_id 
     AND groups.id = user_groups.group_id 
     AND groups.id IN (%s, %s, %s) 
    ) 

Answer 2

Puede utilizar in_:

session.query(ZKUser).filter(ZKGroup.id.in_([1,2])).all() 

Answer 3

Según the docs for any, la consulta se ejecutará más rápido si se utiliza una explícita join lugar:

Debido a que cualquier() utiliza una subconsulta correlacionada, su rendimiento no es tan bueno cuando se compara con grandes tablas de destino como el de usar una unión.

En su caso, usted podría hacer algo como:

users = (
    session.query(ZKUser) 
    .join(user_groups) 
    .filter(user_groups.columns.group_id.in_([1, 2, 3])) 
) 

Este emite SQL como:

SELECT * 
FROM users 
JOIN user_groups ON users.id = user_groups.user_id 
WHERE user_groups.group_id IN (1, 2, 3)