LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-15 02:22:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Remove unintended private imports and simplify sig copy

Browse Source
This commit is contained in:
Lonami Exo 2023-09-03 11:03:07 +02:00
parent e4d88e061d
commit 8727b87130
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tools/copy_client_signatures.py
View File

@ -12,7 +12,6 @@ to avoid being included.
""" """
import ast import ast
import sys import sys
from _ast import AsyncFunctionDef, ClassDef
from pathlib import Path from pathlib import Path
from typing import Dict, List, Union from typing import Dict, List, Union
@ -24,7 +23,7 @@ class FunctionMethodsVisitor(ast.NodeVisitor):
def visit_FunctionDef(self, node: ast.FunctionDef) -> None: def visit_FunctionDef(self, node: ast.FunctionDef) -> None:
self._try_add_def(node) self._try_add_def(node)
def visit_AsyncFunctionDef(self, node: AsyncFunctionDef) -> None: def visit_AsyncFunctionDef(self, node: ast.AsyncFunctionDef) -> None:
self._try_add_def(node) self._try_add_def(node)
def _try_add_def(self, node: Union[ast.FunctionDef, ast.AsyncFunctionDef]) -> None: def _try_add_def(self, node: Union[ast.FunctionDef, ast.AsyncFunctionDef]) -> None:
@ -38,7 +37,7 @@ class MethodVisitor(ast.NodeVisitor):
self._in_client = False self._in_client = False
self.method_docs: Dict[str, str] = {} self.method_docs: Dict[str, str] = {}
def visit_ClassDef(self, node: ClassDef) -> None: def visit_ClassDef(self, node: ast.ClassDef) -> None:
if node.name == "Client": if node.name == "Client":
assert not self._in_client assert not self._in_client
self._in_client = True self._in_client = True
@ -49,7 +48,7 @@ class MethodVisitor(ast.NodeVisitor):
def visit_FunctionDef(self, node: ast.FunctionDef) -> None: def visit_FunctionDef(self, node: ast.FunctionDef) -> None:
self._try_add_doc(node) self._try_add_doc(node)
def visit_AsyncFunctionDef(self, node: AsyncFunctionDef) -> None: def visit_AsyncFunctionDef(self, node: ast.AsyncFunctionDef) -> None:
self._try_add_doc(node) self._try_add_doc(node)
def _try_add_doc(self, node: Union[ast.FunctionDef, ast.AsyncFunctionDef]) -> None: def _try_add_doc(self, node: Union[ast.FunctionDef, ast.AsyncFunctionDef]) -> None:
@ -62,7 +61,7 @@ class MethodVisitor(ast.NodeVisitor):
def main() -> None: def main() -> None:
client_root = Path.cwd() / sys.argv[1] client_root = Path.cwd() / "client/src/telethon/_impl/client/client"
fm_visitor = FunctionMethodsVisitor() fm_visitor = FunctionMethodsVisitor()
m_visitor = MethodVisitor() m_visitor = MethodVisitor()