Update functions.py

functions.py

@@ -19,7 +19,7 @@ GITHUB_API = "https://api.github.com"
 # Function to update headers with a new token
 def get_headers(updated_token=None):
     # Use the provided token or fallback to the existing secret token
-    token = updated_token if updated_token else secrets['GITHUB_API_TOKEN']
+    token = updated_token if updated_token!="" else secrets['GITHUB_API_TOKEN']
     return {
         "Authorization": f"token {token}",
         "Accept": "application/vnd.github.v3+json"
@@ -32,4 +32,25 @@ def get_contributors(repo, updated_token=None):
     response = requests.get(url, headers=headers)
     return [contributor['login'] for contributor in response.json()]
 
-
+# Function to get repository branches
+def get_repo_branches(repo_name, updated_token=None):
+    """
+    Fetch and return a list of branches from a GitHub repository.
+    
+    Parameters:
+    - repo_name: str - The full name of the repository (e.g., "owner/repo").
+    - updated_token: str - Optional. A GitHub token for authentication.
+    
+    Returns:
+    - A list of branch names (str) of the specified repository.
+    """
+    headers = get_headers(updated_token)
+    url = f"{GITHUB_API}/repos/{repo_name}/branches"
+    try:
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()  # Raises an HTTPError if the response status code is 4XX or 5XX
+        branches = [branch['name'] for branch in response.json()]
+        return branches
+    except requests.exceptions.RequestException as e:
+        print(f"Failed to fetch branches: {e}")
+        return []  # Return an empty list in case of failure