Mysql
 sql >> Base de Dados >  >> RDS >> Mysql

Converter uma consulta SQL complexa em SQLAlchemy


Seu HAVING é tratado corretamente, mas você está passando a expressão errada. Parece que você está usando Python 2, pois a comparação relacional entre uma string e um inteiro
'distance' < 25

não gera uma exceção, mas avalia como False em vez de. Em outras palavras, sua consulta é igual a
locations = db.session.query(...).having(False).all()

o que explica por que você obtém zero resultados:todas as linhas são explicitamente filtradas pela cláusula HAVING, conforme visto na versão impressa:
...
HAVING false = 1  -- remove all rows

Uma solução é usar uma construção adequada, como column() , para produzir a expressão:
locations = db.session.query(...).having(column('distance') < 25).all()

Você não deve envolver a expressão complexa do item da lista de seleção em um select() , que representa uma instrução SELECT. Ou rotule o text() como é:
text('( 6371 * acos( cos( radians("53.6209798282177") ) * '
     'cos( radians( lat ) ) * cos( radians( lng ) - radians("13.96948162900808") ) + '
     'sin( radians("53.6209798282177") ) * sin( radians( lat ) ) ) ) '
     'AS distance')

ou construa a expressão usando o modelo:
(6371 *
 func.acos(func.cos(func.radians(53.6209798282177)) *
           func.cos(func.radians(Location.lat)) *
           func.cos(func.radians(Location.lng) - func.radians(13.96948162900808)) +
           func.sin(func.radians(53.6209798282177)) *
           func.sin(func.radians(Location.lat)))).label('distance')

Você pode melhorar a legibilidade da construção da sua consulta criando uma função para a distância do grande círculo a> , e com um pouco de trabalho você pode implementar um método híbrido em Location :
import math

def gc_distance(lat1, lng1, lat2, lng2, math=math):
    ang = math.acos(math.cos(math.radians(lat1)) *
                    math.cos(math.radians(lat2)) *
                    math.cos(math.radians(lng2) -
                             math.radians(lng1)) +
                    math.sin(math.radians(lat1)) *
                    math.sin(math.radians(lat2)))

    return 6371 * ang

class Location(db.Model):
    ...
    @hybrid_method
    def distance(self, lat, lng):
        return gc_distance(lat, lng, self.lat, self.lng)

    @distance.expression
    def distance(cls, lat, lng):
        return gc_distance(lat, lng, cls.lat, cls.lng, math=func)

locations = db.session.query(
        Location,
        Location.distance(53.6209798282177,
                          13.96948162900808).label('distance')).\
    having(column('distance') < 25).\
    order_by('distance').\
    all()

Observe que a maneira como você usa HAVING para eliminar linhas que não são de grupo não é portátil. Por exemplo, no Postgresql, a presença da cláusula HAVING transforma uma consulta em uma consulta agrupada, mesmo sem uma cláusula GROUP BY. Você pode usar uma subconsulta:
stmt = db.session.query(
        Location,
        Location.distance(53.6209798282177,
                          13.96948162900808).label('distance')).\
    subquery()

location_alias = db.aliased(Location, stmt)

locations = db.session.query(location_alias).\
    filter(stmt.c.distance < 25).\
    order_by(stmt.c.distance).\
    all()