Блог

У меня есть тонна диалоговых окон и мастеров в большом приложении QT / Python, но я не могу определить, какой дочерний класс запускает родительский метод exec_. Есть ли способ сделать это с помощью mock или любой другой библиотеки? Конечно, я могу найти его через debug, но мне нужен программный способ сделать это.

 import mock
from PySide2 import QtWidgets

class CustomDialog(QtWidgets.QDialog):
    pass


class AnotherCustomDialog(QtWidgets.QDialog) :
    pass


def launch_custom_dialog() :
    dlg = CustomDialog() 
    dlg.exec_() 


with mock.patch.object(QtWidgets.QDialog, 'exec_') as mock_dialog:
    test_which_calls_launch_custom_dialog()
    if mock_dialog.called:
        # How do I find the name of the child? I.e. CustomDialog
  

Если вы вызываете это самостоятельно:

 def launch_custom_dialog() :
    dlg = CustomDialog()    # or could be any of your custom dialogs
    dlg.exec_()
    return dlg
  

Затем вы можете создать свой собственный промежуточный базовый класс:

 class CustomDialogBase(QtWidgets.QDialog):
    def exec_(self, *args, **kwargs):
        self.note_caller = self
        return super().exec_(*args, **kwargs)

class CustomDialog(CustomDialogBase):
    pass
class AnotherCustomDialog(CustomDialogBase) :
    pass
  

Теперь вы можете проверить, какой это был дочерний класс:

 dlg = launch_custom_dialog()
print(dlg.note_caller)