Mercurial Repositories > eric / file diff

--- a/eric6/Graphics/PackageDiagramBuilder.py	Sun Apr 12 15:03:27 2020 +0200
+++ b/eric6/Graphics/PackageDiagramBuilder.py	Sun Apr 12 18:40:37 2020 +0200
@@ -394,7 +394,8 @@
         if isRbModule:
             name = "{0} (Module)".format(name)
         cl = ClassModel(name, meths[:], attrs[:])
-        cw = ClassItem(cl, False, x, y, noAttrs=self.noAttrs, scene=self.scene)
+        cw = ClassItem(cl, False, x, y, noAttrs=self.noAttrs, scene=self.scene,
+                       colors=self.umlView.getDrawingColors())
         cw.setId(self.umlView.getItemId())
         self.allClasses[className] = cw
     
@@ -411,7 +412,8 @@
         """
         from .ClassItem import ClassItem, ClassModel
         cl = ClassModel(_class)
-        cw = ClassItem(cl, True, x, y, noAttrs=self.noAttrs, scene=self.scene)
+        cw = ClassItem(cl, True, x, y, noAttrs=self.noAttrs, scene=self.scene,
+                       colors=self.umlView.getDrawingColors())
         cw.setId(self.umlView.getItemId())
         self.allClasses[_class] = cw
     
@@ -427,7 +429,8 @@
         """
         from .PackageItem import PackageItem, PackageModel
         pm = PackageModel(name, modules)
-        pw = PackageItem(pm, x, y, scene=self.scene)
+        pw = PackageItem(pm, x, y, scene=self.scene,
+                         colors=self.umlView.getDrawingColors())
         pw.setId(self.umlView.getItemId())
         self.allClasses[name] = pw
     
@@ -444,7 +447,8 @@
                     self.__getCurrentShape(route[1]),
                     self.__getCurrentShape(route[0]),
                     Generalisation,
-                    topToBottom=True)
+                    topToBottom=True,
+                    colors=self.umlView.getDrawingColors())
                 self.scene.addItem(assoc)
     
     def getPersistenceData(self):