core/resolvers/feed.py

from auth.authenticate import login_required
from base.orm import local_session
from base.resolvers import query
from sqlalchemy import and_, desc
from orm.shout import Shout, ShoutAuthor, ShoutTopic
from orm.topic import TopicFollower
from orm.user import AuthorFollower
from typing import List


@query.field("shoutsForFeed")
@login_required
def get_user_feed(_, info, page, size) -> List[Shout]:
    user = info.context["request"].user
    shouts = []
    with local_session() as session:
        shouts = (
            session.query(Shout)
            .join(ShoutAuthor)
            .join(AuthorFollower)
            .where(AuthorFollower.follower == user.slug)
            .order_by(desc(Shout.createdAt))
        )
        topicrows = (
            session.query(Shout)
            .join(ShoutTopic)
            .join(TopicFollower)
            .where(TopicFollower.follower == user.slug)
            .order_by(desc(Shout.createdAt))
        )
        shouts = shouts.union(topicrows).limit(size).offset(page * size).all()
    return shouts


@query.field("myCandidates")
@login_required
async def user_unpublished_shouts(_, info, page=1, size=10) -> List[Shout]:
    user = info.context["request"].user
    shouts = []
    with local_session() as session:
        shouts = (
            session.query(Shout)
            .join(ShoutAuthor)
            .where(and_(Shout.publishedAt == None, ShoutAuthor.user == user.slug))
            .order_by(desc(Shout.createdAt))
            .limit(size)
            .offset(page * size)
            .all()
        )
    return shouts