batch-load-fix

resolvers/follower.py

@@ -11,14 +11,10 @@ from orm.author import Author, AuthorFollower
 from orm.reaction import Reaction
 from orm.shout import Shout, ShoutReactionsFollower
 from orm.topic import Topic, TopicFollower
-from resolvers.author import add_author_stat_columns
+from resolvers.stat import add_author_stat_columns
 from resolvers.community import community_follow, community_unfollow
-from resolvers.topic import (
-    topic_follow,
-    topic_unfollow,
-    add_topic_stat_columns,
-    get_topics_from_query,
-)
+from resolvers.topic import topic_follow, topic_unfollow
+from resolvers.stat import add_topic_stat_columns, get_topics_from_query
 from services.auth import login_required
 from services.db import local_session
 from services.notify import notify_follower

resolvers/stat.py

@@ -0,0 +1,83 @@
+from sqlalchemy import func, distinct
+from sqlalchemy.orm import aliased
+
+from orm.author import Author, AuthorFollower
+from orm.shout import ShoutAuthor, ShoutTopic
+from orm.topic import Topic, TopicFollower
+from services.db import local_session
+# from services.viewed import ViewedStorage
+
+
+def add_author_stat_columns(q):
+    shout_author_aliased = aliased(ShoutAuthor)
+    q = q.outerjoin(shout_author_aliased).add_columns(
+        func.count(distinct(shout_author_aliased.shout)).label('shouts_stat')
+    )
+
+    followers_table = aliased(AuthorFollower)
+    q = q.outerjoin(followers_table, followers_table.author == Author.id).add_columns(
+        func.count(distinct(followers_table.follower)).label('followers_stat')
+    )
+
+    followings_table = aliased(AuthorFollower)
+    q = q.outerjoin(
+        followings_table, followings_table.follower == Author.id
+    ).add_columns(func.count(distinct(followers_table.author)).label('followings_stat'))
+
+    q = q.group_by(Author.id)
+    return q
+
+
+async def get_authors_from_query(q):
+    authors = []
+    with local_session() as session:
+        for [author, shouts_stat, followers_stat, followings_stat] in session.execute(
+            q
+        ):
+            author.stat = {
+                'shouts': shouts_stat,
+                'followers': followers_stat,
+                'followings': followings_stat,
+                # viewed
+            }
+            authors.append(author)
+    return authors
+
+
+def add_topic_stat_columns(q):
+    aliased_shout_author = aliased(ShoutAuthor)
+    aliased_topic_follower = aliased(TopicFollower)
+
+    q = (
+        q.outerjoin(ShoutTopic, Topic.id == ShoutTopic.topic)
+        .add_columns(func.count(distinct(ShoutTopic.shout)).label('shouts_stat'))
+        .outerjoin(aliased_shout_author, ShoutTopic.shout == aliased_shout_author.shout)
+        .add_columns(
+            func.count(distinct(aliased_shout_author.author)).label('authors_stat')
+        )
+        .outerjoin(aliased_topic_follower)
+        .add_columns(
+            func.count(distinct(aliased_topic_follower.follower)).label(
+                'followers_stat'
+            )
+        )
+    )
+
+    q = q.group_by(Topic.id)
+
+    return q
+
+
+async def get_topics_from_query(q):
+    topics = []
+    with local_session() as session:
+        for [topic, shouts_stat, authors_stat, followers_stat] in session.execute(q):
+            topic.stat = {
+                'shouts': shouts_stat,
+                'authors': authors_stat,
+                'followers': followers_stat,
+                # 'viewed': await ViewedStorage.get_topic(topic.slug),
+            }
+            topics.append(topic)
+
+    return topics

resolvers/topic.py

@@ -1,55 +1,15 @@
 from sqlalchemy import and_, distinct, func, select
-from sqlalchemy.orm import aliased
 
 from orm.author import Author
-from orm.shout import ShoutAuthor, ShoutTopic
+from orm.shout import ShoutTopic
 from orm.topic import Topic, TopicFollower
+from resolvers.stat import add_topic_stat_columns, get_topics_from_query
 from services.auth import login_required
 from services.db import local_session
 from services.schema import mutation, query
-from services.viewed import ViewedStorage
 from services.logger import root_logger as logger
 
 
-def add_topic_stat_columns(q):
-    aliased_shout_author = aliased(ShoutAuthor)
-    aliased_topic_follower = aliased(TopicFollower)
-
-    q = (
-        q.outerjoin(ShoutTopic, Topic.id == ShoutTopic.topic)
-        .add_columns(func.count(distinct(ShoutTopic.shout)).label('shouts_stat'))
-        .outerjoin(aliased_shout_author, ShoutTopic.shout == aliased_shout_author.shout)
-        .add_columns(
-            func.count(distinct(aliased_shout_author.author)).label('authors_stat')
-        )
-        .outerjoin(aliased_topic_follower)
-        .add_columns(
-            func.count(distinct(aliased_topic_follower.follower)).label(
-                'followers_stat'
-            )
-        )
-    )
-
-    q = q.group_by(Topic.id)
-
-    return q
-
-
-async def get_topics_from_query(q):
-    topics = []
-    with local_session() as session:
-        for [topic, shouts_stat, authors_stat, followers_stat] in session.execute(q):
-            topic.stat = {
-                'shouts': shouts_stat,
-                'authors': authors_stat,
-                'followers': followers_stat,
-                'viewed': await ViewedStorage.get_topic(topic.slug),
-            }
-            topics.append(topic)
-
-    return topics
-
-
 @query.field('get_topics_all')
 async def get_topics_all(_, _info):
     q = select(Topic)