From: Tomas Mlcoch <tmlcoch@redhat.com>
Date: Sun, 28 Dec 2014 20:18:44 +0000 (+0100)
Subject: Python: Add __contains__ method to Repomd() class
X-Git-Tag: upstream/0.10.0~141
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=9c22993e5c429163cc595ed937c34355664543b1;p=services%2Fcreaterepo_c.git

Python: Add __contains__ method to Repomd() class
---

diff --git a/src/python/__init__.py b/src/python/__init__.py
index 38e7425..763dd9f 100644
--- 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
--- 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)