+# This file was imported from
+# https://bitbucket.org/marcusva/python-utils/overview
+# And slightly adapted
+
+"""OS-specific font detection."""
+import os
+import sys
+from subprocess import Popen, PIPE
+
+if sys.platform in ("win32", "cli"):
+ import winreg
+
+__all__ = ["STYLE_NORMAL", "STYLE_BOLD", "STYLE_ITALIC", "STYLE_LIGHT",
+ "init", "list_fonts", "get_fonts", "get_font"
+ ]
+
+# Font cache entries:
+# { family : [...,
+# (name, styles, fonttype, filename)
+# ...
+# ]
+# }
+__FONTCACHE = None
+
+
+STYLE_NONE = 0x00
+STYLE_NORMAL = 0x01
+STYLE_BOLD = 0x02
+STYLE_ITALIC = 0x04
+STYLE_LIGHT = 0x08
+STYLE_MEDIUM = 0x10
+
+def _add_font(family, name, styles, fonttype, filename):
+ """Adds a font to the internal font cache."""
+ global __FONTCACHE
+
+ if family not in __FONTCACHE:
+ __FONTCACHE[family] = []
+ __FONTCACHE[family].append((name, styles, fonttype, filename))
+
+
+def _cache_fonts_win32():
+ """Caches fonts on a Win32 platform."""
+ key = "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Fonts"
+ regfonts = []
+ try:
+ with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, key) as fontkey:
+ idx = 0
+ enumval = winreg.EnumValue
+ rappend = regfonts.append
+ while True:
+ rappend(enumval(fontkey, idx)[:2])
+ idx += 1
+ except WindowsError:
+ pass
+
+ # TODO: integrate alias handling for fonts within the registry.
+ # TODO: Scan and index fonts from %SystemRoot%\\Fonts that are not in the
+ # registry
+
+ # Received all fonts from the registry.
+ for name, filename in regfonts:
+ fonttype = os.path.splitext(filename)[1][1:].lower()
+ if name.endswith("(TrueType)"):
+ name = name[:-10].strip()
+ if name.endswith("(All Res)"):
+ name = name[:-9].strip()
+ style = STYLE_NORMAL
+ if name.find(" Bold") >= 0:
+ style |= STYLE_BOLD
+ if name.find(" Italic") >= 0 or name.find(" Oblique") >= 0:
+ style |= STYLE_ITALIC
+
+ family = name
+ for rm in ("Bold", "Italic", "Oblique"):
+ family = family.replace(rm, "")
+ family = family.lower().strip()
+
+ fontpath = os.environ.get("SystemRoot", "C:\\Windows")
+ fontpath = os.path.join(fontpath, "Fonts")
+ if filename.find("\\") == -1:
+ # No path delimiter is given; we assume it to be a font in
+ # %SystemRoot%\Fonts
+ filename = os.path.join(fontpath, filename)
+ _add_font(family, name, style, fonttype, filename)
+
+
+def _cache_fonts_darwin():
+ """Caches fonts on Mac OS."""
+ raise NotImplementedError("Mac OS X support is not given yet")
+
+
+def _cache_fonts_fontconfig():
+ """Caches font on POSIX-alike platforms."""
+ try:
+ command = "fc-list : file family style fullname fullnamelang"
+ proc = Popen(command, stdout=PIPE, shell=True, stderr=PIPE)
+ pout = proc.communicate()[0]
+ output = pout.decode("utf-8")
+ except OSError:
+ return
+
+ for entry in output.split(os.linesep):
+ if entry.strip() == "":
+ continue
+ values = entry.split(":")
+ filename = values[0]
+
+ # get the font type
+ fname, fonttype = os.path.splitext(filename)
+ if fonttype == ".gz":
+ fonttype = os.path.splitext(fname)[1][1:].lower()
+ else:
+ fonttype = fonttype.lstrip(".").lower()
+
+ # get the font name
+ name = None
+ if len(values) > 3:
+ fullnames, fullnamelangs = values[3:]
+ langs = fullnamelangs.split(",")
+ try:
+ offset = langs.index("fullnamelang=en")
+ except ValueError:
+ offset = -1
+ if offset == -1:
+ try:
+ offset = langs.index("en")
+ except ValueError:
+ offset = -1
+ if offset != -1:
+ # got an english name, use that one
+ name = fullnames.split(",")[offset]
+ if name.startswith("fullname="):
+ name = name[9:]
+ if name is None:
+ if fname.endswith(".pcf") or fname.endswith(".bdf"):
+ name = os.path.basename(fname[:-4])
+ else:
+ name = os.path.basename(fname)
+ name = name.lower()
+
+ # family and styles
+ family = values[1].strip().lower()
+ stylevals = values[2].strip()
+ style = STYLE_NONE
+
+ if stylevals.find("Bold") >= 0:
+ style |= STYLE_BOLD
+ if stylevals.find("Light") >= 0:
+ style |= STYLE_LIGHT
+ if stylevals.find("Italic") >= 0 or stylevals.find("Oblique") >= 0:
+ style |= STYLE_ITALIC
+ if stylevals.find("Medium") >= 0:
+ style |= STYLE_MEDIUM
+ if style == STYLE_NONE:
+ style = STYLE_NORMAL
+ _add_font(family, name, style, fonttype, filename)
+
+
+def init():
+ """Initialises the internal font cache.
+
+ It does not need to be called explicitly.
+ """
+ global __FONTCACHE
+ if __FONTCACHE is not None:
+ return
+ __FONTCACHE = {}
+ if sys.platform in ("win32", "cli"):
+ _cache_fonts_win32()
+ elif sys.platform == "darwin":
+ _cache_fonts_darwin()
+ else:
+ _cache_fonts_fontconfig()
+
+
+def list_fonts():
+ """Returns an iterator over the cached fonts."""
+ if __FONTCACHE is None:
+ init()
+ if len(__FONTCACHE) == 0:
+ yield None
+ for family, entries in __FONTCACHE.items():
+ for fname, styles, fonttype, filename in entries:
+ yield (family, fname, styles, fonttype, filename)
+
+
+def get_fonts(name, style=STYLE_NONE, ftype=None):
+ """Retrieves all fonts matching the given family or font name."""
+ if __FONTCACHE is None:
+ init()
+ if len(__FONTCACHE) == 0:
+ return None
+
+ results = []
+ rappend = results.append
+
+ name = name.lower()
+ if ftype:
+ ftype = ftype.lower()
+
+ fonts = __FONTCACHE.get(name, [])
+ for fname, fstyles, fonttype, filename in fonts:
+ if ftype and fonttype != ftype:
+ # ignore font filetype mismatches
+ continue
+ if style == STYLE_NONE or fstyles == style:
+ rappend((name, fname, fstyles, fonttype, filename))
+
+ for family, fonts in __FONTCACHE.items():
+ for fname, fstyles, fonttype, filename in fonts:
+ if fname.lower() == name and filename not in results:
+ rappend((family, fname, fstyles, fonttype, filename))
+ return results
+
+
+def get_font(name, style=STYLE_NONE, ftype=None):
+ """Retrieves the best matching font file for the given name and
+ criteria.
+ """
+ retvals = get_fonts(name, style, ftype)
+ if len(retvals) > 0:
+ return retvals[0]
+ return None