reviewly/main.py at main · MambaCodes/reviewly · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
"""Main entry point for reviewly."""
from rich import print
from reviewly.config import get_config
from reviewly.github_client import GitHubClient
from reviewly.analyzer import DeepseekAnalyzer

def main():
    """Main function."""
    # Load configuration
    config = get_config()

    # Initialize clients
    github_client = GitHubClient(config.github_token, config.github_repo_url, cache_dir="./cache")
    analyzer = DeepseekAnalyzer(config.deepseek_api_key)

    try:
        # Fetch PR reviews
        print("[bold blue]Fetching PR reviews...[/bold blue]")
        reviews = github_client.get_pr_reviews(limit=50)  # Get last 50 PRs

        if not reviews:
            print("[bold yellow]No PR reviews found.[/bold yellow]")
            return

        print(f"[green]Found {len(reviews)} review comments[/green]")

        # Analyze reviews
        print("[bold blue]Analyzing reviews with DeepSeek...[/bold blue]")
        checklist = analyzer.analyze_reviews(reviews)

        # Save results
        output_file = "pr_checklist.md"
        with open(output_file, "w", encoding="utf-8") as f:
            f.write("# PR Review Checklist\n\n")
            f.write("This checklist was automatically generated based on recent PR reviews.\n\n")
            f.write(checklist)

        print(f"[bold green]✓ Checklist saved to {output_file}[/bold green]")

    except Exception as e:
        print(f"[bold red]Error: {str(e)}[/bold red]")

if __name__ == "__main__":
    main()