Move if py3 to module level

sebastinas · sebastinas · commit e74bbd20644d · 2015-05-16T20:24:43.000+02:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/bpython/inspection.py b/bpython/inspection.py
@@ -286,8 +286,11 @@ def get_encoding_file(fname):
     return 'ascii'
 
 
-def get_source_unicode(obj):
-    """Returns a decoded source of object"""
-    if py3:
+if py3:
+    def get_source_unicode(obj):
+        """Returns a decoded source of object"""
         return inspect.getsource(obj)
-    return inspect.getsource(obj).decode(get_encoding(obj))
+else:
+    def get_source_unicode(obj):
+        """Returns a decoded source of object"""
+        return inspect.getsource(obj).decode(get_encoding(obj))
