Enhancement: Rule manager method to get rules by list of keys

Author: stclair

dynamic_rules/models.py

@@ -20,6 +20,9 @@ def get_by_group_object(self, obj):
     def get_by_key(self, group_object, key):
         return self.get_by_group_object(group_object).filter(key=key)
 
+    def get_by_keys(self, group_object, keys):
+        return self.get_by_group_object(group_object).filter(key__in=keys)
+
 
 class Rule(models.Model):
     content_type = models.ForeignKey('contenttypes.ContentType')

dynamic_rules/tests/test_models.py

@@ -53,6 +53,17 @@ def test_get_by_key_returns_get_by_group_object_filtered_by_key(self):
         group_obj_query.filter.assert_called_once_with(key=key)
         self.assertEqual(group_obj_query.filter.return_value, rules)
 
+    def test_get_by_keys_returns_get_by_group_object_filtered_by_list_of_keys(self):
+        manager = mock.Mock(spec_set=models.RuleManager)
+        group_obj_query = manager.get_by_group_object.return_value
+        keys = ["abc", "def"]
+
+        rules = models.RuleManager.get_by_keys(manager, self.model_one, keys)
+
+        manager.get_by_group_object.assert_called_once_with(self.model_one)
+        group_obj_query.filter.assert_called_once_with(key__in=keys)
+        self.assertEqual(group_obj_query.filter.return_value, rules)
+
 
 class RuleModelTests(unittest.TestCase):
 

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name="django-dynamic-rules",
-    version="1.0.1",
+    version="1.0.2",
     author="IMT Computer Services",
     author_email="[email protected]",
     description="Allows you to create dynamic rules related to a particular model",