commynity-cudl

resolvers/community.py

@@ -1,7 +1,7 @@
 from orm.author import Author
 from orm.community import Community, CommunityFollower
 from services.db import local_session
-from services.schema import query
+from services.schema import mutation, query
 
 
 @query.field("get_communities_all")
@@ -29,3 +29,69 @@ async def get_communities_by_author(_, _info, slug="", user="", author_id=0):
             q = q.where(CommunityFollower.author == author_id)
         return q.all()
     return []
+
+
+@mutation.field("join_community")
+async def join_community(_, info, slug: str):
+    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.commit()
+        return {"ok": True}
+
+
+@mutation.field("leave_community")
+async def leave_community(_, info, slug: str):
+    author_dict = info.context.get("author", {})
+    author_id = author_dict.get("id")
+    with local_session() as session:
+        session.query(CommunityFollower).where(
+            CommunityFollower.author == author_id, CommunityFollower.community == slug
+        ).delete()
+        session.commit()
+        return {"ok": True}
+
+
+@mutation.field("create_community")
+async def create_community(_, info, community_data):
+    author_dict = info.context.get("author", {})
+    author_id = author_dict.get("id")
+    with local_session() as session:
+        session.add(Community(author=author_id, **community_data))
+        session.commit()
+        return {"ok": True}
+
+
+@mutation.field("update_community")
+async def update_community(_, info, community_data):
+    author_dict = info.context.get("author", {})
+    author_id = author_dict.get("id")
+    slug = community_data.get("slug")
+    if slug:
+        with local_session() as session:
+            try:
+                session.query(Community).where(Community.created_by == author_id, Community.slug == slug).update(
+                    community_data
+                )
+                session.commit()
+            except Exception as e:
+                return {"ok": False, "error": str(e)}
+            return {"ok": True}
+    return {"ok": False, "error": "Please, set community slug in input"}
+
+
+@mutation.field("delete_community")
+async def delete_community(_, info, slug: str):
+    author_dict = info.context.get("author", {})
+    author_id = author_dict.get("id")
+    with local_session() as session:
+        try:
+            session.query(Community).where(Community.slug == slug, Community.created_by == author_id).delete()
+            session.commit()
+            return {"ok": True}
+        except Exception as e:
+            return {"ok": False, "error": str(e)}