Use base64.encodestring under Python 2.x and keep base64.encodebytes for Python 3.

hodgestar · stefanv · commit 19d502988e84 · 2011-03-16T22:53:57.000+02:00
diff --git a/lib/matplotlib/backends/backend_svg.py b/lib/matplotlib/backends/backend_svg.py
@@ -9,6 +9,13 @@
 except ImportError:
     from md5 import md5 #Deprecated in 2.5
 
+if hasattr(base64, 'encodebytes'):
+    # Python 3 case
+    encodebytes = base64.encodebytes
+else:
+    # Python 2 case
+    encodebytes = base64.encodestring
+
 from matplotlib import verbose, __version__, rcParams
 from matplotlib.backend_bases import RendererBase, GraphicsContextBase,\
      FigureManagerBase, FigureCanvasBase
@@ -433,7 +440,7 @@ def draw_image(self, gc, x, y, im):
             _png.write_png(buffer, cols, rows, bytesio)
             im.flipud_out()
             self._svgwriter.write(
-                base64.encodebytes(bytesio.getvalue()).decode('ascii'))
+                encodebytes(bytesio.getvalue()).decode('ascii'))
         else:
             self._imaged[self.basename] = self._imaged.get(self.basename,0) + 1
             filename = '%s.image%d.png'%(self.basename, self._imaged[self.basename])
