python-substack 0.1.13__tar.gz → 0.1.14__tar.gz

{python_substack-0.1.13 → python_substack-0.1.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-substack
-Version: 0.1.13
+Version: 0.1.14
 Summary: A Python wrapper around the Substack API.
 Home-page: https://github.com/ma2za/python-substack
 License: MIT
@@ -44,15 +44,19 @@ You can install python-substack using:
 
 Set the following environment variables by creating a **.env** file:
 
-    PUBLICATION_URL=https://ma2za.substack.com
     EMAIL=
     PASSWORD=
-    USER_ID=
 
-To discover the USER_ID go to your public profile page,
-in the URL bar of the browser you find the substack address
-followed by your USER_ID and your username:
-https://substack.com/profile/[USER_ID]-[username]
+## If you don't have a password
+
+Recently Substack has been setting up new accounts without a password. If you sign-out and sign back in it just uses
+your email address with a "magic" link.
+
+Set a password:
+
+- Sign-out of Substack
+- At the sign-in page click, "Sign in with password" under the `Email` text box
+- Then choose, "Set a new password"
 
 The .env file will be ignored by git but always be careful.
 
@@ -71,13 +75,24 @@ from substack.post import Post
 api = Api(
     email=os.getenv("EMAIL"),
     password=os.getenv("PASSWORD"),
-    publication_url=os.getenv("PUBLICATION_URL"),
 )
 
+user_id = api.get_user_id()
+
+# Switch Publications - The library defaults to your users primary publication. You can retrieve all your publications and change which one you want to use.
+
+# primary publication
+user_publication = api.get_user_primary_publication()
+# all publications
+user_publications = api.get_user_publications()
+
+# This step is only necessary if you are not using your primary publication
+# api.change_publication(user_publication)
+
 post = Post(
     title="How to publish a Substack post using the Python API",
     subtitle="This post was published using the Python API",
-    user_id=os.getenv("USER_ID")
+    user_id=user_id
 )
 
 post.add({'type': 'paragraph', 'content': 'This is how you add a new paragraph to your post!'})
@@ -91,6 +106,9 @@ post.add({'type': "paragraph",
 post.add({'type': 'paragraph', 'content': [
     {'content': "View Link", 'marks': [{'type': "link", 'href': 'https://whoraised.substack.com/'}]}]})
 
+# set paywall boundary
+post.add({'type': 'paywall'})
+
 # add image
 post.add({'type': 'captionedImage', 'src': "https://media.tenor.com/7B4jMa-a7bsAAAAC/i-am-batman.gif"})
 
@@ -109,4 +127,17 @@ api.prepublish_draft(draft.get("id"))
 api.publish_draft(draft.get("id"))
 ```
 
+# Contributing
+
+Install pre-commit:
+
+```shell
+pip install pre-commit
+```
+
+Set up pre-commit
+
+```shell
+pre-commit install
+```
 

{python_substack-0.1.13 → python_substack-0.1.14}/README.md

@@ -20,15 +20,19 @@ You can install python-substack using:
 
 Set the following environment variables by creating a **.env** file:
 
-    PUBLICATION_URL=https://ma2za.substack.com
     EMAIL=
     PASSWORD=
-    USER_ID=
 
-To discover the USER_ID go to your public profile page,
-in the URL bar of the browser you find the substack address
-followed by your USER_ID and your username:
-https://substack.com/profile/[USER_ID]-[username]
+## If you don't have a password
+
+Recently Substack has been setting up new accounts without a password. If you sign-out and sign back in it just uses
+your email address with a "magic" link.
+
+Set a password:
+
+- Sign-out of Substack
+- At the sign-in page click, "Sign in with password" under the `Email` text box
+- Then choose, "Set a new password"
 
 The .env file will be ignored by git but always be careful.
 
@@ -47,13 +51,24 @@ from substack.post import Post
 api = Api(
     email=os.getenv("EMAIL"),
     password=os.getenv("PASSWORD"),
-    publication_url=os.getenv("PUBLICATION_URL"),
 )
 
+user_id = api.get_user_id()
+
+# Switch Publications - The library defaults to your users primary publication. You can retrieve all your publications and change which one you want to use.
+
+# primary publication
+user_publication = api.get_user_primary_publication()
+# all publications
+user_publications = api.get_user_publications()
+
+# This step is only necessary if you are not using your primary publication
+# api.change_publication(user_publication)
+
 post = Post(
     title="How to publish a Substack post using the Python API",
     subtitle="This post was published using the Python API",
-    user_id=os.getenv("USER_ID")
+    user_id=user_id
 )
 
 post.add({'type': 'paragraph', 'content': 'This is how you add a new paragraph to your post!'})
@@ -67,6 +82,9 @@ post.add({'type': "paragraph",
 post.add({'type': 'paragraph', 'content': [
     {'content': "View Link", 'marks': [{'type': "link", 'href': 'https://whoraised.substack.com/'}]}]})
 
+# set paywall boundary
+post.add({'type': 'paywall'})
+
 # add image
 post.add({'type': 'captionedImage', 'src': "https://media.tenor.com/7B4jMa-a7bsAAAAC/i-am-batman.gif"})
 
@@ -85,3 +103,16 @@ api.prepublish_draft(draft.get("id"))
 api.publish_draft(draft.get("id"))
 ```
 
+# Contributing
+
+Install pre-commit:
+
+```shell
+pip install pre-commit
+```
+
+Set up pre-commit
+
+```shell
+pre-commit install
+```

{python_substack-0.1.13 → python_substack-0.1.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-substack"
-version = "0.1.13"
+version = "0.1.14"
 description = "A Python wrapper around the Substack API."
 authors = ["Paolo Mazza <mazzapaolo2019@gmail.com>"]
 license = "MIT"
@@ -28,4 +28,4 @@ PyYAML = "^6.0"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
-build-backend = "poetry.core.masonry.api"
+build-backend = "poetry.core.masonry.api"

{python_substack-0.1.13 → python_substack-0.1.14}/substack/api.py

@@ -1,4 +1,11 @@
+"""
+
+API Wrapper
+
+"""
+
 import base64
+import json
 import logging
 import os
 from datetime import datetime
@@ -21,12 +28,13 @@ class Api:
     """
 
     def __init__(
-            self,
-            email=None,
-            password=None,
-            base_url=None,
-            publication_url=None,
-            debug=False,
+        self,
+        email=None,
+        password=None,
+        cookies_path=None,
+        base_url=None,
+        publication_url=None,
+        debug=False,
     ):
         """
 
@@ -37,12 +45,15 @@ class Api:
         Args:
           email:
           password:
+          cookies_path
+            To re-use your session without logging in each time, you can save your cookies to a json file and
+            then load them in the next session.
+            Make sure to re-save your cookies, as they do update over time.
           base_url:
             The base URL to use to contact the Substack API.
             Defaults to https://substack.com/api/v1.
         """
         self.base_url = base_url or "https://substack.com/api/v1"
-        self.publication_url = urljoin(publication_url, "api/v1")
 
         if debug:
             logging.basicConfig()
@@ -50,8 +61,42 @@ class Api:
 
         self._session = requests.Session()
 
-        if email is not None and password is not None:
+        # Load cookies from file if provided
+        # Helps with Captcha errors by reusing cookies from "local" auth, then switching to running code in the cloud
+        if cookies_path is not None:
+            with open(cookies_path) as f:
+                cookies = json.load(f)
+            self._session.cookies.update(cookies)
+
+        elif email is not None and password is not None:
             self.login(email, password)
+        else:
+            raise ValueError(
+                "Must provide email and password or cookies_path to authenticate."
+            )
+
+        user_publication = None
+        # if the user provided a publication url, then use that
+        if publication_url:
+            import re
+
+            # Regular expression to extract subdomain name
+            match = re.search(r"https://(.*).substack.com", publication_url.lower())
+            subdomain = match.group(1) if match else None
+
+            user_publications = self.get_user_publications()
+            # search through publications to find the publication with the matching subdomain
+            for publication in user_publications:
+                if publication["subdomain"] == subdomain:
+                    # set the current publication to the users publication
+                    user_publication = publication
+                    break
+        else:
+            # get the users primary publication
+            user_publication = self.get_user_primary_publication()
+
+        # set the current publication to the users primary publication
+        self.change_publication(user_publication)
 
     def login(self, email, password) -> dict:
         """
@@ -73,8 +118,37 @@ class Api:
                 "redirect": "/",
             },
         )
+
         return Api._handle_response(response=response)
 
+    def signin_for_pub(self, publication):
+        """
+        Complete the signin process
+        """
+        response = self._session.get(
+            f"https://substack.com/sign-in?redirect=%2F&for_pub={publication['subdomain']}",
+        )
+        return Api._handle_response(response=response)
+
+    def change_publication(self, publication):
+        """
+        Change the publication URL
+        """
+        self.publication_url = urljoin(publication["publication_url"], "api/v1")
+
+        # sign-in to the publication
+        self.signin_for_pub(publication)
+
+    def export_cookies(self, path: str = "cookies.json"):
+        """
+        Export cookies to a json file.
+        Args:
+            path: path to the json file
+        """
+        cookies = self._session.cookies.get_dict()
+        with open(path, "w") as f:
+            json.dump(cookies, f)
+
     @staticmethod
     def _handle_response(response: requests.Response):
         """
@@ -92,6 +166,82 @@ class Api:
         except ValueError:
             raise SubstackRequestException("Invalid Response: %s" % response.text)
 
+    def get_user_id(self):
+        """
+
+        Returns:
+
+        """
+        profile = self.get_user_profile()
+        user_id = profile["id"]
+
+        return user_id
+
+    @staticmethod
+    def get_publication_url(publication: dict) -> str:
+        """
+        Gets the publication url
+
+        Args:
+            publication:
+        """
+        custom_domain = publication["custom_domain"]
+        if not custom_domain:
+            publication_url = f"https://{publication['subdomain']}.substack.com"
+        else:
+            publication_url = f"https://{custom_domain}"
+
+        return publication_url
+
+    def get_user_primary_publication(self):
+        """
+        Gets the users primary publication
+        """
+
+        profile = self.get_user_profile()
+        primary_publication = profile["primaryPublication"]
+        primary_publication["publication_url"] = self.get_publication_url(
+            primary_publication
+        )
+
+        return primary_publication
+
+    def get_user_publications(self):
+        """
+        Gets the users publications
+        """
+
+        profile = self.get_user_profile()
+
+        # Loop through users "publicationUsers" list, and return a list
+        # of dictionaries of "name", and "subdomain", and "id"
+        user_publications = []
+        for publication in profile["publicationUsers"]:
+            pub = publication["publication"]
+            pub["publication_url"] = self.get_publication_url(pub)
+            user_publications.append(pub)
+
+        return user_publications
+
+    def get_user_profile(self):
+        """
+        Gets the users profile
+        """
+        response = self._session.get(f"{self.base_url}/user/profile/self")
+
+        return Api._handle_response(response=response)
+
+    def get_user_settings(self):
+        """
+        Get list of users.
+
+        Returns:
+
+        """
+        response = self._session.get(f"{self.base_url}/settings")
+
+        return Api._handle_response(response=response)
+
     def get_publication_users(self):
         """
         Get list of users.
@@ -111,9 +261,29 @@ class Api:
         Returns:
 
         """
-        response = self._session.get(f"{self.publication_url}/publication_launch_checklist")
+        response = self._session.get(
+            f"{self.publication_url}/publication_launch_checklist"
+        )
+
+        return Api._handle_response(response=response)["subscriberCount"]
+
+    def get_published_posts(
+        self, offset=0, limit=25, order_by="post_date", order_direction="desc"
+    ):
+        """
+        Get list of published posts for the publication.
+        """
+        response = self._session.get(
+            f"{self.publication_url}/post_management/published",
+            params={
+                "offset": offset,
+                "limit": limit,
+                "order_by": order_by,
+                "order_direction": order_direction,
+            },
+        )
 
-        return Api._handle_response(response=response)['subscriberCount']
+        return Api._handle_response(response=response)
 
     def get_posts(self) -> dict:
         """
@@ -142,6 +312,14 @@ class Api:
         )
         return Api._handle_response(response=response)
 
+    def get_draft(self, draft_id):
+        """
+        Gets a draft given it's id.
+
+        """
+        response = self._session.get(f"{self.publication_url}/drafts/{draft_id}")
+        return Api._handle_response(response=response)
+
     def delete_draft(self, draft_id):
         """
 
@@ -166,11 +344,7 @@ class Api:
         response = self._session.post(f"{self.publication_url}/drafts", json=body)
         return Api._handle_response(response=response)
 
-    def put_draft(
-            self,
-            draft,
-            **kwargs
-    ) -> dict:
+    def put_draft(self, draft, **kwargs) -> dict:
         """
 
         Args:
@@ -202,7 +376,7 @@ class Api:
         return Api._handle_response(response=response)
 
     def publish_draft(
-            self, draft, send: bool = True, share_automatically: bool = False
+        self, draft, send: bool = True, share_automatically: bool = False
     ) -> dict:
         """
 
@@ -320,7 +494,7 @@ class Api:
                 page_output = self.get_category(category_id, category_type, page)
                 publications.extend(page_output.get("publications", []))
                 if (
-                        limit is not None and limit <= len(publications)
+                    limit is not None and limit <= len(publications)
                 ) or not page_output.get("more", False):
                     publications = publications[:limit]
                     break
@@ -358,5 +532,9 @@ class Api:
             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]
+        sections = [
+            p.get("sections")
+            for p in content.get("publications")
+            if p.get("hostname") in self.publication_url
+        ]
         return sections[0]

{python_substack-0.1.13 → python_substack-0.1.14}/substack/post.py

@@ -1,3 +1,9 @@
+"""
+
+Post Utilities
+
+"""
+
 import json
 from typing import Dict
 
@@ -7,13 +13,19 @@ from substack.exceptions import SectionNotExistsException
 
 
 class Post:
+    """
+
+    Post utility class
+
+    """
+
     def __init__(
-            self,
-            title: str,
-            subtitle: str,
-            user_id,
-            audience: str = None,
-            write_comment_permissions: str = None,
+        self,
+        title: str,
+        subtitle: str,
+        user_id,
+        audience: str = None,
+        write_comment_permissions: str = None,
     ):
         """
 
@@ -143,20 +155,20 @@ class Post:
         return self
 
     def captioned_image(
-            self,
-            src: str,
-            fullscreen: bool = False,
-            imageSize: str = "normal",
-            height: int = 819,
-            width: int = 1456,
-            resizeWidth: int = 728,
-            bytes: str = None,
-            alt: str = None,
-            title: str = None,
-            type: str = None,
-            href: str = None,
-            belowTheFold: bool = False,
-            internalRedirect: str = None,
+        self,
+        src: str,
+        fullscreen: bool = False,
+        imageSize: str = "normal",
+        height: int = 819,
+        width: int = 1456,
+        resizeWidth: int = 728,
+        bytes: str = None,
+        alt: str = None,
+        title: str = None,
+        type: str = None,
+        href: str = None,
+        belowTheFold: bool = False,
+        internalRedirect: str = None,
     ):
         """
 
@@ -252,7 +264,7 @@ class Post:
         return self
 
     def remove_last_paragraph(self):
-        """ """
+        """Remove last paragraph"""
         del self.draft_body.get("content")[-1]
 
     def get_draft(self):
@@ -282,7 +294,11 @@ class Post:
             Subscribe for free to receive new posts and support my work."""
 
         subscribe = self.draft_body["content"][-1]
-        subscribe["attrs"] = {"url": "%%checkout_url%%", "text": "Subscribe", "language": "en"}
+        subscribe["attrs"] = {
+            "url": "%%checkout_url%%",
+            "text": "Subscribe",
+            "language": "en",
+        }
         subscribe["content"] = [
             {
                 "type": "ctaCaption",