style: Format code and fix import order in badges.py

This commit is contained in:
Paul Gauthier (aider) 2025-03-20 14:10:43 -07:00
parent a5614bfe40
commit 7942ca9206

View file

@ -3,9 +3,9 @@
import argparse import argparse
import os import os
import sys import sys
import yaml
import requests import requests
import yaml
from dotenv import load_dotenv from dotenv import load_dotenv
@ -56,24 +56,27 @@ def get_aider_percentage():
""" """
blame_path = os.path.join( blame_path = os.path.join(
os.path.dirname(os.path.dirname(os.path.abspath(__file__))), os.path.dirname(os.path.dirname(os.path.abspath(__file__))),
"aider", "website", "_data", "blame.yml" "aider",
"website",
"_data",
"blame.yml",
) )
try: try:
with open(blame_path, 'r') as f: with open(blame_path, "r") as f:
blame_data = yaml.safe_load(f) blame_data = yaml.safe_load(f)
# Get the most recent release (first entry in the list) # Get the most recent release (first entry in the list)
if blame_data and len(blame_data) > 0: if blame_data and len(blame_data) > 0:
latest_release = blame_data[0] latest_release = blame_data[0]
percentage = latest_release.get('aider_percentage', 0) percentage = latest_release.get("aider_percentage", 0)
version = latest_release.get('end_tag', 'unknown') version = latest_release.get("end_tag", "unknown")
return percentage, version return percentage, version
return 0, 'unknown' return 0, "unknown"
except Exception as e: except Exception as e:
print(f"Error reading blame data: {e}", file=sys.stderr) print(f"Error reading blame data: {e}", file=sys.stderr)
return 0, 'unknown' return 0, "unknown"
def main(): def main():
@ -115,7 +118,7 @@ def main():
stars = get_github_stars(args.github_repo) stars = get_github_stars(args.github_repo)
if stars is not None: if stars is not None:
print(f"GitHub stars for {args.github_repo}: {stars:,}") print(f"GitHub stars for {args.github_repo}: {stars:,}")
# Get Aider contribution percentage in latest release # Get Aider contribution percentage in latest release
percentage, version = get_aider_percentage() percentage, version = get_aider_percentage()
print(f"Aider wrote {percentage:.2f}% of code in the latest release ({version})") print(f"Aider wrote {percentage:.2f}% of code in the latest release ({version})")