aselimov/pubs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

FIX: Encode unicode before writing to file in python2.

Browse Source 
This is necessary because _open returns a file descriptor in binary mode
for python2.
This commit is contained in:
Olivier Mangin 2017-07-13 21:13:02 -04:00
parent 81f2661593
commit cf596206b0
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pubs/content.py
View File

@ -94,7 +94,16 @@ def remove_file(filepath):
def write_file(filepath, data, mode='w'): def write_file(filepath, data, mode='w'):
"""Write data to file.
Data should be unicode except when binary mode is selected,
in which case data is expected to be binary.
"""
check_directory(os.path.dirname(filepath)) check_directory(os.path.dirname(filepath))
if 'b' not in mode and sys.version_info < (3,):
# _open returns in binary mode for python2
# Data must be encoded
data = data.encode('utf-8')
with _open(filepath, mode) as f: with _open(filepath, mode) as f:
f.write(data) f.write(data)