diff --git a/stats_pull_requests.py b/stats_pull_requests.py index 6d53fc1..555facd 100644 --- a/stats_pull_requests.py +++ b/stats_pull_requests.py @@ -1,28 +1,15 @@ -import os, logging +import os from datetime import datetime import pandas as pd from tqdm import tqdm from github import Github -from utils import has_only_1_round_of_comments, has_only_1_comment +from utils import has_only_1_round_of_comments, has_only_1_comment, move_github_logging_to_file tqdm.pandas() # Initialize GitHub API client g = Github(os.environ["GITHUB_AUTH_TOKEN_CRAB"]) -def move_github_logging_to_file(): - github_logger = logging.getLogger("github") - - # Remove existing handlers to prevent duplicate logging - for handler in github_logger.handlers[:]: - github_logger.removeHandler(handler) - - file_handler = logging.FileHandler("github_api.log") # Log to file - formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s") - file_handler.setFormatter(formatter) - github_logger.addHandler(file_handler) - github_logger.propagate = False # Prevent logging to standard output - def process_pull(repo, pull): commits = pull.get_commits() comments = pull.get_review_comments() diff --git a/utils.py b/utils.py index 28685fe..1ef5890 100644 --- a/utils.py +++ b/utils.py @@ -3,6 +3,20 @@ from github.Commit import Commit from github.PaginatedList import PaginatedList from github.PullRequestComment import PullRequestComment from tqdm import tqdm +import logging + +def move_github_logging_to_file(): + github_logger = logging.getLogger("github") + + # Remove existing handlers to prevent duplicate logging + for handler in github_logger.handlers[:]: + github_logger.removeHandler(handler) + + file_handler = logging.FileHandler("github_api.log") # Log to file + formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s") + file_handler.setFormatter(formatter) + github_logger.addHandler(file_handler) + github_logger.propagate = False # Prevent logging to standard output def parse_date(date: str) -> datetime: return datetime.strptime(date, "%Y-%m-%dT%H:%M:%SZ")