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

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

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-20 14:05:26 -07:00
parent 71884e58d4
commit 0ac2a85a3e
1 changed files with 24 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

35
scripts/badges.py
View file

@ -1,10 +1,12 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import requests
import argparse import argparse
import json
import os import os
import sys import sys
import json
import requests
def get_total_downloads(api_key, package_name="aider-chat"): def get_total_downloads(api_key, package_name="aider-chat"):
""" """
@ -12,33 +14,44 @@ def get_total_downloads(api_key, package_name="aider-chat"):
""" """
url = f"https://api.pepy.tech/api/v2/projects/{package_name}" url = f"https://api.pepy.tech/api/v2/projects/{package_name}"
headers = {"X-API-Key": api_key} headers = {"X-API-Key": api_key}
try: try:
response = requests.get(url, headers=headers) response = requests.get(url, headers=headers)
response.raise_for_status() # Raise an exception for HTTP errors response.raise_for_status() # Raise an exception for HTTP errors
data = response.json() data = response.json()
total_downloads = data.get("total_downloads", 0) total_downloads = data.get("total_downloads", 0)
return total_downloads return total_downloads
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e:
print(f"Error fetching download statistics: {e}", file=sys.stderr) print(f"Error fetching download statistics: {e}", file=sys.stderr)
sys.exit(1) sys.exit(1)
def main(): def main():
parser = argparse.ArgumentParser(description="Get total downloads for a Python package from pepy.tech") parser = argparse.ArgumentParser(
parser.add_argument("--api-key", help="pepy.tech API key (or set PEPY_API_KEY environment variable)") description="Get total downloads for a Python package from pepy.tech"
parser.add_argument("--package", default="aider-chat", help="Package name (default: aider-chat)") )
parser.add_argument(
"--api-key", help="pepy.tech API key (or set PEPY_API_KEY environment variable)"
)
parser.add_argument(
"--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
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("API key not provided. Please set PEPY_API_KEY environment variable or use --api-key", file=sys.stderr) print(
"API key not provided. Please set PEPY_API_KEY environment variable or use --api-key",
file=sys.stderr,
)
sys.exit(1) sys.exit(1)
total_downloads = get_total_downloads(api_key, args.package) total_downloads = get_total_downloads(api_key, args.package)
print(f"Total downloads for {args.package}: {total_downloads:,}") print(f"Total downloads for {args.package}: {total_downloads:,}")
if __name__ == "__main__": if __name__ == "__main__":
main() main()