Merge pull request #25 from sab666/feature/community

feat: community create

pythorhead/community.py

@@ -0,0 +1,47 @@
+from typing import Optional
+
+from pythorhead.requestor import Requestor, Request
+
+
+class Community:
+    def __init__(self, _requestor: Requestor):
+        self._requestor = _requestor
+
+    def create(
+            self,
+            name: str,
+            title: str,
+            description: Optional[str] = None,
+            icon: Optional[str] = None,
+            nsfw: Optional[bool] = None,
+            posting_restricted_to_mods: Optional[bool] = None
+    ) -> Optional[dict]:
+        """
+        Create a community
+
+        Args:
+            name (str)
+            title (str)
+            description (str, optional): Defaults to None
+            icon (str, optional): Defaults to None
+            nsfw (bool, optional): Defaults to None
+            posting_restricted_to_mods (bool, optional): Defaults to None
+
+        Returns:
+            Optional[dict]: post data if successful
+        """
+        new_community = {
+            "name": name,
+            "title": title,
+
+        }
+        if description is not None:
+            new_community['description'] = description
+        if icon is not None:
+            new_community['icon'] = icon
+        if nsfw is not None:
+            new_community['nsfw'] = nsfw
+        if [posting_restricted_to_mods] is not None:
+            new_community['[posting_restricted_to_mods]'] = [posting_restricted_to_mods]
+
+        return self._requestor.api(Request.POST, "/post", json=new_community)

pythorhead/lemmy.py

@@ -2,6 +2,7 @@
 from typing import Optional
 
 from pythorhead.comment import Comment
+from pythorhead.community import Community
 from pythorhead.image import Image
 from pythorhead.post import Post
 from pythorhead.private_message import PrivateMessage
@@ -20,6 +21,7 @@ def __init__(self, api_base_url: str) -> None:
         self._requestor = Requestor()
         self._requestor.set_domain(api_base_url)
         self.post = Post(self._requestor)
+        self.community = Community(self._requestor)
         self.comment = Comment(self._requestor)
         self.site = Site(self._requestor)
         self.user = User(self._requestor)