dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 00:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Refactor main() to use blame() function and print stats

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-07-29 10:58:35 -03:00
parent 99c6e88e4a
commit 17cba2500e
1 changed files with 27 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

50
scripts/blame.py
View file

@ -2,16 +2,39 @@
import subprocess import subprocess
import sys import sys
import tempfile
import argparse import argparse
from collections import defaultdict from collections import defaultdict
from pathlib import Path
from aider.dump import dump from aider.dump import dump
import os
from operator import itemgetter from operator import itemgetter
import re import re
def blame(start_tag, end_tag=None):
commits = get_all_commit_hashes_between_tags(start_tag, end_tag)
commits = [commit[:hash_len] for commit in commits]
authors = get_commit_authors(commits)
py_files = run(['git', 'ls-files', '*.py']).strip().split('\n')
all_file_counts = {}
grand_total = defaultdict(int)
aider_total = 0
for file in py_files:
file_counts = get_counts_for_file(start_tag, end_tag, authors, file)
if file_counts:
all_file_counts[file] = file_counts
for author, count in file_counts.items():
grand_total[author] += count
if "(aider)" in author.lower():
aider_total += count
total_lines = sum(grand_total.values())
aider_percentage = (aider_total / total_lines) * 100 if total_lines > 0 else 0
return all_file_counts, grand_total, total_lines, aider_total, aider_percentage
def get_all_commit_hashes_between_tags(start_tag, end_tag=None): def get_all_commit_hashes_between_tags(start_tag, end_tag=None):
if end_tag: if end_tag:
res = run(["git", "rev-list", f"{start_tag}..{end_tag}"]) res = run(["git", "rev-list", f"{start_tag}..{end_tag}"])
@ -52,34 +75,15 @@ def main():
parser.add_argument("--end-tag", help="The tag to end at (default: HEAD)", default=None) parser.add_argument("--end-tag", help="The tag to end at (default: HEAD)", default=None)
args = parser.parse_args() args = parser.parse_args()
commits = get_all_commit_hashes_between_tags(args.start_tag, args.end_tag) all_file_counts, grand_total, total_lines, aider_total, aider_percentage = blame(args.start_tag, args.end_tag)
commits = [commit[:hash_len] for commit in commits]
authors = get_commit_authors(commits)
py_files = run(['git', 'ls-files', '*.py']).strip().split('\n')
all_file_counts = {}
grand_total = defaultdict(int)
aider_total = 0
for file in py_files:
file_counts = get_counts_for_file(args.start_tag, args.end_tag, authors, file)
if file_counts:
all_file_counts[file] = file_counts
for author, count in file_counts.items():
grand_total[author] += count
if "(aider)" in author.lower():
aider_total += count
dump(all_file_counts) dump(all_file_counts)
print("\nGrand Total:") print("\nGrand Total:")
total_lines = sum(grand_total.values())
for author, count in sorted(grand_total.items(), key=itemgetter(1), reverse=True): for author, count in sorted(grand_total.items(), key=itemgetter(1), reverse=True):
percentage = (count / total_lines) * 100 percentage = (count / total_lines) * 100
print(f"- {author}: {count} lines ({percentage:.2f}%)") print(f"- {author}: {count} lines ({percentage:.2f}%)")
aider_percentage = (aider_total / total_lines) * 100 if total_lines > 0 else 0
print(f"\nAider wrote {aider_percentage:.0f}% of the code in this release ({aider_total}/{total_lines} lines).") print(f"\nAider wrote {aider_percentage:.0f}% of the code in this release ({aider_total}/{total_lines} lines).")
def get_counts_for_file(start_tag, end_tag, authors, fname): def get_counts_for_file(start_tag, end_tag, authors, fname):