python - query - sqlalchemy relationship

Construyendo dinámicamente filtros en SQLAlchemy

Question

Estoy buscando una manera de construir filtros dinámicamente usando SQLAlchemy. Es decir, dada la columna, el nombre del operador y el valor de comparación, construya el filtro correspondiente.

Trataré de ilustrar usando un ejemplo (esto se usaría para construir una API). Digamos que tenemos el siguiente modelo:

class Cat(Model):
id = Column(Integer, primary_key=True)
name = Column(String)
age = Column(Integer)

Me gustaría mapear las consultas a los filtros. Por ejemplo,

• /cats?filter=age;eq;3 debe generar Cat.query.filter(Cat.age == 3)

• /cats?filter=age;in;5,6,7&filter=id;ge;10 debe generar Cat.query.filter(Cat.age.in_([5, 6, 7])).filter(Cat.id >= 10)

Miré a mi alrededor para ver cómo se había hecho, pero no pude encontrar una manera que no implicara mapear manualmente el nombre de cada operador a un comparador o algo similar. Por ejemplo, Flask-Restless mantiene un diccionario de todas las operaciones compatibles y almacena las funciones lambda correspondientes ( código aquí ).

Busqué en los documentos de SQLAlchemy y encontré dos pistas potenciales, pero ninguna me pareció satisfactoria:

• utilizando Column.like , Column.in_ ...: estos operadores están disponibles directamente en la columna, lo que simplificaría el uso de getattr pero aún faltan algunos ( == , > , etc.).

• usando Column.op : por ejemplo, Cat.name.op(''='')(''Hobbes'') pero esto no parece funcionar para todos los operadores ( in concreto).

¿Hay una manera limpia de hacer esto sin funciones lambda ?

Answer 1

En caso de que esto sea útil para alguien, esto es lo que terminé haciendo:

from flask import request
class Parser(object):
sep = '';''
# ...
def filter_query(self, query):
model_class = self._get_model_class(query) # returns the query''s Model
raw_filters = request.args.getlist(''filter'')
for raw in raw_filters:
try:
key, op, value = raw.split(self.sep, 3)
except ValueError:
raise APIError(400, ''Invalid filter: %s'' % raw)
column = getattr(model_class, key, None)
if not column:
raise APIError(400, ''Invalid filter column: %s'' % key)
if op == ''in'':
filt = column.in_(value.split('',''))
else:
try:
attr = filter(
lambda e: hasattr(column, e % op),
[''%s'', ''%s_'', ''__%s__'']
)[0] % op
except IndexError:
raise APIError(400, ''Invalid filter operator: %s'' % op)
if value == ''null'':
value = None
filt = getattr(column, attr)(value)
query = query.filter(filt)
return query

Esto cubre todos los comparadores de columnas de SQLAlchemy:

• eq para ==
• lt para <
• ge para >=
• in para in
• like por me like
• etc.

La lista exhaustiva con sus nombres correspondientes se puede encontrar here .

Answer 2

Puede usar sqlalchemy-elasticquery para construir filtros dinámicamente usando SQLAlchemy.

?filters={ "age" : 3 }

Answer 3

Un truco útil al construir un filtro de expresión múltiple:

filter_group = list(Column.in_(''a'',''b''),Column.like(''%a''))
query = query.filter(and_(*filter_group))

El uso de este enfoque le permitirá combinar expresiones con y / o lógica. También esto te permitirá evitar llamadas recursivas como en tu respuesta.