143 lines
4.7 KiB
Python
143 lines
4.7 KiB
Python
# Pyrogram - Telegram MTProto API Client Library for Python
|
|
# Copyright (C) 2017-2020 Dan <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 csv
|
|
import os
|
|
import re
|
|
import shutil
|
|
|
|
HOME = "compiler/error"
|
|
DEST = "pyrogram/errors/exceptions"
|
|
NOTICE_PATH = "NOTICE"
|
|
|
|
|
|
def snek(s):
|
|
# https://stackoverflow.com/questions/1175208/elegant-python-function-to-convert-camelcase-to-snake-case
|
|
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 caml(s):
|
|
s = snek(s).split("_")
|
|
return "".join([str(i.title()) for i in s])
|
|
|
|
|
|
def start():
|
|
shutil.rmtree(DEST, ignore_errors=True)
|
|
os.makedirs(DEST)
|
|
|
|
files = [i for i in os.listdir("{}/source".format(HOME))]
|
|
|
|
with open(NOTICE_PATH, encoding="utf-8") as f:
|
|
notice = []
|
|
|
|
for line in f.readlines():
|
|
notice.append("# {}".format(line).strip())
|
|
|
|
notice = "\n".join(notice)
|
|
|
|
with open("{}/all.py".format(DEST), "w", encoding="utf-8") as f_all:
|
|
f_all.write(notice + "\n\n")
|
|
f_all.write("count = {count}\n\n")
|
|
f_all.write("exceptions = {\n")
|
|
|
|
count = 0
|
|
|
|
for i in files:
|
|
code, name = re.search(r"(\d+)_([A-Z_]+)", i).groups()
|
|
|
|
f_all.write(" {}: {{\n".format(code))
|
|
|
|
init = "{}/__init__.py".format(DEST)
|
|
|
|
if not os.path.exists(init):
|
|
with open(init, "w", encoding="utf-8") as f_init:
|
|
f_init.write(notice + "\n\n")
|
|
|
|
with open(init, "a", encoding="utf-8") as f_init:
|
|
f_init.write("from .{}_{} import *\n".format(name.lower(), code))
|
|
|
|
with open("{}/source/{}".format(HOME, i), encoding="utf-8") as f_csv, \
|
|
open("{}/{}_{}.py".format(DEST, name.lower(), code), "w", encoding="utf-8") as f_class:
|
|
reader = csv.reader(f_csv, delimiter="\t")
|
|
|
|
super_class = caml(name)
|
|
name = " ".join([str(i.capitalize()) for i in re.sub(r"_", " ", name).lower().split(" ")])
|
|
|
|
sub_classes = []
|
|
|
|
f_all.write(" \"_\": \"{}\",\n".format(super_class))
|
|
|
|
for j, row in enumerate(reader):
|
|
if j == 0:
|
|
continue
|
|
|
|
count += 1
|
|
|
|
if not row: # Row is empty (blank line)
|
|
continue
|
|
|
|
error_id, error_message = row
|
|
|
|
sub_class = caml(re.sub(r"_X", "_", error_id))
|
|
|
|
f_all.write(" \"{}\": \"{}\",\n".format(error_id, sub_class))
|
|
|
|
sub_classes.append((sub_class, error_id, error_message))
|
|
|
|
with open("{}/template/class.txt".format(HOME), "r", encoding="utf-8") as f_class_template:
|
|
class_template = f_class_template.read()
|
|
|
|
with open("{}/template/sub_class.txt".format(HOME), "r", encoding="utf-8") as f_sub_class_template:
|
|
sub_class_template = f_sub_class_template.read()
|
|
|
|
class_template = class_template.format(
|
|
notice=notice,
|
|
super_class=super_class,
|
|
code=code,
|
|
docstring='"""{}"""'.format(name),
|
|
sub_classes="".join([sub_class_template.format(
|
|
sub_class=k[0],
|
|
super_class=super_class,
|
|
id="\"{}\"".format(k[1]),
|
|
docstring='"""{}"""'.format(k[2])
|
|
) for k in sub_classes])
|
|
)
|
|
|
|
f_class.write(class_template)
|
|
|
|
f_all.write(" },\n")
|
|
|
|
f_all.write("}\n")
|
|
|
|
with open("{}/all.py".format(DEST), encoding="utf-8") as f:
|
|
content = f.read()
|
|
|
|
with open("{}/all.py".format(DEST), "w", encoding="utf-8") as f:
|
|
f.write(re.sub("{count}", str(count), content))
|
|
|
|
print("Compiling Errors: [100%]")
|
|
|
|
|
|
if "__main__" == __name__:
|
|
HOME = "."
|
|
DEST = "../../pyrogram/errors/exceptions"
|
|
NOTICE_PATH = "../../NOTICE"
|
|
|
|
start()
|