🚑 fix unicode support on Python 2.7

Fixes two UnicodeEncodeError's occurring when hashing and writing files. Closes #281
moremoban · Jul 9, 2019 · 10bebef · 10bebef
1 parent ec48af4
commit 10bebef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/moban/hashstore.py b/moban/hashstore.py
@@ -65,6 +65,8 @@ def get_file_hash(afile):
 
 def get_hash(content):
     md5 = hashlib.md5()
+    if PY2 and content.__class__.__name__ == "unicode":
+        content = content.encode("utf-8")
     md5.update(content)
     return md5.digest().decode("latin1")
 

diff --git a/moban/utils.py b/moban/utils.py
@@ -7,6 +7,7 @@
 from moban import constants, exceptions
 
 log = logging.getLogger(__name__)
+PY2 = sys.version_info[0] == 2
 
 
 def merge(left, right):
@@ -60,6 +61,8 @@ def file_permissions(afile):
 
 
 def write_file_out(filename, content):
+    if PY2 and content.__class__.__name__ == "unicode":
+        content = content.encode("utf-8")
     dest_folder = os.path.dirname(filename)
     if dest_folder:
         mkdir_p(dest_folder)