| #!/usr/bin/env python3 |
| |
| # Copyright 2022 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| """Creates an HTML page for releases for `pip install --find-links` from GitHub releases.""" |
| # TODO(#10479) since we're generating this we might as well create a PEP 503 |
| # compliant index |
| import argparse |
| import html |
| import json |
| import subprocess |
| import sys |
| import textwrap |
| |
| import requests |
| |
| |
| def parse_arguments(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument("--repo", |
| "--repository", |
| default="openxla/iree", |
| help="The GitHub repository to fetch releases from.") |
| parser.add_argument( |
| "--output", |
| default="-", |
| help="The file to write the HTML to or '-' for stdout (the default)") |
| return parser.parse_args() |
| |
| |
| class ReleaseFetcher: |
| |
| def __init__(self, repo, per_page=100): |
| self._session = requests.Session() |
| self._repo = repo |
| self._per_page = per_page |
| |
| def get_all(self): |
| url = f"https://api.github.com/repos/{self._repo}/releases" |
| page = 1 |
| |
| while True: |
| response = self._session.get(url, |
| params={ |
| "page": page, |
| "per_page": self._per_page, |
| }) |
| for release in response.json(): |
| yield release |
| if "next" not in response.links: |
| break |
| page += 1 |
| |
| |
| def main(args): |
| fetcher = ReleaseFetcher(repo=args.repo) |
| with (sys.stdout if args.output == "-" else open(args.output, "w")) as f: |
| f.write( |
| textwrap.dedent("""\ |
| <!DOCTYPE html> |
| <html> |
| <body> |
| """)) |
| for release in fetcher.get_all(): |
| if release["draft"]: |
| continue |
| for asset in release["assets"]: |
| url = html.escape(asset['browser_download_url']) |
| name = html.escape(asset['name']) |
| f.write(f" <a href={url}>{name}</a><br />\n") |
| f.write(textwrap.dedent("""\ |
| </body> |
| </html> |
| """)) |
| |
| |
| if __name__ == "__main__": |
| main(parse_arguments()) |
