View difference between Paste ID: te3Ap50C and vghD1bVJ

 # implements_to_string class decorator that helps implementing classes with __unicode__ or __str__ methods
-# Based on: http://lucumr.pocoo.org/2013/5/21/porting-to-python-3-redux/
+# Based on: Steven's suggestions
-# However, the code below is supposed to work for an arbitrary encoding, not just utf-8
+
 from __future__ import print_function
 import sys
 
-isPy3k = sys.version_info[0] == 3
+def decorate(cls):
     print("decorate called")
-if isPy3k:
+    if sys.version_info[0] > 2:
-    implements_to_string = lambda x: x
+        cls.__dict__["__str__"].__name__ = '__bytes__'
-else:
+        cls.__dict__["__unicode__"].__name__ = '__str__'
-    print "you're using Python 2"
+        cls.__bytes__ = cls.__dict__["__str__"]
-    def implements_to_string(cls):
+        cls.__str__ = cls.__dict__["__unicode__"]
-        class Klass(cls):
+    return cls
-            def __init__(self, *args, **kwargs):
+
-                super(Klass, self).__init__(*args, **kwargs)
+@decorate
-                print "defining __unicode__"
+
-                cls.__unicode__ = cls.__str__
+
-                print "defining __str__, which uses %s encoding" % self.encoding
+    def __init__(self):
-                cls.__str__ = lambda x: x.__unicode__().encode(self.encoding)
+        self.__dict__["encoding"] = self.encoding
-        return Klass
+
     def __str__(self):
-@implements_to_string
+        return "str called".encode(self.encoding)
 
     def __unicode__(self):
         return "unicode called"
 
     @property
-        return "rot13"
+
         """In reality this method extracts the encoding from a file"""
         return "utf-8" # rot13 no longer exists in Python3
-        return "str called"
+
 if __name__ == "__main__":
     t = Test()
     if sys.version_info[0] == 2:
-    print unicode(t)
+        print(unicode(t))
-    print str(t)
+    print(str(t))