Mercurial Repositories > eric / file diff

--- a/Plugins/PluginWizardQFileDialog.py	Fri Dec 13 22:45:47 2013 +0100
+++ b/Plugins/PluginWizardQFileDialog.py	Fri Dec 13 23:39:14 2013 +0100
@@ -9,6 +9,8 @@
 
 from __future__ import unicode_literals    # __IGNORE_WARNING__
 
+import re
+
 from PyQt4.QtCore import QObject
 from PyQt4.QtGui import QDialog
 
@@ -44,6 +46,8 @@
         """
         super(FileDialogWizard, self).__init__(ui)
         self.__ui = ui
+        
+        self.__pyqtRe = re.compile(r"(?:import|from)\s+PyQt([45])")
 
     def activate(self):
         """
@@ -99,9 +103,15 @@
         @param editor reference to the current editor
         @return the generated code (string)
         """
+        match = self.__pyqtRe.search(editor.text())
+        if match is None:
+            pyqtVariant = 0    # unknown
+        else:
+            pyqtVariant = int(match.group(1))   # 4 or 5
+        
         from WizardPlugins.FileDialogWizard.FileDialogWizardDialog import \
             FileDialogWizardDialog
-        dlg = FileDialogWizardDialog(None)
+        dlg = FileDialogWizardDialog(pyqtVariant, None)
         if dlg.exec_() == QDialog.Accepted:
             line, index = editor.getCursorPosition()
             indLevel = editor.indentation(line) // editor.indentationWidth()