Fixed py3k __str__ implementation

2013-09-17 14:28:37 +03:00
parent 15e3cab6ac
commit 8a3e634430
1 changed files with 6 additions and 2 deletions
--- a/sqlalchemy_utils/types/weekdays.py
+++ b/sqlalchemy_utils/types/weekdays.py
@@ -34,8 +34,12 @@ class WeekDay(object):
    def __repr__(self):
        return '%s(%r)' % (self.__class__.__name__, self.index)

-    def __str__(self):
-        return six.text_type(self).encode('utf-8')
+    if sys.version_info[0] >= 3:  # Python 3
+        def __str__(self):
+            return self.__unicode__()
+    else:  # Python 2
+        def __str__(self):
+            return self.__unicode__().encode('utf8')

    def __unicode__(self):
        return self.name