reused cached excel fonts to avoid max font bug w/ rec2excel

Author: jdh2358

lib/matplotlib/mlab.py

@@ -2350,6 +2350,13 @@ def toval(self, x):
     def fromstr(self, s):
         return s
 
+
+    def __hash__(self):
+        """
+        override the hash function of any of the formatters, so that we don't create duplicate excel format styles
+        """
+        return hash(self.__class__)
+
 class FormatString(FormatObj):
     def tostr(self, x):
         val = repr(x)
@@ -2378,6 +2385,9 @@ def __init__(self, precision=4, scale=1.):
         self.precision = precision
         self.scale = scale
 
+    def __hash__(self):
+        return hash((self.__class__, self.precision, self.scale))
+
     def toval(self, x):
         if x is not None:
             x = x * self.scale
@@ -2425,6 +2435,10 @@ class FormatDate(FormatObj):
     def __init__(self, fmt):
         self.fmt = fmt
 
+    def __hash__(self):
+        return hash((self.__class__, self.fmt))
+
+
     def toval(self, x):
         if x is None: return 'None'
         return x.strftime(self.fmt)

lib/mpl_toolkits/exceltools.py

@@ -1,6 +1,6 @@
 """
 Some io tools for excel -- requires xlwt
-
+l
 Example usage:
 
     import matplotlib.mlab as mlab
@@ -33,9 +33,16 @@ def xlformat_factory(format):
     copy the format, perform any overrides, and attach an xlstyle instance
     copied format is returned
     """
-    format = copy.deepcopy(format)
 
+    #if we have created an excel format already using this format,
+    #don't recreate it; mlab.FormatObj override has to make objs with
+    #the same props hash to the same value
+    key = hash(format)
+    fmt_ = xlformat_factory.created_formats.get(key)
+    if fmt_ is not None:
+        return fmt_
 
+    format = copy.deepcopy(format)
 
     xlstyle = excel.XFStyle()
     if isinstance(format, mlab.FormatPercent):
@@ -55,8 +62,12 @@ def xlformat_factory(format):
 
     format.xlstyle = xlstyle
 
+    xlformat_factory.created_formats[ key ] = format
+
     return format
 
+xlformat_factory.created_formats = {}
+
 def rec2excel(r, ws, formatd=None, rownum=0, colnum=0, nanstr='NaN', infstr='Inf'):
     """
     save record array r to excel xlwt worksheet ws