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也不失为一种简便的方法。