| Geoffrey Martin-Noble | 7d53e56 | 2022-09-28 15:43:04 -0700 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
| 2 | |
| 3 | # Copyright 2022 The IREE Authors |
| 4 | # |
| 5 | # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| 6 | # See https://llvm.org/LICENSE.txt for license information. |
| 7 | # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 8 | """Creates an HTML page for releases for `pip install --find-links` from GitHub releases.""" |
| 9 | # TODO(#10479) since we're generating this we might as well create a PEP 503 |
| 10 | # compliant index |
| 11 | import argparse |
| 12 | import html |
| 13 | import json |
| 14 | import subprocess |
| 15 | import sys |
| 16 | import textwrap |
| 17 | |
| 18 | import requests |
| 19 | |
| 20 | |
| 21 | def parse_arguments(): |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 22 | parser = argparse.ArgumentParser() |
| 23 | parser.add_argument( |
| 24 | "--repo", |
| 25 | "--repository", |
| Scott Todd | 3f51a55 | 2024-04-19 11:00:27 -0700 | [diff] [blame] | 26 | default="iree-org/iree", |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 27 | help="The GitHub repository to fetch releases from.", |
| 28 | ) |
| 29 | parser.add_argument( |
| 30 | "--output", |
| 31 | default="-", |
| 32 | help="The file to write the HTML to or '-' for stdout (the default)", |
| 33 | ) |
| 34 | return parser.parse_args() |
| Geoffrey Martin-Noble | 7d53e56 | 2022-09-28 15:43:04 -0700 | [diff] [blame] | 35 | |
| 36 | |
| 37 | class ReleaseFetcher: |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 38 | def __init__(self, repo, per_page=100): |
| 39 | self._session = requests.Session() |
| 40 | self._repo = repo |
| 41 | self._per_page = per_page |
| Geoffrey Martin-Noble | 7d53e56 | 2022-09-28 15:43:04 -0700 | [diff] [blame] | 42 | |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 43 | def get_all(self): |
| 44 | url = f"https://api.github.com/repos/{self._repo}/releases" |
| 45 | page = 1 |
| Geoffrey Martin-Noble | 7d53e56 | 2022-09-28 15:43:04 -0700 | [diff] [blame] | 46 | |
| Scott Todd | 5cdd6ec | 2024-02-20 20:02:30 -0800 | [diff] [blame] | 47 | # GitHub limits API responses to the first 1000 results. |
| 48 | while page * self._per_page < 1000: |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 49 | response = self._session.get( |
| 50 | url, |
| 51 | params={ |
| 52 | "page": page, |
| 53 | "per_page": self._per_page, |
| 54 | }, |
| 55 | ) |
| 56 | for release in response.json(): |
| 57 | yield release |
| 58 | if "next" not in response.links: |
| 59 | break |
| 60 | page += 1 |
| Geoffrey Martin-Noble | 7d53e56 | 2022-09-28 15:43:04 -0700 | [diff] [blame] | 61 | |
| 62 | |
| 63 | def main(args): |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 64 | fetcher = ReleaseFetcher(repo=args.repo) |
| 65 | with sys.stdout if args.output == "-" else open(args.output, "w") as f: |
| 66 | f.write( |
| 67 | textwrap.dedent( |
| 68 | """\ |
| Geoffrey Martin-Noble | 7d53e56 | 2022-09-28 15:43:04 -0700 | [diff] [blame] | 69 | <!DOCTYPE html> |
| 70 | <html> |
| 71 | <body> |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 72 | """ |
| 73 | ) |
| 74 | ) |
| 75 | for release in fetcher.get_all(): |
| 76 | if release["draft"]: |
| 77 | continue |
| 78 | for asset in release["assets"]: |
| 79 | url = html.escape(asset["browser_download_url"]) |
| 80 | name = html.escape(asset["name"]) |
| 81 | f.write(f" <a href={url}>{name}</a><br />\n") |
| 82 | f.write( |
| 83 | textwrap.dedent( |
| 84 | """\ |
| Geoffrey Martin-Noble | 7d53e56 | 2022-09-28 15:43:04 -0700 | [diff] [blame] | 85 | </body> |
| 86 | </html> |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 87 | """ |
| 88 | ) |
| 89 | ) |
| Geoffrey Martin-Noble | 7d53e56 | 2022-09-28 15:43:04 -0700 | [diff] [blame] | 90 | |
| 91 | |
| 92 | if __name__ == "__main__": |
| Jakub Kuderski | be24f02 | 2023-06-21 14:44:18 -0400 | [diff] [blame] | 93 | main(parse_arguments()) |
