Merge pull request #8 from CanarysAutomations/grenston-branch

multi branch support added

Author: Grenston George

automate-branch-rules.py

@@ -1,6 +1,6 @@
 from github import Github
 from stdiomask import getpass
-from config import branch_name
+from config import branches
 from config import branch_rules
 from config import signed_commit
 from config import add_codeowners_file
@@ -9,48 +9,52 @@
 def add_all(pat):
     print("")
     for repo in git.get_organization(org_name).get_repos():
+        for branch_name in branches:
+            branch = repo.get_branch(branch_name)
+            if(add_codeowners_file):
+               codeowners.add(org_name,pat,repo.name,branch_name)
+            branch.edit_protection(**branch_rules)
+            if(signed_commit):
+                branch.add_required_signatures()
+            else:
+                branch.remove_required_signatures()
+            print("Edited the branch protection rules for: " + repo.name + "," + branch_name)
+
+def add_one(pat):
+    repo_name = input("\nRepository: ")
+    repo = git.get_repo(org_name+"/"+repo_name)
+    for branch_name in branches:
         branch = repo.get_branch(branch_name)
         if(add_codeowners_file):
-            codeowners.add(org_name,pat,repo.name,branch_name)
+            codeowners.add(org_name,pat,repo_name,branch_name)
         branch.edit_protection(**branch_rules)
         if(signed_commit):
             branch.add_required_signatures()
         else:
             branch.remove_required_signatures()
-        print("Edited the branch protection rules for: " + repo.name)
-
-def add_one(pat):
-    repo_name = input("\nRepository: ")
-    repo = git.get_repo(org_name+"/"+repo_name)
-    branch = repo.get_branch(branch_name)
-    if(add_codeowners_file):
-        codeowners.add(org_name,pat,repo_name,branch_name)
-    branch.edit_protection(**branch_rules)
-    if(signed_commit):
-        branch.add_required_signatures()
-    else:
-        branch.remove_required_signatures()
-    print("Edited the branch protection rules for: " + repo.name)  
+        print("Edited the branch protection rules for: " + repo.name + "," + branch_name)  
 
 def remove_one():
     repo_name = input("\nRepository: ")
     repo = git.get_repo(org_name+"/"+repo_name)
-    branch = repo.get_branch(branch_name)
-    if (branch.protected):
-        branch.remove_protection()
-        print("Removed branch protection rules for: " + repo.name)
-    else:
-        print("No branch protection rules for: " + repo.name + "," + branch.name)    
-
-def remove_all():
-    print("")
-    for repo in git.get_organization(org_name).get_repos():
+    for branch_name in branches:
         branch = repo.get_branch(branch_name)
         if (branch.protected):
             branch.remove_protection()
-            print("Removed branch protection rules for: " + repo.name)
+            print("Removed branch protection rules for: " + repo.name + "," + branch_name)
         else:
-            print("No branch protection rules for: " + repo.name + "," + branch.name)
+            print("No branch protection rules for: " + repo.name + "," + branch.name)    
+
+def remove_all():
+    print("")
+    for repo in git.get_organization(org_name).get_repos():
+        for branch_name in branches:
+            branch = repo.get_branch(branch_name)
+            if (branch.protected):
+                branch.remove_protection()
+                print("Removed branch protection rules for: " + repo.name + "," + branch_name)
+            else:
+                print("No branch protection rules for: " + repo.name + "," + branch.name)
 
 print("   ___        _                        _         ______                      _       ______      _           ")
 print("  / _ \      | |                      | |        | ___ \                    | |      | ___ \    | |          ")

config.py

@@ -1,4 +1,4 @@
-branch_name = "master"
+branches = ("master","dev")
 add_codeowners_file = False
 signed_commit = False
 branch_rules = { "required_approving_review_count":1,