Mercurial Repositories > eric / file diff

--- a/eric6/Graphics/UMLClassDiagramBuilder.py	Sun Apr 12 15:03:27 2020 +0200
+++ b/eric6/Graphics/UMLClassDiagramBuilder.py	Sun Apr 12 18:40:37 2020 +0200
@@ -241,7 +241,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
         if _class.name not in self.allModules[self.file]:
@@ -260,7 +261,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
         if _class not in self.allModules[self.file]:
@@ -279,7 +281,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):