Ticket #322: pyamf_classalias.diff

pyamf/tests/test_basic.py

@@ -427 +427 @@
 
         self.assertEquals(pyamf.has_alias(Spam), True)
 
+    def test_required_arguments(self):
+        class Foo(object):
+            def __init__(self, bar, valid=1):
+                pass
+        self.assertRaises(TypeError, pyamf.register_class, Foo)
+
+        class Foo(object):
+            def __init__(self, bar, valid):
+                pass
+        self.assertRaises(TypeError, pyamf.register_class, Foo)
+
+        class Foo(object):
+            def __init__(self, bar=1, valid=1):
+                pass
+        pyamf.register_class(Foo)
+
+
 class UnregisterClassTestCase(ClassCacheClearingTestCase):
     def test_klass(self):
         alias = pyamf.register_class(Spam, 'spam.eggs')

pyamf/__init__.py

@@ -361 +361 @@
                 return self.klass.__new__(self.klass)
             elif type(self.klass) is types.ClassType: # classic class
                 return util.make_classic_instance(self.klass)
-
             raise TypeError, 'invalid class type %r' % self.klass
-
+        
         return self.klass(*args, **kwargs)
 
     def __str__(self):
@@ -634 +633 @@
     if alias is not None and alias in CLASS_CACHE.keys():
         raise ValueError, "alias '%s' already registered" % alias
 
+    # Check that the constructor of the class doesn't require any additonal
+    # arguments.
+    if hasattr(klass, '__init__') and hasattr(klass.__init__, 'im_func'):
+        klass_func = klass.__init__.im_func
+        # Number of arguments - number of default values
+        if klass_func.func_code.co_argcount - len(klass_func.func_defaults or []) > 1:
+            raise TypeError("pyamf doesn't support required init arguments")
+
     x = ClassAlias(klass, alias, attr_func=attr_func, attrs=attrs,
         metadata=metadata)