492 lines
13 KiB
Python
492 lines
13 KiB
Python
# Pyrogram - Telegram MTProto API Client Library for Python
|
|
# Copyright (C) 2017-2019 Dan Tès <https://github.com/delivrance>
|
|
#
|
|
# This file is part of Pyrogram.
|
|
#
|
|
# Pyrogram is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Pyrogram is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import ast
|
|
import os
|
|
import re
|
|
import shutil
|
|
|
|
HOME = "compiler/docs"
|
|
DESTINATION = "docs/source/telegram"
|
|
PYROGRAM_API_DEST = "docs/source/api"
|
|
|
|
FUNCTIONS_PATH = "pyrogram/api/functions"
|
|
TYPES_PATH = "pyrogram/api/types"
|
|
|
|
FUNCTIONS_BASE = "functions"
|
|
TYPES_BASE = "types"
|
|
|
|
|
|
def snek(s: str):
|
|
s = re.sub(r"(.)([A-Z][a-z]+)", r"\1_\2", s)
|
|
return re.sub(r"([a-z0-9])([A-Z])", r"\1_\2", s).lower()
|
|
|
|
|
|
def generate(source_path, base):
|
|
all_entities = {}
|
|
|
|
def build(path, level=0):
|
|
last = path.split("/")[-1]
|
|
|
|
for i in os.listdir(path):
|
|
try:
|
|
if not i.startswith("__"):
|
|
build("/".join([path, i]), level=level + 1)
|
|
except NotADirectoryError:
|
|
with open(path + "/" + i, encoding="utf-8") as f:
|
|
p = ast.parse(f.read())
|
|
|
|
for node in ast.walk(p):
|
|
if isinstance(node, ast.ClassDef):
|
|
name = node.name
|
|
break
|
|
else:
|
|
continue
|
|
|
|
full_path = os.path.basename(path) + "/" + snek(name).replace("_", "-") + ".rst"
|
|
|
|
if level:
|
|
full_path = base + "/" + full_path
|
|
|
|
os.makedirs(os.path.dirname(DESTINATION + "/" + full_path), exist_ok=True)
|
|
|
|
with open(DESTINATION + "/" + full_path, "w", encoding="utf-8") as f:
|
|
f.write(
|
|
page_template.format(
|
|
title=name,
|
|
title_markup="=" * len(name),
|
|
full_class_path="pyrogram.api.{}".format(
|
|
".".join(full_path.split("/")[:-1]) + "." + name
|
|
)
|
|
)
|
|
)
|
|
|
|
if last not in all_entities:
|
|
all_entities[last] = []
|
|
|
|
all_entities[last].append(name)
|
|
|
|
build(source_path)
|
|
|
|
for k, v in sorted(all_entities.items()):
|
|
v = sorted(v)
|
|
entities = []
|
|
|
|
for i in v:
|
|
entities.append(snek(i).replace("_", "-"))
|
|
|
|
if k != base:
|
|
inner_path = base + "/" + k + "/index" + ".rst"
|
|
module = "pyrogram.api.{}.{}".format(base, k)
|
|
else:
|
|
for i in sorted(list(all_entities), reverse=True):
|
|
if i != base:
|
|
entities.insert(0, "{0}/index".format(i))
|
|
|
|
inner_path = base + "/index" + ".rst"
|
|
module = "pyrogram.api.{}".format(base)
|
|
|
|
with open(DESTINATION + "/" + inner_path, "w", encoding="utf-8") as f:
|
|
if k == base:
|
|
f.write(":tocdepth: 1\n\n")
|
|
k = "Raw " + k
|
|
|
|
f.write(
|
|
toctree.format(
|
|
title=k.title(),
|
|
title_markup="=" * len(k),
|
|
module=module,
|
|
entities="\n ".join(entities)
|
|
)
|
|
)
|
|
|
|
f.write("\n")
|
|
|
|
|
|
def pyrogram_api():
|
|
def get_title_list(s: str) -> list:
|
|
return [i.strip() for i in [j.strip() for j in s.split("\n") if j] if i]
|
|
|
|
# Methods
|
|
|
|
categories = dict(
|
|
utilities="""
|
|
Utilities
|
|
start
|
|
stop
|
|
restart
|
|
idle
|
|
run
|
|
add_handler
|
|
remove_handler
|
|
stop_transmission
|
|
export_session_string
|
|
""",
|
|
messages="""
|
|
Messages
|
|
send_message
|
|
forward_messages
|
|
send_photo
|
|
send_audio
|
|
send_document
|
|
send_sticker
|
|
send_animated_sticker
|
|
send_video
|
|
send_animation
|
|
send_voice
|
|
send_video_note
|
|
send_media_group
|
|
send_location
|
|
send_venue
|
|
send_contact
|
|
send_cached_media
|
|
edit_message_text
|
|
edit_message_caption
|
|
edit_message_media
|
|
edit_message_reply_markup
|
|
edit_inline_text
|
|
edit_inline_caption
|
|
edit_inline_media
|
|
edit_inline_reply_markup
|
|
send_chat_action
|
|
delete_messages
|
|
get_messages
|
|
get_history
|
|
get_history_count
|
|
read_history
|
|
iter_history
|
|
send_poll
|
|
vote_poll
|
|
stop_poll
|
|
retract_vote
|
|
download_media
|
|
""",
|
|
chats="""
|
|
Chats
|
|
join_chat
|
|
leave_chat
|
|
kick_chat_member
|
|
unban_chat_member
|
|
restrict_chat_member
|
|
promote_chat_member
|
|
export_chat_invite_link
|
|
set_chat_photo
|
|
delete_chat_photo
|
|
set_chat_title
|
|
set_chat_description
|
|
pin_chat_message
|
|
unpin_chat_message
|
|
get_chat
|
|
get_chat_member
|
|
get_chat_members
|
|
get_chat_members_count
|
|
iter_chat_members
|
|
get_dialogs
|
|
iter_dialogs
|
|
get_dialogs_count
|
|
restrict_chat
|
|
update_chat_username
|
|
archive_chats
|
|
unarchive_chats
|
|
""",
|
|
users="""
|
|
Users
|
|
get_me
|
|
get_users
|
|
get_profile_photos
|
|
get_profile_photos_count
|
|
iter_profile_photos
|
|
set_profile_photo
|
|
delete_profile_photos
|
|
update_username
|
|
get_user_dc
|
|
block_user
|
|
unblock_user
|
|
""",
|
|
contacts="""
|
|
Contacts
|
|
add_contacts
|
|
get_contacts
|
|
get_contacts_count
|
|
delete_contacts
|
|
""",
|
|
password="""
|
|
Pssword
|
|
enable_cloud_password
|
|
change_cloud_password
|
|
remove_cloud_password
|
|
""",
|
|
bots="""
|
|
Bots
|
|
get_inline_bot_results
|
|
send_inline_bot_result
|
|
answer_callback_query
|
|
answer_inline_query
|
|
request_callback_answer
|
|
send_game
|
|
set_game_score
|
|
get_game_high_scores
|
|
""",
|
|
advanced="""
|
|
Advanced
|
|
send
|
|
resolve_peer
|
|
save_file
|
|
"""
|
|
)
|
|
|
|
root = PYROGRAM_API_DEST + "/methods"
|
|
|
|
shutil.rmtree(root, ignore_errors=True)
|
|
os.mkdir(root)
|
|
|
|
with open(HOME + "/template/methods.rst") as f:
|
|
template = f.read()
|
|
|
|
with open(root + "/index.rst", "w") as f:
|
|
fmt_keys = {}
|
|
|
|
for k, v in categories.items():
|
|
name, *methods = get_title_list(v)
|
|
fmt_keys.update({k: "\n ".join("{0} <{0}>".format(m) for m in methods)})
|
|
|
|
for method in methods:
|
|
with open(root + "/{}.rst".format(method), "w") as f2:
|
|
title = "{}()".format(method)
|
|
|
|
f2.write(title + "\n" + "=" * len(title) + "\n\n")
|
|
f2.write(".. automethod:: pyrogram.Client.{}()".format(method))
|
|
|
|
f.write(template.format(**fmt_keys))
|
|
|
|
# Types
|
|
|
|
categories = dict(
|
|
users_chats="""
|
|
Users & Chats
|
|
User
|
|
UserStatus
|
|
Chat
|
|
ChatPreview
|
|
ChatPhoto
|
|
ChatMember
|
|
ChatPermissions
|
|
Dialog
|
|
""",
|
|
messages_media="""
|
|
Messages & Media
|
|
Message
|
|
MessageEntity
|
|
Photo
|
|
Thumbnail
|
|
Audio
|
|
Document
|
|
Animation
|
|
Video
|
|
Voice
|
|
VideoNote
|
|
Contact
|
|
Location
|
|
Venue
|
|
Sticker
|
|
Game
|
|
WebPage
|
|
Poll
|
|
PollOption
|
|
""",
|
|
bots_keyboard="""
|
|
Bots & Keyboards
|
|
ReplyKeyboardMarkup
|
|
KeyboardButton
|
|
ReplyKeyboardRemove
|
|
InlineKeyboardMarkup
|
|
InlineKeyboardButton
|
|
ForceReply
|
|
CallbackQuery
|
|
GameHighScore
|
|
CallbackGame
|
|
""",
|
|
input_media="""
|
|
Input Media
|
|
InputMedia
|
|
InputMediaPhoto
|
|
InputMediaVideo
|
|
InputMediaAudio
|
|
InputMediaAnimation
|
|
InputMediaDocument
|
|
InputPhoneContact
|
|
""",
|
|
inline_mode="""
|
|
Inline Mode
|
|
InlineQuery
|
|
InlineQueryResult
|
|
InlineQueryResultArticle
|
|
""",
|
|
input_message_content="""
|
|
InputMessageContent
|
|
InputMessageContent
|
|
InputTextMessageContent
|
|
"""
|
|
)
|
|
|
|
root = PYROGRAM_API_DEST + "/types"
|
|
|
|
shutil.rmtree(root, ignore_errors=True)
|
|
os.mkdir(root)
|
|
|
|
with open(HOME + "/template/types.rst") as f:
|
|
template = f.read()
|
|
|
|
with open(root + "/index.rst", "w") as f:
|
|
fmt_keys = {}
|
|
|
|
for k, v in categories.items():
|
|
name, *types = get_title_list(v)
|
|
|
|
fmt_keys.update({k: "\n ".join(types)})
|
|
|
|
# noinspection PyShadowingBuiltins
|
|
for type in types:
|
|
with open(root + "/{}.rst".format(type), "w") as f2:
|
|
title = "{}".format(type)
|
|
|
|
f2.write(title + "\n" + "=" * len(title) + "\n\n")
|
|
f2.write(".. autoclass:: pyrogram.{}()".format(type))
|
|
|
|
f.write(template.format(**fmt_keys))
|
|
|
|
# Bound Methods
|
|
|
|
categories = dict(
|
|
message="""
|
|
Message
|
|
Message.click
|
|
Message.delete
|
|
Message.download
|
|
Message.forward
|
|
Message.pin
|
|
Message.edit_text
|
|
Message.edit_caption
|
|
Message.edit_media
|
|
Message.edit_reply_markup
|
|
Message.reply_text
|
|
Message.reply_animation
|
|
Message.reply_audio
|
|
Message.reply_cached_media
|
|
Message.reply_chat_action
|
|
Message.reply_contact
|
|
Message.reply_document
|
|
Message.reply_game
|
|
Message.reply_inline_bot_result
|
|
Message.reply_location
|
|
Message.reply_media_group
|
|
Message.reply_photo
|
|
Message.reply_poll
|
|
Message.reply_sticker
|
|
Message.reply_venue
|
|
Message.reply_video
|
|
Message.reply_video_note
|
|
Message.reply_voice
|
|
""",
|
|
chat="""
|
|
Chat
|
|
Chat.archive
|
|
Chat.unarchive
|
|
Chat.set_title
|
|
Chat.set_description
|
|
Chat.set_photo
|
|
Chat.kick_member
|
|
Chat.unban_member
|
|
Chat.restrict_member
|
|
Chat.promote_member
|
|
""",
|
|
user="""
|
|
User
|
|
User.archive
|
|
User.unarchive
|
|
""",
|
|
callback_query="""
|
|
Callback Query
|
|
CallbackQuery.answer
|
|
CallbackQuery.edit_message_text
|
|
CallbackQuery.edit_message_caption
|
|
CallbackQuery.edit_message_media
|
|
CallbackQuery.edit_message_reply_markup
|
|
""",
|
|
inline_query="""
|
|
InlineQuery
|
|
InlineQuery.answer
|
|
"""
|
|
)
|
|
|
|
root = PYROGRAM_API_DEST + "/bound-methods"
|
|
|
|
shutil.rmtree(root, ignore_errors=True)
|
|
os.mkdir(root)
|
|
|
|
with open(HOME + "/template/bound-methods.rst") as f:
|
|
template = f.read()
|
|
|
|
with open(root + "/index.rst", "w") as f:
|
|
fmt_keys = {}
|
|
|
|
for k, v in categories.items():
|
|
name, *bound_methods = get_title_list(v)
|
|
|
|
fmt_keys.update({"{}_hlist".format(k): "\n ".join("- :meth:`~{}`".format(bm) for bm in bound_methods)})
|
|
|
|
fmt_keys.update(
|
|
{"{}_toctree".format(k): "\n ".join("{} <{}>".format(bm.split(".")[1], bm) for bm in bound_methods)})
|
|
|
|
# noinspection PyShadowingBuiltins
|
|
for bm in bound_methods:
|
|
with open(root + "/{}.rst".format(bm), "w") as f2:
|
|
title = "{}()".format(bm)
|
|
|
|
f2.write(title + "\n" + "=" * len(title) + "\n\n")
|
|
f2.write(".. automethod:: pyrogram.{}()".format(bm))
|
|
|
|
f.write(template.format(**fmt_keys))
|
|
|
|
|
|
def start():
|
|
global page_template
|
|
global toctree
|
|
|
|
shutil.rmtree(DESTINATION, ignore_errors=True)
|
|
|
|
with open(HOME + "/template/page.txt", encoding="utf-8") as f:
|
|
page_template = f.read()
|
|
|
|
with open(HOME + "/template/toctree.txt", encoding="utf-8") as f:
|
|
toctree = f.read()
|
|
|
|
generate(TYPES_PATH, TYPES_BASE)
|
|
generate(FUNCTIONS_PATH, FUNCTIONS_BASE)
|
|
pyrogram_api()
|
|
|
|
|
|
if "__main__" == __name__:
|
|
FUNCTIONS_PATH = "../../pyrogram/api/functions"
|
|
TYPES_PATH = "../../pyrogram/api/types"
|
|
HOME = "."
|
|
DESTINATION = "../../docs/source/telegram"
|
|
PYROGRAM_API_DEST = "../../docs/source/api"
|
|
|
|
start()
|