|
| 1 | +# SPDX-FileCopyrightText: 2023 Alec Delaney |
| 2 | +# SPDX-License-Identifier: MIT |
| 3 | + |
| 4 | +"""Download the images from the GraphQL query to the static folder.""" |
| 5 | + |
| 6 | +import json |
| 7 | +import pathlib |
| 8 | +import sys |
| 9 | + |
| 10 | +import requests |
| 11 | + |
| 12 | +URL = "https://api.github.com/graphql" |
| 13 | + |
| 14 | +with open("/etc/config.json", encoding="utf-8") as jsonfile: |
| 15 | + config = json.load(jsonfile) |
| 16 | + |
| 17 | +with open("assets/graphql_query.txt", mode="r", encoding="utf-8") as queryfile: |
| 18 | + query_param = {"query": queryfile.read()} |
| 19 | + |
| 20 | +resp = requests.post( |
| 21 | + URL, |
| 22 | + json=query_param, |
| 23 | + headers={ |
| 24 | + "Authorization": f'Bearer {config["GH_TOKEN"]}', |
| 25 | + }, |
| 26 | + timeout=5, |
| 27 | +) |
| 28 | + |
| 29 | +json_resp = json.loads(resp.content)["data"]["user"] |
| 30 | +base_dir = pathlib.Path(sys.argv[1]) |
| 31 | +resp_dir = base_dir / "assets/contrib/" |
| 32 | +card_dir = base_dir / "flask_app/static/img/gh_cards/" |
| 33 | +resp_dir.mkdir(exist_ok=True) |
| 34 | +card_dir.mkdir(exist_ok=True) |
| 35 | + |
| 36 | +with open(str(resp_dir / "recent.json"), mode="w", encoding="utf-8") as contribfile: |
| 37 | + json.dump(json_resp, contribfile) |
| 38 | + |
| 39 | + |
| 40 | +for index, node in enumerate(json_resp["repositories"]["nodes"]): |
| 41 | + for _ in range(5): |
| 42 | + img_resp = requests.get(node["openGraphImageUrl"], timeout=5) |
| 43 | + if img_resp.status_code == 200: |
| 44 | + with open(str(card_dir / f"card{index}.png"), mode="wb") as imgfile: |
| 45 | + for data_chunk in img_resp: |
| 46 | + imgfile.write(data_chunk) |
0 commit comments