SQLAlchemy复杂查询


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

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

  1. 多条件组合,可以用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()
  2. 动态组合条件。针对不同的场景,可能需要不同的查询条件,类似动态的拼接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()
  3. 关联查询。对应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()
  4. 使用别名用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()
  5. 聚合查询和使用数据库函数。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()
  6.  子查询。 

    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()
  7. 直接运行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()
  8. 分页查询。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也不失为一种简便的方法。