core/resolvers/community.py
2024-06-05 17:45:55 +03:00

40 lines
1.3 KiB
Python

from sqlalchemy import distinct, func, select
from orm.author import Author
from orm.community import Community
from orm.shout import ShoutCommunity
from services.db import local_session
from services.schema import query
def get_communities_from_query(q):
ccc = []
with local_session() as session:
for [c, shouts_stat, followers_stat] in session.execute(q):
c.stat = {
"shouts": session.execute(
select(func.count(distinct(ShoutCommunity.shout))).filter(ShoutCommunity.community == c.id)
),
# "authors": session.execute(select(func.count(distinct(ShoutCommunity.shout))).filter(ShoutCommunity.community == c.id)),
# "followers": session.execute(select(func.count(distinct(ShoutCommunity.shout))).filter(ShoutCommunity.community == c.id)),
# "commented": commented_stat,
}
ccc.append(c)
return ccc
@query.field("get_communities_all")
async def get_communities_all(_, _info):
q = select(Author)
return get_communities_from_query(q)
@query.field("get_community")
async def get_community(_, _info, slug: str):
q = select(Community).where(Community.slug == slug)
communities = get_communities_from_query(q)
return communities[0]