mirror of
https://github.com/ClementTsang/bottom
synced 2024-11-25 05:30:22 +00:00
98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
#!/bin/python3
|
|
|
|
# A simple script to clean caches matching a PR ID.
|
|
#
|
|
# Expects a GitHub token in the environment variables as GITHUB_TOKEN.
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
import time
|
|
from urllib.error import HTTPError, URLError
|
|
|
|
from urllib.request import Request, urlopen
|
|
|
|
URL = "https://api.github.com/repos/ClementTsang/bottom/actions/caches"
|
|
|
|
|
|
def cache_list_request(key):
|
|
request = Request(URL, method="GET")
|
|
request.add_header("Accept", "application/vnd.github+json")
|
|
request.add_header("Authorization", "Bearer {}".format(key))
|
|
return request
|
|
|
|
|
|
def delete_cache_request(key, id):
|
|
request = Request("{}/{}".format(URL, id), method="DELETE")
|
|
request.add_header("Accept", "application/vnd.github+json")
|
|
request.add_header("Authorization", "Bearer {}".format(key))
|
|
return request
|
|
|
|
|
|
def main():
|
|
args = sys.argv
|
|
env = os.environ
|
|
|
|
key = env["GITHUB_TOKEN"]
|
|
if args[1].isnumeric():
|
|
pr_id = int(args[1])
|
|
ref = "refs/pull/{}/merge".format(pr_id)
|
|
|
|
print("Clearing any caches generated by PR {}".format(pr_id))
|
|
with urlopen(cache_list_request(key)) as response:
|
|
response = json.load(response)
|
|
caches = response["actions_caches"]
|
|
for cache in caches:
|
|
if cache["ref"] == ref:
|
|
id = cache["id"]
|
|
try:
|
|
print("Deleting ID {}...".format(id))
|
|
urlopen(delete_cache_request(key, id))
|
|
except HTTPError as e:
|
|
print("HTTPError with delete, error code {}.".format(e.code))
|
|
except URLError as _:
|
|
print("URLError with delete.")
|
|
else:
|
|
print("Successfully deleted cache ID {}!".format(id))
|
|
time.sleep(0.1)
|
|
elif args[1] == "keep-main" or args[1] == "keep-master":
|
|
print("Clearing all but default branch cache.")
|
|
with urlopen(cache_list_request(key)) as response:
|
|
response = json.load(response)
|
|
caches = response["actions_caches"]
|
|
for cache in caches:
|
|
if not ("master" in cache["ref"] or "main" in cache["ref"]):
|
|
id = cache["id"]
|
|
try:
|
|
print("Deleting ID {}...".format(id))
|
|
urlopen(delete_cache_request(key, id))
|
|
except HTTPError as e:
|
|
print("HTTPError with delete, error code {}.".format(e.code))
|
|
except URLError as _:
|
|
print("URLError with delete.")
|
|
else:
|
|
print("Successfully deleted cache ID {}!".format(id))
|
|
time.sleep(0.1)
|
|
elif args[1] == "main" or args[1] == "master" or args[1] == "all":
|
|
print("Clearing all caches.")
|
|
with urlopen(cache_list_request(key)) as response:
|
|
response = json.load(response)
|
|
caches = response["actions_caches"]
|
|
for cache in caches:
|
|
id = cache["id"]
|
|
try:
|
|
print("Deleting ID {}...".format(id))
|
|
urlopen(delete_cache_request(key, id))
|
|
except HTTPError as e:
|
|
print("HTTPError with delete, error code {}.".format(e.code))
|
|
except URLError as _:
|
|
print("URLError with delete.")
|
|
else:
|
|
print("Successfully deleted cache ID {}!".format(id))
|
|
time.sleep(0.1)
|
|
else:
|
|
print(f"Skipping, given argument {args[1]}.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|