--- a/Project/CreateDialogCodeDialog.py Sun Sep 04 16:04:20 2011 +0200 +++ b/Project/CreateDialogCodeDialog.py Sun Sep 04 16:05:27 2011 +0200 @@ -33,6 +33,8 @@ """ Class implementing a dialog to generate code for a Qt4 dialog. """ + DialogClasses = {"QDialog", "QWidget", "QMainWindow", "QWizard"} + def __init__(self, formName, project, parent = None): """ Constructor @@ -83,7 +85,8 @@ classesList = [] for cls in list(self.__module.classes.values()): - classesList.append(cls.name) + if not set(cls.super).isdisjoint(CreateDialogCodeDialog.DialogClasses): + classesList.append(cls.name) classesList.sort() self.classNameCombo.addItems(classesList)