From d26df6c1782f564b202e0fa39d9d40280a979416 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <paul@aider.chat>
Date: Thu, 20 Mar 2025 14:18:41 -0700
Subject: [PATCH] feat: add get_badges_md() function to generate badges
 markdown

---
 scripts/badges.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/scripts/badges.py b/scripts/badges.py
index 167784ab5..19043913e 100755
--- a/scripts/badges.py
+++ b/scripts/badges.py
@@ -140,6 +140,35 @@ src="https://img.shields.io/badge/🔄%20Singularity-{aider_percent_rounded}%25-
     return markdown
 
 
+def get_badges_md():
+    """
+    Get all statistics and return the generated badges markdown
+    """
+    # Load environment variables from .env file
+    load_dotenv()
+    
+    # Get API key from environment variable
+    api_key = os.environ.get("PEPY_API_KEY")
+    if not api_key:
+        print(
+            "API key not provided. Please set PEPY_API_KEY environment variable",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+    
+    # Get PyPI downloads for the default package
+    total_downloads = get_total_downloads(api_key, "aider-chat")
+    
+    # Get GitHub stars for the default repo
+    stars = get_github_stars("paul-gauthier/aider")
+    
+    # Get Aider contribution percentage in latest release
+    percentage, _ = get_latest_release_aider_percentage()
+    
+    # Generate and return badges markdown
+    return generate_badges_md(total_downloads, stars, percentage)
+
+
 def main():
     # Load environment variables from .env file
     load_dotenv()