Update piechart1.py

Author: JorjMcKie

demo/piechart1.py

@@ -33,7 +33,7 @@
           (56, "dodgerblue", "CSU"),
           (193, "red", "SPD"),
           (64, "violetred", "Die Linke"),
-          (63, "green", "Die Grünen"),
+          (63, "green", "Die Grünen"),
           (1, "gray", "fraktionslos"),
          )
 
@@ -49,18 +49,19 @@
 
 for i, c in enumerate(table):
     beta = c[0] / seats * 360          # express no. of seats as angle
+    color = getColor(c[1])             # avoid multiple color lookups
     # the method delivers point of other end of the constructed arc
     # we will use it as input for next round
     point = page.drawSector(center, point, beta, color = white,
-                    fullSector = True, fill = getColor(c[1]))
+                    fullSector = True, fill = color)
     # legend text (takes care of German plural, too)
     text = "%s, %i %s" % (c[2], c[0], "Sitze" if c[0] > 1 else "Sitz")
     pos  = fitz.Point(ts_h, ts_v + i*lineheight)
     page.insertText(pos, text, color = blue)     # legend text
     tl = fitz.Point(pos.x - 30, ts_v - 10 + i*lineheight)
     br = fitz.Point(pos.x - 10, ts_v + i*lineheight)
     rect = fitz.Rect(tl, br)                # legend color bar
-    page.drawRect(rect, fill = getColor(c[1]), color = getColor(c[1]))
+    page.drawRect(rect, fill = color, color = color)
 
 doc.save("piechart1.pdf")