Merge pull request #75 from whisk3y/master

Add helper functions to replace all or one ruleset

Author: whisk3y

dyn/tm/services/dsf.py

@@ -27,6 +27,7 @@
            'DSFResponsePool', 'DSFRuleset', 'DSFMonitorEndpoint', 'DSFMonitor',
            'TrafficDirector']
 
+
 def get_all_dsf_services():
     """:return: A ``list`` of :class:`TrafficDirector` Services"""
     uri = '/DSF/'
@@ -3370,6 +3371,48 @@ def remove_orphans(self):
         api_args = {'remove_orphans': 'Y'}
         self._update(api_args)
 
+    def replace_all_rulesets(self, rulesets):
+        """
+            This request will replace all rulesets with a new list of rulesets.
+            :param rulesets: a list of rulesets :class:DSFRuleset to be published to the service
+            Warning! This call takes extra time as it is several api calls.
+            """
+        if (type(rulesets) is list or type(rulesets) is tuple) and isinstance(rulesets[0], DSFRuleset):
+            old_rulesets = self.all_rulesets
+            for old_rule in old_rulesets:
+                old_rule.delete()
+            for new_rule in rulesets:
+                new_rule._dsf_ruleset_id = None
+                new_rule.create(self)
+        else:
+            raise Exception("rulesets parameter must be a list of DSFRuleset objects")
+
+    def replace_one_ruleset(self, ruleset):
+        """
+            This request will replace a single ruleset and maintain the order of the list.
+            :param ruleset: A single object of :class:DSFRuleset`
+            The replacement is keyed by the DSFRuleset label value
+            Warning! This call takes extra time as it is several api calls.
+            """
+        if isinstance(ruleset, DSFRuleset):
+            old_rulesets = self.all_rulesets
+            old_rule = None
+            rule_index = 0
+            for rule in old_rulesets:
+                if rule.label == ruleset.label:
+                    old_rule = rule
+                    rule_index += 1
+                    break
+                rule_index += 1
+            if old_rule is not None:
+                old_rule.delete()
+                ruleset._dsf_ruleset_id = None
+                ruleset.create(self, index=rule_index)
+            else:
+                ruleset.create(self)
+        else:
+            raise Exception("rulesets parameter must be a single object of class DSFRuleset")
+
     @property
     def service_id(self):
         """The unique System id of this DSF Service"""