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

namespace = path.split("/")[-1]

if namespace in ["base", "types", "functions"]:

namespace = ""

full_name = f"{(namespace + '.') if namespace else ''}{name}"

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=full_name,

title_markup="=" * len(full_name),

full_class_path="pyrogram.raw.{}".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(f'{i} <{snek(i).replace("_", "-")}>')

if k != base:

inner_path = base + "/" + k + "/index" + ".rst"

module = "pyrogram.raw.{}.{}".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.raw.{}".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)

)

)

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="""

""",

authorization="""

""",

messages="""

""",

chats="""

""",

invite_links="""

""",

chat_topics="""

""",

users="""

""",

contacts="""

""",

password="""

""",

bots="""

""",

phone="""

""",

stickers="""

""",

stories="""

""",

payments="""

""",

advanced="""

)

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

functions = ["idle", "compose"]

for func in functions:

with open(root + "/{}.rst".format(func), "w") as f2:

title = "{}()".format(func)

f2.write(title + "\n" + "=" * len(title) + "\n\n")

f2.write(".. autofunction:: pyrogram.{}()".format(func))

f.write(template.format(**fmt_keys))

# Types

categories = dict(

users_chats="""

""",

messages_media="""

""",

chat_topics="""

""",

bot_commands="""

""",

bot_keyboards="""

""",

inline_mode="""

""",

authorization="""

""",

input_media="""

""",

input_paid_media="""

""",

input_message_content="""

""",

payments="""

)

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.types.{}()\n".format(type))

f.write(template.format(**fmt_keys))

# Bound Methods

categories = dict(

message="""

""",

chat="""

""",

user="""

""",

callback_query="""

""",

inline_query="""

""",

pre_checkout_query="""

""",

shipping_query="""

""",

chat_join_request="""

""",

story="""

""",

active_session="""

""",

)

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.types.{}()".format(bm))

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)

generate(BASE_PATH, BASE_BASE)