Update functions.py

functions.py

@@ -54,3 +54,57 @@ def get_repo_branches(repo_name, updated_token=None):
     except requests.exceptions.RequestException as e:
         print(f"Failed to fetch branches: {e}")
         return []  # Return an empty list in case of failure
+# Function to calculate the start date based on the period
+def get_start_date(period):
+    if period == "month":
+        return datetime.now() - timedelta(days=30)
+    elif period == "3 months":
+        return datetime.now() - timedelta(days=30*3)
+    elif period == "6 months":
+        return datetime.now() - timedelta(days=30*6)
+    elif period == "year":
+        return datetime.now() - timedelta(days=365)
+    else:
+        raise ValueError("Invalid period specified")
+
+# Function to fetch commits by a contributor in a given period
+def get_contributor_commits(repo_name, contributor, period, updated_token=None):
+    """
+    Fetch all commits made by a specific contributor within a specified period for a given repository.
+    
+    Parameters:
+    - repo_name: str - The full name of the repository (e.g., "owner/repo").
+    - contributor: str - The GitHub username of the contributor.
+    - period: str - The period for which to fetch commits ("month", "3 months", "6 months", "year").
+    - updated_token: str - Optional. A GitHub token for authentication.
+    
+    Returns:
+    - A list of commits made by the specified contributor in the given period.
+    """
+    headers = get_headers(updated_token)
+    start_date = get_start_date(period)
+    commits = []
+
+    # Assuming the repository might have multiple branches, we start by fetching branches
+    branches_url = f"{GITHUB_API}/repos/{repo_name}/branches"
+    try:
+        branches_response = requests.get(branches_url, headers=headers)
+        branches_response.raise_for_status()
+        branches = [branch['name'] for branch in branches_response.json()]
+        
+        # Fetch commits for each branch
+        for branch in branches:
+            commits_url = f"{GITHUB_API}/repos/{repo_name}/commits"
+            params = {
+                "sha": branch,
+                "since": start_date.isoformat(),
+                "author": contributor
+            }
+            commits_response = requests.get(commits_url, headers=headers, params=params)
+            commits_response.raise_for_status()
+            branch_commits = commits_response.json()
+            commits.extend(branch_commits)
+    except requests.exceptions.RequestException as e:
+        print(f"Failed to fetch commits: {e}")
+    
+    return commits