Add support for Python2 & 3 compatible metaclass usage

2 files changed, 28 insertions, 1 deletions

diff --git a/pies/overrides.py b/pies/overrides.py
index 926d51b..e8376f5 100644
--- a/pies/overrides.py
+++ b/pies/overrides.py
@@ -41,7 +41,7 @@ native_next = next
 
 common = ['native_dict', 'native_round', 'native_filter', 'native_map', 'native_range', 'native_str', 'native_chr',
           'native_input', 'PY2', 'PY3', 'u', 'itemsview', 'valuesview', 'keysview', 'execute', 'integer_types',
-          'native_next']
+          'native_next', 'with_metaclass']
 
 if PY3:
     import urllib
@@ -200,3 +200,28 @@ else:
 
     __all__ = common + ['round', 'dict', 'apply', 'cmp', 'coerce', 'execfile', 'raw_input', 'unpacks', 'str', 'chr',
                         'input', 'range', 'filter', 'map', 'zip']
+
+def with_metaclass(meta, *bases):
+    """
+        Enables use of meta classes across Python Versions.
+        taken from jinja2/_compat.py
+
+        Use it like this::
+
+            class BaseForm(object):
+                pass
+
+            class FormType(type):
+                pass
+
+            class Form(with_metaclass(FormType, BaseForm)):
+                pass
+    """
+    class metaclass(meta):
+        __call__ = type.__call__
+        __init__ = type.__init__
+        def __new__(cls, name, this_bases, d):
+            if this_bases is None:
+                return type.__new__(cls, name, (), d)
+            return meta(name, bases, d)
+    return metaclass('temporary_class', None, {})
diff --git a/pies/unittest.py b/pies/unittest.py
index 03b130e..328e554 100644
--- a/pies/unittest.py
+++ b/pies/unittest.py
@@ -3,6 +3,8 @@ from __future__ import absolute_import
 import sys
 from unittest import *
 
+NativeTestCase = TestCase
+
 if sys.version_info < (2, 7):
     skip = lambda why: (lambda func: 'skip')
     skipIf = lambda cond, why: (skip(why) if cond else lambda func: func)