diff options
Diffstat (limited to 'lib/python/qmk/path.py')
| -rw-r--r-- | lib/python/qmk/path.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/lib/python/qmk/path.py b/lib/python/qmk/path.py index cf087265f..e7e3ec53f 100644 --- a/lib/python/qmk/path.py +++ b/lib/python/qmk/path.py | |||
| @@ -5,6 +5,7 @@ import os | |||
| 5 | 5 | ||
| 6 | from qmk.errors import NoSuchKeyboardError | 6 | from qmk.errors import NoSuchKeyboardError |
| 7 | 7 | ||
| 8 | from bs4 import UnicodeDammit | ||
| 8 | 9 | ||
| 9 | def keymap(keyboard): | 10 | def keymap(keyboard): |
| 10 | """Locate the correct directory for storing a keymap. | 11 | """Locate the correct directory for storing a keymap. |
| @@ -33,3 +34,20 @@ def normpath(path): | |||
| 33 | return os.path.normpath(path) | 34 | return os.path.normpath(path) |
| 34 | 35 | ||
| 35 | return os.path.normpath(os.path.join(os.environ['ORIG_CWD'], path)) | 36 | return os.path.normpath(os.path.join(os.environ['ORIG_CWD'], path)) |
| 37 | |||
| 38 | def unicode_text(filename): | ||
| 39 | """Returns the contents of filename as a UTF-8 string. Tries to DTRT when it comes to encoding. | ||
| 40 | """ | ||
| 41 | with open(filename, "rb") as fd: | ||
| 42 | text = UnicodeDammit(fd.read()) | ||
| 43 | |||
| 44 | if text.contains_replacement_characters: | ||
| 45 | log_warning("%s: Could not determine file encoding, some characters were replaced." % (filename,)) | ||
| 46 | |||
| 47 | return text.unicode_markup or "" | ||
| 48 | |||
| 49 | |||
| 50 | def unicode_lines(filename): | ||
| 51 | """Returns the contents of filename as a UTF-8 string. Tries to DTRT when it comes to encoding. | ||
| 52 | """ | ||
| 53 | return unicode_text(filename).split("\n") | ||
