[ADD] util.direct_inherit_parents

Well named function to get only the direct inherted parents.

Needed for usages in functions that call themself recursively.
Mostly useful for inheritS.

Rewrite `util.inherit_parents` to use this function.

Part-of: #22

Author: KangOl

src/base/tests/test_util.py

@@ -513,6 +513,15 @@ def test_inherit_parents(self, model, expected):
         result = sorted(util.inherit_parents(cr, model))
         self.assertEqual(result, sorted(expected))
 
+    def test_direct_inherit_parents(self):
+        cr = self.env.cr
+        result = sorted(util.direct_inherit_parents(cr, "product.product"))
+        self.assertEqual(len(result), 3)
+        parents, inhs = zip(*result)
+        self.assertEqual(parents, ("mail.activity.mixin", "mail.thread", "product.template"))
+        self.assertTrue(all(inh.model == "product.product" for inh in inhs))
+        self.assertEqual([inh.via for inh in inhs], [None, None, "product_tmpl_id"])
+
 
 class TestNamedCursors(UnitTestCase):
     @staticmethod

src/util/inherit.py

@@ -93,7 +93,10 @@ def for_each_inherit(cr, model, skip=(), interval="[)"):
             yield inh
 
 
-def inherit_parents(cr, model, skip=(), interval="[)"):
+def direct_inherit_parents(cr, model, skip=(), interval="[)"):
+    """
+    Yield the *direct* inherits parents.
+    """
     if skip == "*":
         return
     skip = set(skip)
@@ -103,7 +106,21 @@ def inherit_parents(cr, model, skip=(), interval="[)"):
             continue
         for inh in inhs:
             if inh.model == model and cmp_(inh):
-                yield parent
+                yield parent, inh
                 skip.add(parent)
-                for grand_parent in inherit_parents(cr, parent, skip=skip, interval=interval):
-                    yield grand_parent
+
+
+def inherit_parents(cr, model, skip=(), interval="[)"):
+    """
+    Recursively yield all inherit parents model names
+    """
+    if skip == "*":
+        return
+    skip = set(skip)
+    for parent, _inh in direct_inherit_parents(cr, model, skip=skip, interval=interval):
+        if parent in skip:
+            continue
+        yield parent
+        skip.add(parent)
+        for grand_parent in inherit_parents(cr, parent, skip=skip, interval=interval):
+            yield grand_parent