Replace most uses of getfilesystemencoding by os.fs{en,de}code.

anntzer · anntzer · commit bd0065e6fb29 · 2018-05-01T10:58:57.000-07:00
diff --git a/lib/matplotlib/__init__.py b/lib/matplotlib/__init__.py
@@ -686,13 +686,6 @@ def _get_cachedir():
 get_cachedir = _wrap('CACHEDIR=%s', _get_cachedir, always=False)
 
 
-def _decode_filesystem_path(path):
-    if not isinstance(path, str):
-        return path.decode(sys.getfilesystemencoding())
-    else:
-        return path
-
-
 def _get_data_path():
     'get the path to matplotlib data'
 
@@ -703,35 +696,25 @@ def _get_data_path():
                                'directory')
         return path
 
-    _file = _decode_filesystem_path(__file__)
-    path = os.sep.join([os.path.dirname(_file), 'mpl-data'])
-    if os.path.isdir(path):
-        return path
-
-    # setuptools' namespace_packages may highjack this init file
-    # so need to try something known to be in matplotlib, not basemap
-    import matplotlib.afm
-    _file = _decode_filesystem_path(matplotlib.afm.__file__)
-    path = os.sep.join([os.path.dirname(_file), 'mpl-data'])
-    if os.path.isdir(path):
-        return path
-
-    # py2exe zips pure python, so still need special check
-    if getattr(sys, 'frozen', None):
-        exe_path = os.path.dirname(_decode_filesystem_path(sys.executable))
-        path = os.path.join(exe_path, 'mpl-data')
-        if os.path.isdir(path):
-            return path
-
-        # Try again assuming we need to step up one more directory
-        path = os.path.join(os.path.split(exe_path)[0], 'mpl-data')
-        if os.path.isdir(path):
-            return path
-
-        # Try again assuming sys.path[0] is a dir not a exe
-        path = os.path.join(sys.path[0], 'mpl-data')
-        if os.path.isdir(path):
-            return path
+    # Yield some filenames next to which we may be able to find mpl-data.
+    def get_candidate_siblings():
+        yield __file__
+        # setuptools' namespace_packages may highjack this init file
+        # so need to try something known to be in Matplotlib, not basemap.
+        import matplotlib.afm
+        yield matplotlib.afm.__file__
+        # py2exe zips pure python, so still need special check.
+        if getattr(sys, 'frozen', None):
+            yield sys.executable
+            # Try again assuming we need to step up one more directory.
+            yield os.path.dirname(sys.executable)
+            # Try again assuming sys.path[0] is a dir not a exe.
+            yield sys.path[0]
+
+    for sibling in get_candidate_siblings():
+        path = Path(sibling).with_name('mpl-data')
+        if path.is_dir():
+            return str(path)
 
     raise RuntimeError('Could not find the matplotlib data files')
 
diff --git a/lib/matplotlib/backends/backend_pdf.py b/lib/matplotlib/backends/backend_pdf.py
@@ -972,7 +972,7 @@ def get_char_width(charcode):
             # Make the charprocs array (using ttconv to generate the
             # actual outlines)
             rawcharprocs = ttconv.get_pdf_charprocs(
-                filename.encode(sys.getfilesystemencoding()), glyph_ids)
+                os.fsencode(filename), glyph_ids)
             charprocs = {}
             for charname in sorted(rawcharprocs):
                 stream = rawcharprocs[charname]
diff --git a/lib/matplotlib/backends/backend_ps.py b/lib/matplotlib/backends/backend_ps.py
@@ -1128,10 +1128,8 @@ def print_figure_impl(fh):
                                 "time; consider using the Cairo backend")
                         else:
                             fh.flush()
-                            convert_ttf_to_ps(
-                                font_filename.encode(
-                                    sys.getfilesystemencoding()),
-                                fh, fonttype, glyph_ids)
+                            convert_ttf_to_ps(os.fsencode(font_filename),
+                                              fh, fonttype, glyph_ids)
             print("end", file=fh)
             print("%%EndProlog", file=fh)
 
diff --git a/lib/matplotlib/dviread.py b/lib/matplotlib/dviread.py
@@ -25,7 +25,6 @@
 import re
 import struct
 import subprocess
-import sys
 import textwrap
 
 import numpy as np
@@ -810,10 +809,7 @@ class PsfontsMap(object):
 
     def __init__(self, filename):
         self._font = {}
-        self._filename = filename
-        if isinstance(filename, bytes):
-            encoding = sys.getfilesystemencoding() or 'utf-8'
-            self._filename = filename.decode(encoding, errors='replace')
+        self._filename = os.fsdecode(filename)
         with open(filename, 'rb') as file:
             self._parse(file)
 
