Python: Add __contains__ method to Repomd() class

diff --git a/src/python/__init__.py b/src/python/__init__.py
index 38e7425..763dd9f 100644 (file)
--- a/src/python/__init__.py
+++ b/src/python/__init__.py
@@ -158,6 +158,12 @@ class Repomd(_createrepo_c.Repomd):
     def __missing__(self, key):
         raise KeyError("Record with type '%s' doesn't exist" % key)
 
+    def __contains__(self, key):
+        for rec in self.records:
+            if rec.type == key:
+                return True
+        return False
+
 # RepomdRecord class
 
 class RepomdRecord(_createrepo_c.RepomdRecord):

diff --git a/tests/python/tests/test_repomd.py b/tests/python/tests/test_repomd.py
index 2244ba2..e54ce20 100644 (file)
--- a/tests/python/tests/test_repomd.py
+++ b/tests/python/tests/test_repomd.py
@@ -146,3 +146,5 @@ class TestCaseRepomd(unittest.TestCase):
         self.assertEqual(rec.type, "primary")
 
         self.assertRaises(KeyError, repomd.__getitem__, "foobar")
+
+        self.assertTrue("primary" in repomd)