Skip to content

Commit 49daf9d

Browse files
author
Adam Maris
committed
use utf-8 codec if ensure_ascii is False to avoid UnicodeError
1 parent 3978d77 commit 49daf9d

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

cortexutils/worker.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,8 +119,13 @@ def __write_output(self, data, ensure_ascii=False):
119119
os.makedirs('%s/output' % self.job_directory)
120120
except:
121121
pass
122-
with open('%s/output/output.json' % self.job_directory, mode='w') as f_output:
123-
json.dump(data, f_output, ensure_ascii=ensure_ascii)
122+
123+
f_output = open('%s/output/output.json' % self.job_directory, mode='w')
124+
125+
if not ensure_ascii:
126+
f_output = codecs.getwriter('utf-8')(f_output, 'strict')
127+
128+
json.dump(data, f_output, ensure_ascii=ensure_ascii)
124129

125130
def get_data(self):
126131
"""Wrapper for getting data from input dict.

0 commit comments

Comments
 (0)