cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-04-28 16:17:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Default skip telegraph account

Browse Source 
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>
This commit is contained in:
Alter-xyz 2024-06-24 21:26:44 -04:00
parent e2ffd7fa57
commit 9c8638279e
5 changed files with 141 additions and 219 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

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

14
README.md
View File

@ -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`

227
handlers/__init__.py
View File

@ -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
from bs4 import BeautifulSoup # pip install beautifulsoup4
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
import requests
import os
from bs4 import BeautifulSoup
import markdown
class TelegraphAPI:
def __init__(self, access_token):
self.access_token = access_token
def __init__(
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):
"""
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
url = f'{self.base_url}/getAccountInfo?access_token={self.access_token}&fields=["short_name","author_name","author_url","auth_url"]'
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,
):
"""
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)
content = self._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):
"""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
def _md_to_dom(self, markdown_text):
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",
]

13
handlers/cohere.py
View File

@ -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)

18
handlers/useful.py
View File

@ -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:
ph = TelegraphAPI(TELEGRA_PH_TOKEN)
else:
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"
#### Telegraph init ####
TELEGRA_PH_TOKEN = environ.get("TELEGRA_PH_TOKEN")
ph = TelegraphAPI(TELEGRA_PH_TOKEN)
#### Telegraph done ####
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)