feat: Default skip telegraph account

It will print out the token in local terminal by default.

If Store_Token = True, it will store token in "token_key.json"

View my lite ver and try at <https://github.com/alterxyz/tg_bot_collections_lite>

.gitignore

@@ -169,4 +169,4 @@ nohup.out
 *.pdf
 .pdm-python
 *.wav
-telegraph_token.json
+token_key.json

README.md

@@ -81,7 +81,19 @@ Note, currently its support dify Chatbot with instructions(System prompt) and di
 2. export COHERE_API_KEY=${the_key}
 3. use `cohere: ${message}` to ask
 
-## Bot -> `Telegra.ph`
+## Function -> Telegraph
+
+### Skip token (default)
+
+You do not need to do anything.
+
+But you may not be able to edit any generated post since you do not have the token.
+
+### Store token (recommended)
+
+Change "Store_Token" to "True" in "handlers/__init__.py" TelegraphAPI/_create_ph_account. It will store the token in "token_key.json".
+
+### Get token manually from Telegram account
 
 1. https://t.me/telegraph Create or login Telegraph account
 2. `Log in as ${Account} on this device`

handlers/__init__.py

@@ -200,66 +200,26 @@ def image_to_data_uri(file_path):
         return f"data:image/png;base64,{encoded_image}"
 
 
-import requests
 import json
-import markdown  # pip install Markdown
+import requests
-from bs4 import BeautifulSoup  # pip install beautifulsoup4
+import os
-
+from bs4 import BeautifulSoup
-
+import markdown
-def create_ph_account(short_name: str, author_name: str, author_url: str = None) -> str:
-    """
-        Creates a new account on the Telegra.ph platform.
-        If an account already exists (stored in a local JSON file), returns the existing access token.
-        Otherwise, creates a new account and stores the information locally.
-        Sample request
-    https://api.telegra.ph/editAccountInfo?access_token=d3b25feccb89e508a9114afb82aa421fe2a9712b963b387cc5ad71e58722&short_name=Sandbox&author_name=Anonymous
-
-        Args:
-            short_name (str): The short name of the account.
-            author_name (str): The name of the author.
-            author_url (str, optional): The URL of the author's profile. Defaults to None.
-
-        Returns:
-            str: The access token for the account.
-
-        Raises:
-            requests.RequestException: If the API request fails.
-            json.JSONDecodeError: If the API response is not valid JSON.
-            KeyError: If the API response does not contain the expected data.
-    """
-    TELEGRAPH_API_URL = "https://api.telegra.ph/createAccount"
-
-    # Try to load existing account information
-    try:
-        with open("telegraph_token.json", "r") as f:
-            account = json.load(f)
-        return account["result"]["access_token"]
-    except FileNotFoundError:
-        # If no existing account, create a new one
-        data = {
-            "short_name": short_name,
-            "author_name": author_name,
-        }
-        if author_url:
-            data["author_url"] = author_url
-
-        # Make API request
-        response = requests.post(TELEGRAPH_API_URL, data=data)
-        response.raise_for_status()  # Raises an HTTPError for bad responses
-
-        account = response.json()
-        access_token = account["result"]["access_token"]
-
-        # Store the new account information
-        with open("telegraph_token.json", "w") as f:
-            json.dump(account, f)
-
-        return access_token
 
 
 class TelegraphAPI:
-    def __init__(self, access_token):
+    def __init__(
-        self.access_token = access_token
+        self,
+        access_token=None,
+        short_name="tg_bot_collections",
+        author_name="Telegram Bot Collections",
+        author_url=None,
+    ):
+        self.access_token = (
+            access_token
+            if access_token
+            else self._create_ph_account(short_name, author_name, author_url)
+        )
         self.base_url = "https://api.telegra.ph"
 
         # Get account info on initialization
@@ -268,34 +228,58 @@ class TelegraphAPI:
         self.author_name = account_info.get("author_name")
         self.author_url = account_info.get("author_url")
 
+    def _create_ph_account(self, short_name, author_name, author_url):
+        Store_Token = False
+        TELEGRAPH_API_URL = "https://api.telegra.ph/createAccount"
+        TOKEN_FILE = "token_key.json"
+
+        # Try to load existing token information
+        try:
+            with open(TOKEN_FILE, "r") as f:
+                tokens = json.load(f)
+            if "TELEGRA_PH_TOKEN" in tokens and tokens["TELEGRA_PH_TOKEN"] != "example":
+                return tokens["TELEGRA_PH_TOKEN"]
+        except FileNotFoundError:
+            tokens = {}
+
+        # If no existing valid token in TOKEN_FILE, create a new account
+        data = {
+            "short_name": short_name,
+            "author_name": author_name,
+            "author_url": author_url,
+        }
+
+        # Make API request
+        response = requests.post(TELEGRAPH_API_URL, data=data)
+        response.raise_for_status()
+
+        account = response.json()
+        access_token = account["result"]["access_token"]
+
+        # Update the token in the dictionary
+        tokens["TELEGRA_PH_TOKEN"] = access_token
+
+        # Store the updated tokens
+        if Store_Token:
+            with open(TOKEN_FILE, "w") as f:
+                json.dump(tokens, f, indent=4)
+        else:
+            print(f"Token not stored to file, but here is your token:\n{access_token}")
+
+        # Store it to the environment variable
+        os.environ["TELEGRA_PH_TOKEN"] = access_token
+
+        return access_token
+
     def create_page(
         self, title, content, author_name=None, author_url=None, return_content=False
     ):
-        """
-        Creates a new Telegraph page.
-
-        Args:
-            title (str): Page title (1-256 characters).
-            content (list): Content of the page as a list of Node dictionaries.
-            author_name (str, optional): Author name (0-128 characters). Defaults to account's author_name.
-            author_url (str, optional): Profile link (0-512 characters). Defaults to account's author_url.
-            return_content (bool, optional): If True, return the content field in the response.
-
-        Returns:
-            str: URL of the created page.
-
-        Raises:
-            requests.exceptions.RequestException: If the request fails.
-
-
-        """
         url = f"{self.base_url}/createPage"
         data = {
             "access_token": self.access_token,
             "title": title,
             "content": json.dumps(content),
             "return_content": return_content,
-            # Use provided author info or fall back to account info
             "author_name": author_name if author_name else self.author_name,
             "author_url": author_url if author_url else self.author_url,
         }
@@ -307,14 +291,7 @@ class TelegraphAPI:
         return page_url
 
     def get_account_info(self):
-        """
+        url = f'{self.base_url}/getAccountInfo?access_token={self.access_token}&fields=["short_name","author_name","author_url","auth_url"]'
-        Gets information about the Telegraph account.
-
-        Returns:
-            dict: Account information including short_name, author_name, and author_url.
-                 Returns None if there's an error.
-        """
-        url = f"{self.base_url}/getAccountInfo?access_token={self.access_token}"  # &fields=[\"author_name\",\"author_url\"] for specific fields
         response = requests.get(url)
 
         if response.status_code == 200:
@@ -332,23 +309,6 @@ class TelegraphAPI:
         author_url=None,
         return_content=False,
     ):
-        """
-        Edits an existing Telegraph page.
-
-        Args:
-            path (str): Path of the page to edit.
-            title (str): New page title (1-256 characters).
-            content (list): New content of the page as a list of Node dictionaries.
-            author_name (str, optional): Author name (0-128 characters). Defaults to account's author_name.
-            author_url (str, optional): Profile link (0-512 characters). Defaults to account's author_url.
-            return_content (bool, optional): If True, return the content field in the response.
-
-        Returns:
-            str: URL of the edited page.
-
-        Raises:
-            requests.exceptions.RequestException: If the request fails.
-        """
         url = f"{self.base_url}/editPage"
         data = {
             "access_token": self.access_token,
@@ -356,7 +316,6 @@ class TelegraphAPI:
             "title": title,
             "content": json.dumps(content),
             "return_content": return_content,
-            # Use provided author info or fall back to account info
             "author_name": author_name if author_name else self.author_name,
             "author_url": author_url if author_url else self.author_url,
         }
@@ -369,15 +328,6 @@ class TelegraphAPI:
         return page_url
 
     def get_page(self, path):
-        """
-        Gets information about a Telegraph page.
-
-        Args:
-            path (str): Path of the page to get.
-
-        Returns:
-            dict: Information about the page.
-        """
         url = f"{self.base_url}/getPage/{path}?return_content=true"
         response = requests.get(url)
         response.raise_for_status()
@@ -391,23 +341,7 @@ class TelegraphAPI:
         author_url=None,
         return_content=False,
     ):
-        """
+        content = self._md_to_dom(markdown_text)
-        Creates a new Telegraph page from markdown text.
-
-        Args:
-            title (str): Page title (1-256 characters).
-            markdown_text (str): Markdown text to convert to HTML.
-            author_name (str, optional): Author name (0-128 characters). Defaults to account's author_name.
-            author_url (str, optional): Profile link (0-512 characters). Defaults to account's author_url.
-            return_content (bool, optional): If True, return the content field in the response.
-
-        Returns:
-            str: URL of the created page.
-
-        Raises:
-            requests.exceptions.RequestException: If the request fails.
-        """
-        content = md_to_dom(markdown_text)
         return self.create_page(title, content, author_name, author_url, return_content)
 
     def edit_page_md(
@@ -419,34 +353,23 @@ class TelegraphAPI:
         author_url=None,
         return_content=False,
     ):
-        content = md_to_dom(markdown_text)
+        content = self._md_to_dom(markdown_text)
         return self.edit_page(
             path, title, content, author_name, author_url, return_content
         )
 
+    def authorize_browser(self):
+        url = f'{self.base_url}/getAccountInfo?access_token={self.access_token}&fields=["auth_url"]'
+        response = requests.get(url)
+        response.raise_for_status()
+        return response.json()["result"]["auth_url"]
 
-def md_to_dom(markdown_text):
+    def _md_to_dom(self, markdown_text):
-    """Converts markdown text to a Python dictionary representing the DOM,
-    limiting heading levels to h3 and h4.
-
-    Args:
-        markdown_text: The markdown text to convert.
-
-    Returns:
-        A Python list representing the DOM, where each element is a dictionary
-        with the following keys:
-            - 'tag': The tag name of the element.
-            - 'attributes': A dictionary of attributes for the element (optional).
-            - 'children': A list of child elements (optional).
-    """
-
-    # Convert markdown to HTML
         html = markdown.markdown(
             markdown_text,
             extensions=["markdown.extensions.extra", "markdown.extensions.sane_lists"],
         )
 
-    # Parse the HTML with BeautifulSoup
         soup = BeautifulSoup(html, "html.parser")
 
         def parse_element(element):
@@ -458,18 +381,18 @@ def md_to_dom(markdown_text):
                     tag_dict["tag"] = "h4"
                 else:
                     tag_dict["tag"] = "p"
-                tag_dict["children"] = [{"tag": "strong", "children": element.contents}]
+                    tag_dict["children"] = [
+                        {"tag": "strong", "children": element.contents}
+                    ]
 
-            # Correctly handle children for h1-h6
                 if element.attrs:
                     tag_dict["attributes"] = element.attrs
                 if element.contents:
                     children = []
                     for child in element.contents:
                         if isinstance(child, str):
-                        # Remove leading/trailing whitespace from text nodes
                             children.append(child.strip())
-                    else:  # it's another tag
+                        else:
                             children.append(parse_element(child))
                     tag_dict["children"] = children
             else:
@@ -479,9 +402,8 @@ def md_to_dom(markdown_text):
                     children = []
                     for child in element.contents:
                         if isinstance(child, str):
-                        # Remove leading/trailing whitespace from text nodes
                             children.append(child.strip())
-                    else:  # it's another tag
+                        else:
                             children.append(parse_element(child))
                     if children:
                         tag_dict["children"] = children
@@ -490,10 +412,8 @@ def md_to_dom(markdown_text):
         new_dom = []
         for element in soup.contents:
             if isinstance(element, str) and not element.strip():
-            # Skip empty text nodes
                 continue
             elif isinstance(element, str):
-            # Treat remaining text nodes as separate elements for clarity
                 new_dom.append({"tag": "text", "content": element.strip()})
             else:
                 new_dom.append(parse_element(element))
@@ -508,5 +428,4 @@ __all__ = [
     "enrich_text_with_urls",
     "image_to_data_uri",
     "TelegraphAPI",
-    "create_ph_account",
 ]

handlers/cohere.py

@@ -17,19 +17,12 @@ markdown_symbol.link = "🔗"  # If you want, Customizing the link symbol
 
 COHERE_API_KEY = environ.get("COHERE_API_KEY")
 COHERE_MODEL = "command-r-plus"
-
-TELEGRA_PH_TOKEN = environ.get("TELEGRA_PH_TOKEN")
-if TELEGRA_PH_TOKEN:
-    ph = TelegraphAPI(TELEGRA_PH_TOKEN)
-else:
-    TELEGRA_PH_TOKEN = create_ph_account(
-        short_name="cohere", author_name="A Telegram Bot"
-    )
-    ph = TelegraphAPI(TELEGRA_PH_TOKEN)
-
 if COHERE_API_KEY:
     co = cohere.Client(api_key=COHERE_API_KEY)
 
+TELEGRA_PH_TOKEN = environ.get("TELEGRA_PH_TOKEN")
+ph = TelegraphAPI(TELEGRA_PH_TOKEN)
+
 # Global history cache
 cohere_player_dict = ExpiringDict(max_len=1000, max_age_seconds=300)
 

handlers/useful.py

@@ -19,20 +19,18 @@ from . import *
 
 from telegramify_markdown.customize import markdown_symbol
 
+#### Cohere init ####
 import cohere
 
 COHERE_API_KEY = environ.get("COHERE_API_KEY")
-TELEGRA_PH_TOKEN = environ.get("TELEGRA_PH_TOKEN")
+COHERE_MODEL = "command-r-plus"
 if COHERE_API_KEY:
     co = cohere.Client(api_key=COHERE_API_KEY)
-if TELEGRA_PH_TOKEN:
+
+#### Telegraph init ####
+TELEGRA_PH_TOKEN = environ.get("TELEGRA_PH_TOKEN")
 ph = TelegraphAPI(TELEGRA_PH_TOKEN)
-else:
+#### Telegraph done ####
-    TELEGRA_PH_TOKEN = create_ph_account(
-        short_name="Answer it", author_name="A Telegram Bot"
-    )
-    ph = TelegraphAPI(TELEGRA_PH_TOKEN)
-COHERE_MODEL = "command-r-plus"
 
 chat_message_dict = ExpiringDict(max_len=100, max_age_seconds=120)
 chat_user_dict = ExpiringDict(max_len=100, max_age_seconds=20)
@@ -203,7 +201,7 @@ def answer_it_handler(message: Message, bot: TeleBot):
     else:
         pass
 
-    ##### Answer #####
+    ##### Telegraph #####
     final_answer(latest_message, bot, full, chat_id_list)