SQLAlchemy复杂查询

最近个人用python + flask搞了一个小项目，ORM用到的是SQLAlchemy。

　　 SQLAlchemy的查询方式非常灵活，你所能想像到的复杂SQL 语句，基本上都可以实现。这里简单的总结一下常用的查询技巧。

多条件组合，可以用and_，or_实现。最外层时，and_可以省略，默认用逗号分开条件。

db.session.query(User).filter(

        and_(

            or_(User.name==name1,User.name==name2),

            or_(User.status==1,User.status==2)

        ),

        User.active==1

    ).first()

动态组合条件。针对不同的场景，可能需要不同的查询条件，类似动态的拼接SQL 语句。

        if filter_type == 1:

            search = and_(GameRoom.status ==1,or_(

                and_(GameRoom.white_user_id == user_id,

                     GameRoom.active_player == 1),

                and_(GameRoom.black_user_id == user_id,

                     GameRoom.active_player == 0)))

        elif filter_type == 2:

            search = and_(GameRoom.status ==1,or_(

                and_(GameRoom.white_user_id == user_id,

                     GameRoom.active_player == 0),

                and_(GameRoom.black_user_id == user_id,

                     GameRoom.active_player == 1)))

        elif filter_type == 3:

            search = GameRoom.create_by == user_id

        db.session.query(GameRoom).filter(search).all()

关联查询。对应SQL的join和left join等。

    session.query(User, Address).filter(User.id == Address.user_id).all()

    session.query(User).join(User.addresses).all()

    session.query(User).outerjoin(User.addresses).all()

使用别名用aliased，aliased在orm包中。当要对同一个表使用多次关联时，可能需要用到别名。同时，如果查询的结果有多个同名的字段，可以使用label重命名。

black_user = orm.aliased(User)

white_user = orm.aliased(User)

db.session.query(

            GameRoom,

            black_user.score.label("black_score"),

            white_user.score.label("white_score")

            ).outerjoin(black_user,GameRoom.black_user_id==black_user.user_id).outerjoin(

                white_user,GameRoom.white_user_id==white_user.user_id).filter(

                    GameRoom.id==room_id

            ).all()

聚合查询和使用数据库函数。func可以调用各种聚合函数，和当前数据库支持的其它函数。

session.query(User.name, func.count('*').label("user_count")).group_by(User.name).all()

session.query(User.name, func.sum(User.id).label("user_id_sum")).filter(func.to_days(User.create_date)==func.to_days(func.now())).group_by(User.name).all()

子查询。　

stmt = db.session.query(Address.user_id, func.count('*').label("address_count")).group_by(Address.user_id).subquery()

db.session.query(User, stmt.c.address_count).outerjoin((stmt, User.id == stmt.c.user_id)).order_by(User.id).all()

直接运行SQL语句查询。如果查询实在太复杂，觉得用SQLAlchemy查询方式很难实现，或者要通过存储过程实现查询，可以让SQLAlchemy直接运行SQL语句返回结果。

        sql ="""select b.user_id,b.user_name,b.icon,b.score,a.add_score from

            (select user_id, sum(score_new - score_old) as add_score from user_score_log

            where year(create_date)=year(now()) and month(create_date)=month(now())

            group by user_id) a join users b on a.user_id=b.user_id

            order by a.add_score desc limit 50"""

        list_top = db.session.execute(sql).fetchall()

分页查询。sqlalchemy中分页用到pagination，先不说性能怎么样，使用起来是真的非常方便。

        pagination = GameMessage.query.filter(GameMessage.game_id==game_id).\

            order_by(GameMessage.id.desc()).\

            paginate(page, per_page=20, error_out=True)

        pages = pagination.pages

        total = pagination.total

        items = pagination.items

　　总的来说，SQLAlchemy是我用过的最好的ORM 框架之一（其实我最熟的是.net，python也只用过这一个ORM工具）。基本上常见的SQL 语句，这里都已经实现。如果你的查询实在太复杂的，可能需要用存储过程来实现，直接运行SQL也不失为一种简便的方法。

秒客网

SQLAlchemy复杂查询

相关文章