use base64.encodestring on python2.7

tomoemon · tomoemon · commit c7e98aa744a7 · 2015-11-27T20:30:17.000+09:00
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
@@ -928,7 +933,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)
