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

feat: Add .env support for PEPY_API_KEY in badges script

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-20 14:06:16 -07:00
parent 5bf248164a
commit 4c7d99b26c
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
scripts/badges.py
View file

@ -5,6 +5,7 @@ import os
import sys
import requests
from dotenv import load_dotenv
def get_total_downloads(api_key, package_name="aider-chat"):
@ -28,18 +29,21 @@ def get_total_downloads(api_key, package_name="aider-chat"):
def main():
# Load environment variables from .env file
load_dotenv()
parser = argparse.ArgumentParser(
description="Get total downloads for a Python package from pepy.tech"
)
parser.add_argument(
"--api-key", help="pepy.tech API key (or set PEPY_API_KEY environment variable)"
"--api-key", help="pepy.tech API key (can also be set via PEPY_API_KEY in .env file or environment variable)"
)
parser.add_argument(
"--package", default="aider-chat", help="Package name (default: aider-chat)"
)
args = parser.parse_args()
# Get API key from args or environment variable
# Get API key from args or environment variable (which may be loaded from .env)
api_key = args.api_key or os.environ.get("PEPY_API_KEY")
if not api_key:
print(