Mercurial Repositories > eric / file diff

--- a/Graphics/UMLDialog.py	Sat Sep 08 13:02:43 2012 +0200
+++ b/Graphics/UMLDialog.py	Sat Sep 08 17:36:14 2012 +0200
@@ -20,10 +20,12 @@
     """
     Class implementing a dialog showing UML like diagrams.
     """
-    def __init__(self, diagramName="Unnamed", parent=None, name=None):
+    def __init__(self, buildFunction=None, diagramName="Unnamed", parent=None, name=None):
         """
         Constructor
         
+        @param buildFunction function to build the diagram contents (function)
+        @param diagramName name of the diagram (string)
         @param parent parent widget of the view (QWidget)
         @param name name of the view widget (string)
         """
@@ -34,6 +36,7 @@
         else:
             self.setObjectName(name)
         
+        self.buildFunction = buildFunction
         self.scene = QGraphicsScene(0.0, 0.0, 800.0, 600.0)
         self.umlView = UMLGraphicsView(self.scene, diagramName, self, "umlView")
         
@@ -60,3 +63,11 @@
         @param name diagram name (string)
         """
         self.umlView.setDiagramName(name)
+    
+    def show(self):
+        """
+        Overriden method to show the dialog.
+        """
+        if self.buildFunction:
+            self.buildFunction()
+        super().show()