generate.py

import argparse

from svg_wheel import generate_svg_wheel
from utils import (
    annotate_support,
    get_top_packages,
    remove_irrelevant_packages,
    save_to_file,
)


def main(to_chart=360, versions=["2.6"]):
    packages = remove_irrelevant_packages(get_top_packages(), to_chart)
    annotate_support(packages, versions)
    save_to_file(packages, "results.json")
    generate_svg_wheel(packages, to_chart, versions)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Generate", formatter_class=argparse.ArgumentDefaultsHelpFormatter
    )
    parser.add_argument(
        "-n", "--number", default=360, type=int, help="Number of packages to chart"
    )
    parser.add_argument(
        "-v",
        "--version",
        default=["2.6", "3.2", "3.3"],
        nargs="+",
        help="Python version or versions to check",
    )
    args = parser.parse_args()

    main(args.number, args.version)