More flexible wrapping for singletons.

shakefu · Aug 4, 2015 · 90592a9 · 90592a9
1 parent ef53ace
commit 90592a9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytool/lang.py b/pytool/lang.py
@@ -108,7 +108,8 @@ class Test(object):
     # Mirror original class
     cls_name = klass.__name__
     for attr in functools.WRAPPER_ASSIGNMENTS:
-        cls_dict[attr] = getattr(klass, attr)
+        if hasattr(klass, attr):
+            cls_dict[attr] = getattr(klass, attr)
 
     # Make new method that controls singleton behavior
     def __new__(cls, *args, **kwargs):