recce-nightly 1.2.0.20250506__py3-none-any.whl → 1.26.0.20251124__py3-none-any.whl

recce/github.py

@@ -3,10 +3,10 @@ import os
 import re
 import zipfile
 from datetime import datetime
-from typing import List, Tuple, Optional
+from typing import List, Optional, Tuple
 
 import requests
-from github import Artifact, Github, Auth, UnknownObjectException, PullRequest
+from github import Artifact, Auth, Github, PullRequest, UnknownObjectException
 
 from recce.git import current_branch, hosting_repo
 
@@ -16,7 +16,7 @@ def download_artifact(github_token: str, artifact: Artifact) -> List[str]:
     Download the artifact from the Github.
     """
     headers = {
-        'Authorization': f'Bearer {github_token}',
+        "Authorization": f"Bearer {github_token}",
     }
     r = requests.get(artifact.archive_download_url, headers=headers)
     r.raise_for_status()
@@ -27,17 +27,17 @@ def download_artifact(github_token: str, artifact: Artifact) -> List[str]:
 
 def recce_ci_artifact(**kwargs):
     """
-        Download the artifact from the GitHub CI.
+    Download the artifact from the GitHub CI.
     """
-    from rich.console import Console
-
     import git
+
     # Authentication is defined via github.Auth
     from github import Auth, Github
+    from rich.console import Console
 
     console = Console()
-    github_token = kwargs.get('github_token')
-    github_repository = kwargs.get('github_repository')
+    github_token = kwargs.get("github_token")
+    github_repository = kwargs.get("github_repository")
 
     if github_token is None:
         console.print("[[red]Error[/red]] Missing GitHub token. Please provide a GitHub token.")
@@ -70,11 +70,11 @@ def recce_ci_artifact(**kwargs):
 
     head_branch = github_repo.get_branch(current_branch)
 
-    console.rule('GitHub Repository Information')
-    console.print(f'Repository:     {github_repository}')
-    console.print(f'Current Branch: {current_branch}')
+    console.rule("GitHub Repository Information")
+    console.print(f"Repository:     {github_repository}")
+    console.print(f"Current Branch: {current_branch}")
 
-    pull_requests = github_repo.get_pulls(head=f'{github_owner}:{current_branch}')
+    pull_requests = github_repo.get_pulls(head=f"{github_owner}:{current_branch}")
     if pull_requests.totalCount == 0:
         # No pull request found for the current branch
         console.print("[[red]Error[/red]] No pull request found for the current branch.")
@@ -82,32 +82,32 @@ def recce_ci_artifact(**kwargs):
 
     pr = pull_requests[0]
 
-    console.rule('GitHub Pull Request Information')
-    console.print(f'{pr.title} - {pr.html_url}')
-    console.print(f'State: {pr.state.title()}')
-    console.print(f'Author: {pr.user.name}')
-    console.print(f'Created At: {pr.created_at}')
+    console.rule("GitHub Pull Request Information")
+    console.print(f"{pr.title} - {pr.html_url}")
+    console.print(f"State: {pr.state.title()}")
+    console.print(f"Author: {pr.user.name}")
+    console.print(f"Created At: {pr.created_at}")
 
     workflow_runs = github_repo.get_workflow_runs(
-        event='pull_request',
-        status='success',
+        event="pull_request",
+        status="success",
         branch=head_branch,
     )
     if workflow_runs.totalCount == 0:
         console.print("[[yellow]Skip[/yellow]] No successful workflow runs found.")
         return 0
     last_workflow_run = workflow_runs[0]
-    console.rule('GitHub Workflow Run Information')
-    console.print(f'Last Workflow Run: {last_workflow_run.name} {last_workflow_run.html_url}')
+    console.rule("GitHub Workflow Run Information")
+    console.print(f"Last Workflow Run: {last_workflow_run.name} {last_workflow_run.html_url}")
     artifacts = last_workflow_run.get_artifacts()
     if artifacts.totalCount == 0:
         console.print("[[yellow]Skip[/yellow]] No artifacts found.")
         return 0
     for artifact in artifacts:
-        console.print(f'Artifact: {artifact.name} {artifact.archive_download_url}')
+        console.print(f"Artifact: {artifact.name} {artifact.archive_download_url}")
         artifact_files = download_artifact(github_token, artifact)
         console.print(f'Extracted Files: {", ".join(artifact_files)}')
-    console.rule('Complete')
+    console.rule("Complete")
     return 0
 
 
@@ -121,7 +121,7 @@ def get_pull_request(branch, owner, repo_name, github_token=None) -> Tuple[Optio
 
     try:
         repo = g.get_repo(f"{owner}/{repo_name}")
-        pulls = repo.get_pulls(state='open')
+        pulls = repo.get_pulls(state="open")
 
         for pr in pulls:
             if pr.head.ref == branch:
@@ -129,7 +129,10 @@ def get_pull_request(branch, owner, repo_name, github_token=None) -> Tuple[Optio
 
     except UnknownObjectException:
         if github_token is not None:
-            return None, f"Repository {owner}/{repo_name} not found. If it is private repo, please add the 'repo' scope to the token."
+            return (
+                None,
+                f"Repository {owner}/{repo_name} not found. If it is private repo, please add the 'repo' scope to the token.",
+            )
 
     return None, None
 
@@ -139,53 +142,54 @@ def recce_pr_information(github_token=None) -> Tuple[Optional[type(PullRequest)]
     repo = hosting_repo()
 
     if not repo:
-        return None, 'This is not a git repository.'
-    if '/' not in repo:
-        return None, 'This is not a GitHub repository.'
+        return None, "This is not a git repository."
+    if "/" not in repo:
+        return None, "This is not a GitHub repository."
 
-    owner, repo_name = repo.split('/')
+    owner, repo_name = repo.split("/")
 
     github_token = github_token if github_token else os.getenv("GITHUB_TOKEN")
     return get_pull_request(branch, owner, repo_name, github_token)
 
 
 def is_github_codespace():
-    return os.getenv('CODESPACES') == 'true'
+    return os.getenv("CODESPACES") == "true"
 
 
 def get_github_codespace_name():
-    return os.getenv('CODESPACE_NAME')
+    return os.getenv("CODESPACE_NAME")
 
 
 def get_github_codespace_info():
     if is_github_codespace() is False:
         return None
 
-    codespace_name = os.environ.get('CODESPACE_NAME')
-    github_token = os.environ.get('GITHUB_TOKEN')
+    codespace_name = os.environ.get("CODESPACE_NAME")
+    github_token = os.environ.get("GITHUB_TOKEN")
 
     response = requests.get(
-        f'https://api.github.com/user/codespaces/{codespace_name}',
+        f"https://api.github.com/user/codespaces/{codespace_name}",
         headers={
-            'Accept': 'application/vnd.github+json',
-            'Authorization': f'token {github_token}',
-            'X-GitHub-Api-Version': '2022-11-28',
-        })
+            "Accept": "application/vnd.github+json",
+            "Authorization": f"token {github_token}",
+            "X-GitHub-Api-Version": "2022-11-28",
+        },
+    )
 
     if response.status_code != 200:
         return None
     codespace_info = response.json()
 
     return dict(
-        name=codespace_info.get('name'),
-        machine=codespace_info.get('machine'),
-        prebuild=codespace_info.get('prebuild'),
-        created_at=codespace_info.get('created_at'),
-        updated_at=codespace_info.get('updated_at'),
-        last_used_at=codespace_info.get('last_used_at'),
-        state=codespace_info.get('state'),
-        location=codespace_info.get('location'),
-        idle_timeout_minutes=codespace_info.get('idle_timeout_minutes'),
+        name=codespace_info.get("name"),
+        machine=codespace_info.get("machine"),
+        prebuild=codespace_info.get("prebuild"),
+        created_at=codespace_info.get("created_at"),
+        updated_at=codespace_info.get("updated_at"),
+        last_used_at=codespace_info.get("last_used_at"),
+        state=codespace_info.get("state"),
+        location=codespace_info.get("location"),
+        idle_timeout_minutes=codespace_info.get("idle_timeout_minutes"),
     )
 
 
@@ -194,25 +198,25 @@ def get_github_codespace_available_at(codespace):
         return None
 
     def search_in_file(file_path, search_string):
-        with open(file_path, 'r') as f:
+        with open(file_path, "r", encoding="utf-8") as f:
             for _, line in enumerate(f, 1):
                 if search_string in line:
                     return line
         return None
 
     def extract_datatime(log_line):
-        pattern = r'\[(\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{3}).*\]'
+        pattern = r"\[(\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{3}).*\]"
         match = re.search(pattern, log_line)
         if match:
             datetime_str = match.group(1)
-            return datetime.strptime(datetime_str, '%Y-%m-%d %H:%M:%S.%f')
+            return datetime.strptime(datetime_str, "%Y-%m-%d %H:%M:%S.%f")
         return None
 
-    github_codepsce_log_dir = '/tmp/codespaces_logs'
+    github_codepsce_log_dir = "/tmp/codespaces_logs"
     try:
         log_file = os.listdir(github_codepsce_log_dir)[-1]  # Get the latest log file
-        start_monitor_line = search_in_file(f'{github_codepsce_log_dir}/{log_file}', 'Starting monitor')
+        start_monitor_line = search_in_file(f"{github_codepsce_log_dir}/{log_file}", "Starting monitor")
         return extract_datatime(start_monitor_line)
     except Exception:
         # If there is any error, use the updated_at time from the codespace info
-        return datetime.fromisoformat(codespace.get('updated_at'))
+        return datetime.fromisoformat(codespace.get("updated_at"))