exception handled for branch not found

Author: grenston

automate-branch-rules.py

@@ -1,4 +1,6 @@
 from github import Github
+from github import GithubException
+import sys
 from stdiomask import getpass
 from config import branches
 from config import branch_rules
@@ -10,51 +12,71 @@ 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()
+            try:
+                repo.get_branch(branch_name)
+            except GithubException:
+                print("Error:",repo.name,",",branch_name,"-->",sys.exc_info()[1])
             else:
-                branch.remove_required_signatures()
-            print("Edited the branch protection rules for: " + repo.name + "," + branch_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 + "," + 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)
-        branch.edit_protection(**branch_rules)
-        if(signed_commit):
-            branch.add_required_signatures()
+        try:
+            repo.get_branch(branch_name)
+        except GithubException:
+            print("Error:",repo.name,",",branch_name,"-->",sys.exc_info()[1])
         else:
-            branch.remove_required_signatures()
-        print("Edited the branch protection rules for: " + repo.name + "," + branch_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 + "," + branch_name)  
 
 def remove_one():
     repo_name = input("\nRepository: ")
     repo = git.get_repo(org_name+"/"+repo_name)
     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)
+        try:
+            repo.get_branch(branch_name)
+        except GithubException:
+            print("Error:",repo.name,",",branch_name,"-->",sys.exc_info()[1])
         else:
-            print("No branch protection rules for: " + repo.name + "," + branch.name)    
+            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)    
 
 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)
+            try:
+                repo.get_branch(branch_name)
+            except GithubException:
+                print("Error:",repo.name,",",branch_name,"-->",sys.exc_info()[1])
             else:
-                print("No branch protection rules for: " + repo.name + "," + branch.name)
+                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,8 +1,8 @@
-branches = ("master","dev")
-add_codeowners_file = False
+branches = ("dev","master")
+add_codeowners_file = True
 signed_commit = False
 branch_rules = { "required_approving_review_count":1,
                  "require_code_owner_reviews":True,
                  "contexts":["CodeQL"],
                  "strict":True
-                 }
+                 }