mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-22 21:34:59 +00:00
feat: Add .env support for PEPY_API_KEY in badges script
This commit is contained in:
parent
5bf248164a
commit
4c7d99b26c
1 changed files with 6 additions and 2 deletions
|
@ -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(
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue