Fix GH-21 ("u function doesn't work on Python 2")

1 files changed, 5 insertions, 2 deletions

diff --git a/pies/overrides.py b/pies/overrides.py
index 834904b..1d64f66 100644
--- a/pies/overrides.py
+++ b/pies/overrides.py
@@ -110,8 +110,11 @@ else:
     for removed in ('apply', 'cmp', 'coerce', 'execfile', 'raw_input', 'unpacks'):
         globals()[removed] = _create_not_allowed(removed)
 
-    def u(string):
-        return codecs.unicode_escape_decode(string[0])
+    def u(s):
+        if isinstance(s, unicode):
+            return s
+        else:
+            return unicode(s.replace(r'\\', r'\\\\'), "unicode_escape")
 
     def execute(_code_, _globs_=None, _locs_=None):
         """Execute code in a namespace."""