update rit github update repo formula to treat account with +100 repos

Signed-off-by: GuillaumeFalourd <[email protected]>

Author: GuillaumeFalourd

github/update/repo/src/formula/formula.py

@@ -7,58 +7,62 @@
 def run(token, username, default_branch, repos):
 
     url_repos = f"https://api.github.com/user/repos?type=owner&per_page=100&sort=full_name"
-
+    
     authorization = f"token {token}"
     headers = {
         "Accept": "application/vnd.github.v3+json",
         "Authorization" : authorization,
         }
 
-    r1 = requests.get(
-        url = url_repos,
-        headers = headers
-        )
+    all_repositories = get_repositories(url_repos, headers)
 
-    if r1.status_code == 200:
-        datas = r1.json()
-        all_repositories = []
+    if repos == "all":
+        repositories = all_repositories
+        
+    else: # I want to select
+        question1 = [
+            inquirer.Checkbox("repositories",
+                    message = f"\033[1m\033[36m{username}\033[0m \033[1mrepository to delete:\033[0m ",
+                    choices = all_repositories,
+                ),
+        ]
+        answer = inquirer.prompt(question1)
+        repositories = answer["repositories"]            
 
-        for d in datas:
-            all_repositories.append(d["name"])
+    for repository in repositories:
+        url_heads = f"https://api.github.com/repos/{username}/{repository}/git/refs/heads"
 
-        if repos == "all":
-            repos = all_repositories
-            
-        else: # I want to select
-            question1 = [
-                inquirer.Checkbox("repositories",
-                        message = f"\033[1m\033[36m{username}\033[0m \033[1mrepository to delete:\033[0m ",
-                        choices = all_repositories,
-                    ),
-            ]
-            answer = inquirer.prompt(question1)
-            repositories = answer["repositories"]            
-        
-        for repository in repositories:
-            url_heads = f"https://api.github.com/repos/{username}/{repository}/git/refs/heads"
+        r = requests.get(
+            url = url_heads,
+            headers = headers
+            )
 
-            r2 = requests.get(
-                url = url_heads,
-                headers = headers
-                )
-    
-            datas = r2.json()
-            branches = []
+        datas = r.json()
+        branches = []
+        
+        for data in datas:
+            branch_name = re.search("(?<=refs\/heads\/).*", data["ref"]).group()
+            branches.append(branch_name)
 
-            for data in datas:
-                branch_name = re.search("(?<=refs\/heads\/).*", data["ref"]).group()
-                branches.append(branch_name)
-                
-            if default_branch in branches:
-                print(f"\n🛠  Updating default branch for \033[36m{repository}\033[0m...")
-                input_flag_cmd = f"rit github update default-branch --rit_repo_owner=\"{username}\" --rit_git_repo=\"{repository}\" --rit_repo_branch=\"{default_branch}\""
-                os.system(f"{input_flag_cmd}")
-            else:
-                print(f"\n🛠  Creating new \033[36m{default_branch}\033[0m branch for \033[36m{repository}\033[0m...")
-                input_flag_cmd = f"rit github create branch --rit_repo_owner=\"{username}\" --rit_git_repo=\"{repository}\" --rit_repo_branch=\"{default_branch}\" --rit_branch_default=\"yes\""
-                os.system(f"{input_flag_cmd}")
+        if default_branch in branches:
+            print(f"\n🛠  Updating default branch for \033[36m{repository}\033[0m...")
+            input_flag_cmd = f"rit github update default-branch --rit_repo_owner=\"{username}\" --rit_git_repo=\"{repository}\" --rit_repo_branch=\"{default_branch}\""
+            os.system(f"{input_flag_cmd}")
+        else:
+            print(f"\n🛠  Creating new \033[36m{default_branch}\033[0m branch for \033[36m{repository}\033[0m...")
+            input_flag_cmd = f"rit github create branch --rit_repo_owner=\"{username}\" --rit_git_repo=\"{repository}\" --rit_repo_branch=\"{default_branch}\" --rit_branch_default=\"yes\""
+            os.system(f"{input_flag_cmd}")
+
+def get_repositories(url, headers):
+    result = []
+    r = requests.get(
+        url = url,
+        headers = headers
+        )
+    if "next" in r.links :
+        result += get_repositories(r.links["next"]["url"], headers)
+
+    for repository in r.json():
+        result.append(repository.get("name"))
+
+    return result