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") | ||