feat: Add .env support for PEPY_API_KEY in badges script

This commit is contained in:
Paul Gauthier (aider) 2025-03-20 14:06:16 -07:00
parent 5bf248164a
commit 4c7d99b26c

View file

@ -5,6 +5,7 @@ import os
import sys import sys
import requests import requests
from dotenv import load_dotenv
def get_total_downloads(api_key, package_name="aider-chat"): 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(): def main():
# Load environment variables from .env file
load_dotenv()
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description="Get total downloads for a Python package from pepy.tech" description="Get total downloads for a Python package from pepy.tech"
) )
parser.add_argument( 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( parser.add_argument(
"--package", default="aider-chat", help="Package name (default: aider-chat)" "--package", default="aider-chat", help="Package name (default: aider-chat)"
) )
args = parser.parse_args() 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") api_key = args.api_key or os.environ.get("PEPY_API_KEY")
if not api_key: if not api_key:
print( print(