add-metadata.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Preparing to run it:
#   brew install pipenv   # or other installation method
#   pipenv install
#   generate a personal access token at https://github.com/settings/tokens

# Running it:
#   GITHUB_TOKEN=xxxxxxx pipenv run python add-metadata.py < template.md > README.md

import fileinput
import os
import re
import sys

from datetime import datetime, timedelta
from github import Github

g = Github(os.environ['GITHUB_TOKEN'])

gh_repo_regex = re.compile('\[([\w\._-]+\/[\w\._-]+)\]\(@ghRepo\)')

def github_table_row(repo):
    name = repo.name
    if repo.stargazers_count >= 500:
        name = f"**{name}**"

    project_link = f"[{name}]({repo.html_url})"
    stars_shield = f"![GitHub stars](https://img.shields.io/github/stars/{repo.full_name})"
    commit_shield = f"![GitHub commit activity](https://img.shields.io/github/commit-activity/y/{repo.full_name})"

    return f"{project_link} | {repo.description} | {stars_shield} {commit_shield}"


def warn(msg):
    print(f"Warn: {msg}", file=sys.stderr)


def retrieve_repo(name):
    try:
        repo = g.get_repo(name)
    except Exception:
        warn(f"Error occured while getting {name} repo")
        raise
        
    print('.', file=sys.stderr, end='', flush=True)
    
    if is_stale(repo):
        raise Exception("Repo is too old or inactive")

    return repo


def is_stale(repo):
    if repo.archived:
        warn(f"Repo {repo.full_name} is archived")
        return True
    elif repo.pushed_at < datetime.utcnow() - timedelta(days=365):
        warn(f"Repo {repo.full_name} has not been pushed to since {repo.pushed_at}")
        return True

    return False


def parse(line):
    m = gh_repo_regex.search(line)
    if m:
        [repo_name] = m.groups()
        try:
            row = github_table_row(retrieve_repo(repo_name))
        except Exception: 
            row = "skip"
        return row
    else:
        return line.rstrip()


def run():
    print('<!--- This file is automatically generated. Do not edit directly. -->')
    for line in fileinput.input():
        parsed = parse(line)
        if parsed != "skip":
            print(parsed)


run()