Update Branch.protect such that its arguments are optional.

bboe · bboe · commit 1b7ea5a96865 · 2015-11-18T21:35:47.000-08:00
diff --git a/github3/repos/branch.py b/github3/repos/branch.py
@@ -32,17 +32,24 @@ def _update_attributes(self, branch):
     def _repr(self):
         return '<Repository Branch [{0}]>'.format(self.name)
 
-    def protect(self, enforcement, status_checks):
+    def protect(self, enforcement=None, status_checks=None):
         """Enable force push protection and configure status check enforcement.
 
         See: http://git.io/v4Gvu
 
-        :param str enforcement: (required), Specifies the enforcement level of
+        :param str enforcement: (optional), Specifies the enforcement level of
             the status checks. Must be one of 'off', 'non_admins', or
-            'everyone'.
-        :param list status_checks: (required), An iterable of strings naming
-            status checks that must pass before merging.
+            'everyone'. Use `None` or omit to use the already associated value.
+        :param list status_checks: (optional), An list of strings naming
+            status checks that must pass before merging. Use `None` or omit to
+            use the already associated value.
         """
+        previous_values = self.protection['required_status_checks']
+        if enforcement is None:
+            enforcement = previous_values['enforcement_level']
+        if status_checks is None:
+            status_checks = previous_values['contexts']
+
         edit = {'protection': {'enabled': True, 'required_status_checks': {
             'enforcement_level': enforcement, 'contexts': status_checks}}}
         json = self._json(self._patch(self.links['self'], data=dumps(edit),
