🦍 set post section

Author: ma2za

README.md

@@ -58,17 +58,28 @@ post = Post(
 
 post.add({'type': 'paragraph', 'content': 'This is how you add a new paragraph to your post!'})
 
-#bolden text
-post.add({'type': "paragraph",'content':[{'content': "This is how you "},{'content': "bolden ",'marks':[{'type': "strong"}]},{'content': "a word."}]})
+# bolden text
+post.add({'type': "paragraph",
+          'content': [{'content': "This is how you "}, {'content': "bolden ", 'marks': [{'type': "strong"}]},
+                      {'content': "a word."}]})
 
-#add hyperlink to text
-post.add({'type':'paragraph','content':[{'content': "View Link",'marks':[{'type': "link",'href':'https://whoraised.substack.com/'}]}]})
+# add hyperlink to text
+post.add({'type': 'paragraph', 'content': [
+    {'content': "View Link", 'marks': [{'type': "link", 'href': 'https://whoraised.substack.com/'}]}]})
 
-#add image
-post.add({'type':'captionedImage','src': "https://media.tenor.com/7B4jMa-a7bsAAAAC/i-am-batman.gif"})
+# add image
+post.add({'type': 'captionedImage', 'src': "https://media.tenor.com/7B4jMa-a7bsAAAAC/i-am-batman.gif"})
+
+# add local image
+image = api.get_image('image.png')
+post.add({"type": "captionedImage", "src": image.get("url")})
 
 draft = api.post_draft(post.get_draft())
 
+# set section (THIS CAN BE DONE ONLY AFTER HAVING FIRST POSTED THE DRAFT)
+post.set_section("rick rolling", api.get_sections())
+api.put_draft(draft.get("id"), draft_section_id=post.draft_section_id)
+
 api.prepublish_draft(draft.get("id"))
 
 api.publish_draft(draft.get("id"))

examples/draft.yaml

@@ -6,6 +6,8 @@ audience:
   "everyone" # everyone, only_paid, founding, only_free
 write_comment_permissions:
   "none" # none, only_paid, everyone
+section:
+  "rick"
 body:
   0:
     type: "heading"

examples/publish_post.py

@@ -20,7 +20,7 @@
         type=str,
     )
     parser.add_argument(
-        "--publish", help="Publish the draft.", action="store_true", default=False
+        "--publish", help="Publish the draft.", action="store_true", default=True
     )
     args = parser.parse_args()
 
@@ -53,6 +53,9 @@
 
     draft = api.post_draft(post.get_draft())
 
+    post.set_section(post_data.get("section"), api.get_sections())
+    api.put_draft(draft.get("id"), draft_section_id=post.draft_section_id)
+
     if args.publish:
         api.prepublish_draft(draft.get("id"))
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-substack"
-version = "0.1.12"
+version = "0.1.13"
 description = "A Python wrapper around the Substack API."
 authors = ["Paolo Mazza <mazzapaolo2019@gmail.com>"]
 license = "MIT"

substack/api.py

@@ -169,32 +169,20 @@ def post_draft(self, body) -> dict:
     def put_draft(
             self,
             draft,
-            title=None,
-            subtitle=None,
-            body=None,
-            cover_image=None,
+            **kwargs
     ) -> dict:
         """
 
         Args:
-            draft: draft id
-            title:
-            subtitle:
-            body:
-            cover_image:
+            draft:
+            **kwargs:
 
         Returns:
 
         """
-
         response = self._session.put(
             f"{self.publication_url}/drafts/{draft}",
-            json={
-                "draft_title": title,
-                "draft_subtitle": subtitle,
-                "draft_body": body,
-                "cover_image": cover_image,
-            },
+            json=kwargs,
         )
         return Api._handle_response(response=response)
 
@@ -357,3 +345,18 @@ def delete_all_drafts(self):
             for draft in drafts:
                 response = self.delete_draft(draft.get("id"))
         return response
+
+    def get_sections(self):
+        """
+        Get a list of the sections of your publication.
+
+        TODO: this is hacky but I cannot find another place where to get the sections.
+        Returns:
+
+        """
+        response = self._session.get(
+            f"{self.publication_url}/subscriptions",
+        )
+        content = Api._handle_response(response=response)
+        sections = [p.get("sections") for p in content.get("publications") if p.get("hostname") in self.publication_url]
+        return sections[0]

substack/exceptions.py

@@ -26,3 +26,7 @@ def __init__(self, message):
 
     def __str__(self):
         return f"SubstackRequestException: {self.message}"
+
+
+class SectionNotExistsException(SubstackRequestException):
+    pass

substack/post.py

@@ -3,6 +3,8 @@
 
 __all__ = ["Post"]
 
+from substack.exceptions import SectionNotExistsException
+
 
 class Post:
     def __init__(
@@ -27,13 +29,31 @@ def __init__(
         self.draft_body = {"type": "doc", "content": []}
         self.draft_bylines = [{"id": int(user_id), "is_guest": False}]
         self.audience = audience if audience is not None else "everyone"
+        self.draft_section_id = None
+        self.section_chosen = True
 
         # TODO better understand the possible values and combinations with audience
         if write_comment_permissions is not None:
             self.write_comment_permissions = write_comment_permissions
         else:
             self.write_comment_permissions = self.audience
 
+    def set_section(self, name: str, sections: list):
+        """
+
+        Args:
+            name:
+            sections:
+
+        Returns:
+
+        """
+        section = [s for s in sections if s.get("name") == name]
+        if len(section) != 1:
+            raise SectionNotExistsException(name)
+        section = section[0]
+        self.draft_section_id = section.get("id")
+
     def add(self, item: Dict):
         """