嵌套联接子查询的SQLAlchemy计数函数

Question

I'm very new at SQLAlchemy and as you can see I have 3 models:

categories = db.Table(
    'categories',
    db.Column('post_id', db.Integer, db.ForeignKey('post.id')),
    db.Column('category_id', db.Integer, db.ForeignKey('category.id'))
)


class User(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     nickname = db.Column(db.String(64), index=True, unique=True)
     email = db.Column(db.String(120), index=True, unique=True)
     posts = db.relationship('Post', backref='author', lazy='dynamic')
     about_me = db.Column(db.String(140))
     last_seen = db.Column(db.DateTime)


class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    body = db.Column(db.String(140))
    timestamp = db.Column(db.DateTime)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    categories = db.relationship("Category",
                             secondary="categories",
                             backref="posts")



class Category(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100))

I can count how many post each user has by some subquery like this:

stmt = db.session.query(models.Post.user_id, func.count('*').label('post_count')).group_by(models.Post.user_id).subquery()

db.session.query(models.User, stmt.c.post_count).outerjoin(stmt, models.User.id==stmt.c.user_id).order_by(models.User.id)

but I can't figure it out the right query to calculate how many categories each users has.

Answer 1

You could for example add a join to categories in your subquery and count distinct Post.id s and category_id s grouped by user_id :

stmt = db.session.query(
    models.Post.user_id,
    func.count(models.Post.id.distinct()).
        label('post_count'),
    func.count(models.categories.c.category_id.distinct()).
        label('category_count')
).\
    join(models.categories).\
    group_by(models.Post.user_id).\
    subquery()

db.session.query(models.User,
                 stmt.c.post_count,
                 stmt.c.category_count).\
    outerjoin(stmt, models.User.id==stmt.c.user_id).\
    order_by(models.User.id)

The post id's have to be distinct as well in case a post has multiple categories, which will manifest as multiple rows for the post after the join.