Improve topic sorting: add popular sorting by publications and authors count

resolvers/community.py

@@ -1,3 +1,7 @@
+from typing import Any
+
+from graphql import GraphQLResolveInfo
+
 from auth.orm import Author
 from orm.community import Community, CommunityFollower
 from services.db import local_session
@@ -5,18 +9,20 @@ from services.schema import mutation, query
 
 
 @query.field("get_communities_all")
-async def get_communities_all(_, _info):
+async def get_communities_all(_: None, _info: GraphQLResolveInfo) -> list[Community]:
     return local_session().query(Community).all()
 
 
 @query.field("get_community")
-async def get_community(_, _info, slug: str):
+async def get_community(_: None, _info: GraphQLResolveInfo, slug: str) -> Community | None:
     q = local_session().query(Community).where(Community.slug == slug)
     return q.first()
 
 
 @query.field("get_communities_by_author")
-async def get_communities_by_author(_, _info, slug="", user="", author_id=0):
+async def get_communities_by_author(
+    _: None, _info: GraphQLResolveInfo, slug: str = "", user: str = "", author_id: int = 0
+) -> list[Community]:
     with local_session() as session:
         q = session.query(Community).join(CommunityFollower)
         if slug:
@@ -32,20 +38,20 @@ async def get_communities_by_author(_, _info, slug="", user="", author_id=0):
 
 
 @mutation.field("join_community")
-async def join_community(_, info, slug: str):
+async def join_community(_: None, info: GraphQLResolveInfo, slug: str) -> dict[str, Any]:
     author_dict = info.context.get("author", {})
     author_id = author_dict.get("id")
     with local_session() as session:
         community = session.query(Community).where(Community.slug == slug).first()
         if not community:
             return {"ok": False, "error": "Community not found"}
-        session.add(CommunityFollower(community=community.id, author=author_id))
+        session.add(CommunityFollower(community=community.id, follower=author_id))
         session.commit()
         return {"ok": True}
 
 
 @mutation.field("leave_community")
-async def leave_community(_, info, slug: str):
+async def leave_community(_: None, info: GraphQLResolveInfo, slug: str) -> dict[str, Any]:
     author_dict = info.context.get("author", {})
     author_id = author_dict.get("id")
     with local_session() as session:
@@ -57,7 +63,7 @@ async def leave_community(_, info, slug: str):
 
 
 @mutation.field("create_community")
-async def create_community(_, info, community_data):
+async def create_community(_: None, info: GraphQLResolveInfo, community_data: dict[str, Any]) -> dict[str, Any]:
     author_dict = info.context.get("author", {})
     author_id = author_dict.get("id")
     with local_session() as session:
@@ -67,7 +73,7 @@ async def create_community(_, info, community_data):
 
 
 @mutation.field("update_community")
-async def update_community(_, info, community_data):
+async def update_community(_: None, info: GraphQLResolveInfo, community_data: dict[str, Any]) -> dict[str, Any]:
     author_dict = info.context.get("author", {})
     author_id = author_dict.get("id")
     slug = community_data.get("slug")
@@ -85,7 +91,7 @@ async def update_community(_, info, community_data):
 
 
 @mutation.field("delete_community")
-async def delete_community(_, info, slug: str):
+async def delete_community(_: None, info: GraphQLResolveInfo, slug: str) -> dict[str, Any]:
     author_dict = info.context.get("author", {})
     author_id = author_dict.get("id")
     with local_session() as session: