Enhance total commits calculation by incorporating commits from multiple emails and usernames. Implement a fallback to GraphQL for total commits if the search API fails. Update output messages for clarity.

Author: AxelPCG

github_stats.py

@@ -77,6 +77,11 @@ async def query_rest(self, path: str, params: Optional[Dict] = None) -> Dict:
             headers = {
                 "Authorization": f"token {self.access_token}",
             }
+            
+            # API de busca de commits requer header especial
+            if "/search/commits" in path:
+                headers["Accept"] = "application/vnd.github.cloak-preview+json"
+                
             if params is None:
                 params = dict()
             if path.startswith("/"):
@@ -773,25 +778,67 @@ async def forks_made(self) -> int:
 
     async def get_all_time_commits(self) -> None:
         """
-        Get total commits from all years
+        Get total commits from all time, including commits with different emails
         """
-        # Não verificar se _total_commits já existe, sempre recalcular
-        print("Fetching commits from all years...")
+        print("Fetching total commits...")
+        total_commits = 0
 
-        # Buscar commits de todos os anos
-        years = (
-            (await self.queries.query(Queries.contrib_years()))
-            .get("data", {})
-            .get("viewer", {})
-            .get("contributionsCollection", {})
-            .get("contributionYears", [])
-        )
+        # Se temos emails, buscar commits para cada email
+        if self._emails:
+            print(f"Using Git emails: {self._emails}")
+            for email in self._emails:
+                # Usar a API de busca para contar commits por email
+                search_url = f"/search/commits"
+                params = {
+                    "q": f"author-email:{email}",
+                    "per_page": 1
+                }
+                
+                result = await self.queries.query_rest(search_url, params)
+                
+                if result and "total_count" in result:
+                    email_commits = result["total_count"]
+                    total_commits += email_commits
+                    print(f"  Email {email}: {email_commits} commits")
 
-        total_commits = 0
+        # Também buscar por username
+        search_url = f"/search/commits"
+        params = {
+            "q": f"author:{self.username}",
+            "per_page": 1
+        }
 
-        # Para cada ano, buscar o total de commits
-        for year in years:
-            query = f"""
+        result = await self.queries.query_rest(search_url, params)
+        
+        if result and "total_count" in result:
+            username_commits = result["total_count"]
+            print(f"  Username {self.username}: {username_commits} commits")
+            
+            # Se não tínhamos emails, usar apenas commits por username
+            if not self._emails:
+                total_commits = username_commits
+            else:
+                # Se temos emails, adicionar commits por username que não foram contados
+                # (isso pode resultar em duplicatas, mas é melhor que perder commits)
+                if username_commits > total_commits:
+                    total_commits = username_commits
+        
+        # Se a busca falhou completamente, usar GraphQL como fallback
+        if total_commits == 0:
+            print("Search API failed or returned 0, using GraphQL fallback...")
+            
+            # Buscar commits de todos os anos via GraphQL
+            years = (
+                (await self.queries.query(Queries.contrib_years()))
+                .get("data", {})
+                .get("viewer", {})
+                .get("contributionsCollection", {})
+                .get("contributionYears", [])
+            )
+            
+            # Para cada ano, buscar o total de commits
+            for year in years:
+                query = f"""
 query {{
   viewer {{
     contributionsCollection(from: "{year}-01-01T00:00:00Z", to: "{int(year) + 1}-01-01T00:00:00Z") {{
@@ -801,18 +848,18 @@ async def get_all_time_commits(self) -> None:
   }}
 }}
 """
-            result = await self.queries.query(query)
-            if result:
-                contrib = result.get("data", {}).get("viewer", {}).get("contributionsCollection", {})
-                year_commits = (
-                    contrib.get("totalCommitContributions", 0) + 
-                    contrib.get("restrictedContributionsCount", 0)
-                )
-                total_commits += year_commits
-                print(f"Year {year}: {year_commits} commits")
+                result = await self.queries.query(query)
+                if result:
+                    contrib = result.get("data", {}).get("viewer", {}).get("contributionsCollection", {})
+                    year_commits = (
+                        contrib.get("totalCommitContributions", 0) + 
+                        contrib.get("restrictedContributionsCount", 0)
+                    )
+                    total_commits += year_commits
+                    print(f"  Year {year}: {year_commits} commits")
 
         self._total_commits = total_commits
-        print(f"Total commits from all years: {total_commits}")
+        print(f"Total commits from all sources: {total_commits}")
 
     @property
     async def total_forks(self) -> int: