40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import re
|
|
import sys
|
|
import io
|
|
from PyQt6.uic import compileUi
|
|
|
|
ui_file = sys.argv[1]
|
|
py_file = sys.argv[2]
|
|
buf = io.StringIO()
|
|
compileUi(open(ui_file), buf)
|
|
|
|
outdata = buf.getvalue()
|
|
outdata = outdata.replace(
|
|
"from PyQt6 import QtCore, QtGui, QtWidgets",
|
|
"from PyQt6 import QtCore, QtGui, QtWidgets\nfrom aqt.utils import tr\n"
|
|
)
|
|
outdata = re.sub(
|
|
r'(?:QtGui\.QApplication\.)?_?translate\(".*?", "(.*?)"', "tr.\\1(", outdata
|
|
)
|
|
|
|
|
|
outlines = []
|
|
qt_bad_types = [
|
|
".connect(",
|
|
]
|
|
for line in outdata.splitlines():
|
|
for substr in qt_bad_types:
|
|
if substr in line:
|
|
line = line + " # type: ignore"
|
|
break
|
|
if line == "from . import icons_rc":
|
|
continue
|
|
line = line.replace(":/icons/", "icons:")
|
|
line = line.replace("QAction.PreferencesRole", "QAction.MenuRole.PreferencesRole")
|
|
line = line.replace("QAction.AboutRole", "QAction.MenuRole.AboutRole")
|
|
line = line.replace("QComboBox.AdjustToMinimumContentsLength", "QComboBox.SizeAdjustPolicy.AdjustToMinimumContentsLength")
|
|
outlines.append(line)
|
|
|
|
with open(py_file, "w") as file:
|
|
file.write("\n".join(outlines))
|