Merge pull request #5570 from tomoemon/fix_base64_animation

WeatherGod · WeatherGod · commit 2baedb012e7a · 2015-11-30T10:33:53.000-05:00
use base64.encodestring on python2.7
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -27,7 +27,12 @@
 import platform
 import sys
 import itertools
-import base64
+try:
+    # python3
+    from base64 import encodebytes
+except ImportError:
+    # python2
+    from base64 import encodestring as encodebytes
 import contextlib
 import tempfile
 from matplotlib.cbook import iterable, is_string_like
@@ -927,7 +932,7 @@ def to_html5_video(self):
 
             # Now open and base64 encode
             with open(f.name, 'rb') as video:
-                vid64 = base64.encodebytes(video.read())
+                vid64 = encodebytes(video.read())
                 self._base64_video = vid64.decode('ascii')
                 self._video_size = 'width="{0}" height="{1}"'.format(
                         *writer.frame_size)
