Skip to content

Commit

Permalink
issue #26: migrated script from github-worflows. Missing unit and int…
Browse files Browse the repository at this point in the history
…egration test
  • Loading branch information
ThomasCardin committed Jan 30, 2024
1 parent a816f16 commit afcca69
Showing 1 changed file with 84 additions and 0 deletions.
84 changes: 84 additions & 0 deletions remove-previous-image/remove-previous-image.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,84 @@
"""
This script is based on this documentation:
https://docs.github.com/en/rest/packages/packages?apiVersion=2022-11-28
"""

import os
import requests
from requests.auth import HTTPBasicAuth

"""
Delete the old container (based on the previous tag)
"""
def delete_old_image(version_id, org, headers, auth):
url_delete_previous_version = f"https://api.github.com/orgs/{org}/packages/container/{container_name}/versions/{version_id}"
response = requests.delete(url_delete_previous_version, headers=headers, auth=auth)
if response.status_code == 204:
print(f'Previous container deleted!')
else:
raise Exception(f"error deleting the previous container: {response.status_code} {response.text}")

"""
Find the previous tag for a specific container.
Delete the previous tag if it exists and it is not the current one. This wont delete the current tag or older PR tags.
Check if tags is not empty and check if the len is == 1. If that is the case, it means that the only tag is the previous one.
"""
def find_previous_container_tag(response, unique_tag):
version_id = None
for version in response:
print(f"Found tags {version['metadata']['container']['tags']})")
tags = version['metadata']['container']['tags']
if unique_tag not in tags and len(tags) == 1 and tags:
version_id = version['id']
print(f"Previous tag found {tags[0]} with version_id {version_id}")
return tags[0], version_id

raise Exception(f"Container name {container_name} not found or the only tag found was the current one. If that is the case, you can ignore this error.")

"""
Get all GCR containers information
"""
def get_container_tags(org, container_name, auth, headers, container_path):
get_versions = f"https://api.github.com/orgs/{org}/packages/container/{container_name}/versions"
response = requests.get(get_versions, headers=headers, auth=auth)

try:
response.raise_for_status()
except requests.exceptions.HTTPError as e:
raise Exception(f"Error getting the previous tag for the container {container_path} : {e}")

return response

def print_console(message):
print("====================================")
print(message)

if __name__ == "__main__":
registry = os.getenv("REGISTRY")
github_token = os.getenv("GITHUB_TOKEN")
container_name = os.getenv("CONTAINER_NAME")
unique_tag = os.getenv("UNIQUE_TAG")
user = os.getenv("USER")
current_commit = os.getenv("CURRENT_COMMIT")

headers = {
"Accept": "application/vnd.github.v3+json",
}
auth = HTTPBasicAuth(user, github_token)

container_path = f"{registry}/{container_name}:{unique_tag}"
org = registry.split("/")[1]

print_console(f"Getting all tags for this container {container_path}...")
response = get_container_tags(org, container_name, auth, headers, container_path)
print("Done!")

print_console(f"Looking for the previous tag...")
previous_tag, version_id = find_previous_container_tag(response.json(), unique_tag)
print("Done!")

print_console(f"Deleting the previous container with tag ({previous_tag}) and version_id {version_id}...")
delete_old_image(version_id, org, headers, auth)
print("Done!")

0 comments on commit afcca69

Please sign in to comment.