diff options
| -rw-r--r-- | python_modules/codegen.py | 6 | ||||
| -rwxr-xr-x | spice_codegen.py | 6 |
2 files changed, 6 insertions, 6 deletions
diff --git a/python_modules/codegen.py b/python_modules/codegen.py index 55f513b..f324498 100644 --- a/python_modules/codegen.py +++ b/python_modules/codegen.py @@ -121,10 +121,10 @@ class CodeWriter: def write(self, s): # Ensure its a unicode string - if six.PY2: - s = unicode(s) - else: + if six.PY3: s = str(s) + else: + s = unicode(s) if len(s) == 0: return diff --git a/spice_codegen.py b/spice_codegen.py index 16ad478..84790af 100755 --- a/spice_codegen.py +++ b/spice_codegen.py @@ -265,10 +265,10 @@ if options.keep_identical_file: pass f = open(dest_file, 'wb') -if six.PY2: - f.write(content) -else: +if six.PY3: f.write(bytes(content, 'UTF-8')) +else: + f.write(content) f.close() six.print_("Wrote %s" % dest_file) |