FIX: Use unicode

larsoner · larsoner · commit cec3991630d5 · 2017-06-25T15:14:15.000-07:00
diff --git a/sphinx_gallery/gen_gallery.py b/sphinx_gallery/gen_gallery.py
@@ -181,24 +181,23 @@ def generate_gallery_rst(app):
         computation_times += this_computation_times
 
         # we create an index.rst with all examples
-        fhindex = open(os.path.join(gallery_dir, 'index.rst'), 'w')
-        # :orphan: to suppress "not included in TOCTREE" sphinx warnings
-        fhindex.write(":orphan:\n\n" + this_fhindex)
-        for directory in sorted(os.listdir(examples_dir)):
-            if os.path.isdir(os.path.join(examples_dir, directory)):
-                src_dir = os.path.join(examples_dir, directory)
-                target_dir = os.path.join(gallery_dir, directory)
-                this_fhindex, this_computation_times = generate_dir_rst(src_dir, target_dir, gallery_conf,
-                                                                        seen_backrefs)
-                fhindex.write(this_fhindex)
-                computation_times += this_computation_times
-
-        if gallery_conf['download_all_examples']:
-            download_fhindex = generate_zipfiles(gallery_dir)
-            fhindex.write(download_fhindex)
-
-        fhindex.write(SPHX_GLR_SIG)
-        fhindex.flush()
+        with open(os.path.join(gallery_dir, 'index.rst'), 'wb') as fhindex:
+            # :orphan: to suppress "not included in TOCTREE" sphinx warnings
+            fhindex.write((u":orphan:\n\n" + this_fhindex).encode('utf-8'))
+            for directory in sorted(os.listdir(examples_dir)):
+                if os.path.isdir(os.path.join(examples_dir, directory)):
+                    src_dir = os.path.join(examples_dir, directory)
+                    target_dir = os.path.join(gallery_dir, directory)
+                    this_fhindex, this_computation_times = generate_dir_rst(src_dir, target_dir, gallery_conf,
+                                                                            seen_backrefs)
+                    fhindex.write(this_fhindex.encode('utf-8'))
+                    computation_times += this_computation_times
+
+            if gallery_conf['download_all_examples']:
+                download_fhindex = generate_zipfiles(gallery_dir)
+                fhindex.write(download_fhindex.encode('utf-8'))
+
+            fhindex.write(SPHX_GLR_SIG)
 
     if gallery_conf['plot_gallery']:
         print("Computation time summary:")
diff --git a/sphinx_gallery/gen_rst.py b/sphinx_gallery/gen_rst.py
@@ -399,8 +399,8 @@ def generate_dir_rst(src_dir, target_dir, gallery_conf, seen_backrefs):
         print(80 * '_')
         return "", []  # because string is an expected return type
 
-    with open(os.path.join(src_dir, 'README.txt')) as fid:
-        fhindex = fid.read()
+    with open(os.path.join(src_dir, 'README.txt'), 'rb') as fid:
+        fhindex = fid.read().decode('utf-8')
     # Add empty lines to avoid bug in issue #165
     fhindex += "\n\n"
 
