From 3cc9566a00cb4deb1ead414c390a29beaccc0bd0 Mon Sep 17 00:00:00 2001
From: Christophe Fergeau <cfergeau@redhat.com>
Date: Tue, 14 Apr 2015 16:08:43 +0200
Subject: codegen: Use six.PY3 rather than six.PY2

Older versions of python-six (at least 1.3.0) defined six.PY3 but not
six.PY2. six.PY2 is only used twice in straightforward tests so it's
easy to use six.PY3 instead.
---
 python_modules/codegen.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'python_modules/codegen.py')

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
-- 
cgit