Pathlibify font_manager (only internally, doesn't change the API).

anntzer · anntzer · commit 18ff0952aa71 · 2018-02-17T00:19:07.000-08:00
diff --git a/lib/matplotlib/cbook/__init__.py b/lib/matplotlib/cbook/__init__.py
@@ -1023,6 +1023,7 @@ def dedent(s):
     return result
 
 
+@deprecated("3.0")
 def listFiles(root, patterns='*', recurse=1, return_folders=0):
     """
     Recursively list files
diff --git a/lib/matplotlib/font_manager.py b/lib/matplotlib/font_manager.py
@@ -46,11 +46,12 @@
 
 from collections import Iterable
 import json
+import logging
 import os
+from pathlib import Path
 import sys
 from threading import Timer
 import warnings
-import logging
 
 from matplotlib import afm, cbook, ft2font, rcParams, get_cachedir
 from matplotlib.compat import subprocess
@@ -146,13 +147,8 @@
 ]
 
 if not USE_FONTCONFIG and sys.platform != 'win32':
-    home = os.environ.get('HOME')
-    if home is not None:
-        # user fonts on OSX
-        path = os.path.join(home, 'Library', 'Fonts')
-        OSXFontDirectories.append(path)
-        path = os.path.join(home, '.fonts')
-        X11FontDirectories.append(path)
+    OSXFontDirectories.append(str(Path.home() / "Library/Fonts"))
+    X11FontDirectories.append(str(Path.home() / ".fonts"))
 
 
 def get_fontext_synonyms(fontext):
@@ -168,19 +164,20 @@ def get_fontext_synonyms(fontext):
 def list_fonts(directory, extensions):
     """
     Return a list of all fonts matching any of the extensions,
-    possibly upper-cased, found recursively under the directory.
+    found recursively under the directory.
     """
-    pattern = ';'.join(['*.%s;*.%s' % (ext, ext.upper())
-                        for ext in extensions])
-    return cbook.listFiles(directory, pattern)
+    return [str(path)
+            for ext in extensions
+            for path in filter(Path.is_file,
+                               Path(directory).rglob("*.{}".format(ext)))]
 
 
 def win32FontDirectory():
-    """
+    r"""
     Return the user-specified font directory for Win32.  This is
     looked up from the registry key::
 
-      \\\\HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders\\Fonts
+      \\HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders\Fonts
 
     If the key is not found, $WINDIR/Fonts will be returned.
     """
@@ -234,16 +231,10 @@ def win32InstalledFonts(directory=None, fontext='ttf'):
                     # Work around for https://bugs.python.org/issue25778, which
                     # is fixed in Py>=3.6.1.
                     direc = direc.split("\0", 1)[0]
-                    if not os.path.dirname(direc):
-                        direc = os.path.join(directory, direc)
-                    direc = os.path.abspath(direc).lower()
-                    if os.path.splitext(direc)[1][1:] in fontext:
-                        items.add(direc)
-                except EnvironmentError:
-                    continue
-                except WindowsError:
-                    continue
-                except MemoryError:
+                    path = Path(directory, direc).resolve()
+                    if path.suffix.lower() in fontext:
+                        items.add(str(path))
+                except (OSError, MemoryError):
                     continue
             return list(items)
         finally:
@@ -263,6 +254,9 @@ def OSXInstalledFonts(directories=None, fontext='ttf'):
     files = []
     for path in directories:
         if fontext is None:
+            cbook.warn_deprecated(
+                "3.0", "Support for listing all files regardless of extension "
+                "is deprecated.")
             files.extend(cbook.listFiles(path, '*'))
         else:
             files.extend(list_fonts(path, fontext))
@@ -299,7 +293,7 @@ def get_fontconfig_fonts(fontext='ttf'):
     """
     fontext = get_fontext_synonyms(fontext)
     return [fname for fname in _call_fc_list()
-            if os.path.splitext(fname)[1][1:] in fontext]
+            if Path(fname).suffix in fontext]
 
 
 def findSystemFonts(fontpaths=None, fontext='ttf'):
@@ -320,8 +314,7 @@ def findSystemFonts(fontpaths=None, fontext='ttf'):
             fontpaths = [fontdir]
             # now get all installed fonts directly...
             for f in win32InstalledFonts(fontdir):
-                base, ext = os.path.splitext(f)
-                if len(ext)>1 and ext[1:].lower() in fontexts:
+                if Path(f).suffix in fontexts:
                     fontfiles.add(f)
         else:
             fontpaths = X11FontDirectories
@@ -1290,16 +1283,12 @@ def _findfont_cached(self, prop, fontext, directory, fallback_to_default,
         else:
             fontlist = self.ttflist
 
-        if directory is not None:
-            directory = os.path.normcase(directory)
-
         best_score = 1e64
         best_font = None
 
         for font in fontlist:
             if (directory is not None and
-                    os.path.commonprefix([os.path.normcase(font.fname),
-                                          directory]) != directory):
+                    Path(directory) not in Path(font.fname).parents):
                 continue
             # Matching family should have highest priority, so it is multiplied
             # by 10.0
