mirror of
https://github.com/HIllya51/LunaTranslator.git
synced 2024-12-28 08:04:13 +08:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import os, re
|
|
import ast
|
|
|
|
|
|
class TypeHintRemover_1(ast.NodeTransformer):
|
|
def visit_FunctionDef(self, node):
|
|
node.returns = None
|
|
if node.args.args:
|
|
for arg in node.args.args:
|
|
arg.annotation = None
|
|
return node
|
|
|
|
|
|
class TypeHintRemover_2(ast.NodeTransformer):
|
|
def visit_AnnAssign(self, node):
|
|
obj = ast.Assign([node.target], node.value)
|
|
obj.lineno = node.lineno
|
|
return obj
|
|
|
|
|
|
def typeHintRemover(source):
|
|
parsed_source = ast.parse(source)
|
|
transformed = TypeHintRemover_1().visit(parsed_source)
|
|
transformed = TypeHintRemover_2().visit(transformed)
|
|
return ast.unparse(transformed)
|
|
|
|
|
|
def parsecode(code: str):
|
|
# PyQt
|
|
code = code.replace("self.screen().geometry().height()", "99999")
|
|
code = code.replace("self.parent().devicePixelRatioF()", "1")
|
|
code = code.replace("self.devicePixelRatioF()", "1")
|
|
code = re.sub(
|
|
r"(Q[a-zA-Z0-9_]+)\.[a-zA-Z0-9_]+\.([a-zA-Z0-9_]+)([ \)\n,:])", r"\1.\2\3", code
|
|
)
|
|
code = typeHintRemover(code)
|
|
return code
|
|
|
|
|
|
for _dir, _, _fs in os.walk("./LunaTranslator"):
|
|
for _f in _fs:
|
|
if not _f.endswith(".py"):
|
|
continue
|
|
path = os.path.normpath(os.path.abspath(os.path.join(_dir, _f)))
|
|
with open(path, "r", encoding="utf8") as ff:
|
|
code = ff.read()
|
|
with open(path, "w", encoding="utf8") as ff:
|
|
ff.write(parsecode(code))
|