Adds config handler for class, interface and enum bases.

Troy Melhase · Troy Melhase · commit c8dd7ad47833 · 2011-10-30T05:05:12.000Z
diff --git a/java2python/compiler/template.py b/java2python/compiler/template.py
@@ -403,7 +403,9 @@ class Class(ClassMethodSharedMixin, Base):
 
     def iterBases(self):
 	""" Yields the base classes for this type. """
-	return iter(self.bases or ['object'])
+        for handler in self.configHandlers('Base'):
+            for base in handler(self):
+                yield base
 
     def iterDecl(self):
 	""" Yields the declaration for this type. """
diff --git a/java2python/config/default.py b/java2python/config/default.py
@@ -48,6 +48,13 @@
     basic.simpleDocString,
 ]
 
+classBaseHandlers = [
+    basic.defaultBases,
+]
+
+interfaceBaseHandlers = [
+    basic.zopeInterfaceBases,
+]
 
 # These generators are called after a class has been completely
 # generated.  The class content sorter sorts the methods of a class by
diff --git a/java2python/mod/basic.py b/java2python/mod/basic.py
@@ -172,3 +172,10 @@ def zopeInterfaceMethodMutator(obj):
 	if method.parameters and method.parameters[0]['name'] == 'self':
 	    method.parameters.pop(0)
 
+
+def defaultBases(obj):
+    return iter(obj.bases or ['object'])
+
+
+def zopeInterfaceBases(obj):
+    return ['zope.interface.Interface']
