import os, re


def parsecode(code: str):
    # PyQt
    code = code.replace(
        "QApplication.setAttribute(Qt.ApplicationAttribute.AA_EnableHighDpiScaling)", ""
    )
    code = code.replace("QApplication.setHighDpiScaleFactorRoundingPolicy", "print")
    code = code.replace("Qt.HighDpiScaleFactorRoundingPolicy.PassThrough", "")
    code = code.replace("self.screen().geometry().height()", "99999")
    code = code.replace(
        '"./files/runtime/PyQt5/Qt5/plugins"',
        '"./files/runtime/Lib/site-packages/PyQt5/plugins"',
    )
    code = code.replace(
        "./files/runtime/PyQt5/Qt5/bin/Qt5Core.dll",
        "./files/runtime/Lib/site-packages/PyQt5/Qt5Core.dll",
    )
    code = code.replace(
        '    windows.addenvpath("./files/runtime/")',
        '    windows.addenvpath("./files/runtime/")\n    windows.addenvpath("./files/runtime/Lib/site-packages/PyQt5")',
    )
    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 = re.sub(r": [a-zA-Z0-9_]+\)", ")", code)
    code = re.sub(r": [a-zA-Z0-9_]+,", ",", code)
    code = re.sub(r": [a-zA-Z0-9_]+ =", " =", 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))