Get GitHub repo from environment

This commit is contained in:
Bruno Alla 2021-11-09 10:32:08 +00:00
parent 5e906a6636
commit e2c9808c7e
2 changed files with 14 additions and 4 deletions

View File

@ -7,6 +7,7 @@ import datetime as dt
CURRENT_FILE = Path(__file__) CURRENT_FILE = Path(__file__)
ROOT = CURRENT_FILE.parents[1] ROOT = CURRENT_FILE.parents[1]
GITHUB_TOKEN = os.getenv("GITHUB_TOKEN", None) GITHUB_TOKEN = os.getenv("GITHUB_TOKEN", None)
GITHUB_REPO = os.getenv("GITHUB_REPOSITORY", None)
# Generate changelog for PRs merged yesterday # Generate changelog for PRs merged yesterday
MERGED_DATE = dt.date.today() - dt.timedelta(days=1) MERGED_DATE = dt.date.today() - dt.timedelta(days=1)
@ -39,9 +40,7 @@ def main() -> None:
def iter_pulls(): def iter_pulls():
"""Fetch merged pull requests at the date we're interested in.""" """Fetch merged pull requests at the date we're interested in."""
repo = Github(login_or_token=GITHUB_TOKEN).get_repo( repo = Github(login_or_token=GITHUB_TOKEN).get_repo(GITHUB_REPO)
"cookiecutter/cookiecutter-django"
)
recent_pulls = repo.get_pulls( recent_pulls = repo.get_pulls(
state="closed", sort="updated", direction="desc" state="closed", sort="updated", direction="desc"
).get_page(0) ).get_page(0)
@ -77,4 +76,8 @@ def generate_md(grouped_pulls):
if __name__ == "__main__": if __name__ == "__main__":
if GITHUB_REPO is None:
raise RuntimeError(
"No github repo, please set the environment variable GITHUB_REPOSITORY"
)
main() main()

View File

@ -1,4 +1,5 @@
import json import json
import os
from pathlib import Path from pathlib import Path
from github import Github from github import Github
from github.NamedUser import NamedUser from github.NamedUser import NamedUser
@ -7,6 +8,8 @@ from jinja2 import Template
CURRENT_FILE = Path(__file__) CURRENT_FILE = Path(__file__)
ROOT = CURRENT_FILE.parents[1] ROOT = CURRENT_FILE.parents[1]
BOT_LOGINS = ["pyup-bot"] BOT_LOGINS = ["pyup-bot"]
GITHUB_TOKEN = os.getenv("GITHUB_TOKEN", None)
GITHUB_REPO = os.getenv("GITHUB_REPOSITORY", None)
def main() -> None: def main() -> None:
@ -39,7 +42,7 @@ def iter_recent_authors():
Use Github API to fetch recent authors rather than Use Github API to fetch recent authors rather than
git CLI to work with Github usernames. git CLI to work with Github usernames.
""" """
repo = Github(per_page=5).get_repo("cookiecutter/cookiecutter-django") repo = Github(login_or_token=GITHUB_TOKEN, per_page=5).get_repo(GITHUB_REPO)
recent_pulls = repo.get_pulls( recent_pulls = repo.get_pulls(
state="closed", sort="updated", direction="desc" state="closed", sort="updated", direction="desc"
).get_page(0) ).get_page(0)
@ -101,4 +104,8 @@ def write_md_file(contributors):
if __name__ == "__main__": if __name__ == "__main__":
if GITHUB_REPO is None:
raise RuntimeError(
"No github repo, please set the environment variable GITHUB_REPOSITORY"
)
main() main()