0.4.5-api-update

resolvers/community.py

@@ -1,36 +1,31 @@
-from sqlalchemy import select
-
 from orm.author import Author
-from orm.community import Community
+from orm.community import Community, CommunityFollower
 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": shouts_stat,
-                "followers": followers_stat,
-                # "authors": 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)
+    return local_session().query(Community).all()
 
 
 @query.field("get_community")
 async def get_community(_, _info, slug: str):
-    q = select(Community).where(Community.slug == slug)
+    q = local_session().query(Community).where(Community.slug == slug)
+    return q.first()
 
-    communities = get_communities_from_query(q)
-    return communities[0]
+
+@query.field("get_communities_by_author")
+async def get_communities_by_author(_, _info, slug="", user="", author_id=0):
+    with local_session() as session:
+        q = session.query(Community).join(CommunityFollower)
+        if slug:
+            author_id = session.query(Author).where(Author.slug == slug).first().id
+            q = q.where(CommunityFollower.author == author_id)
+        if user:
+            author_id = session.query(Author).where(Author.user == user).first().id
+            q = q.where(CommunityFollower.author == author_id)
+        if author_id:
+            q = q.where(CommunityFollower.author == author_id)
+        return q.all()
+    return []