issue #26: Migrated script from github-worflows. Missing unit and int…

remove-previous-image/remove_previous_image.py

@@ -0,0 +1,92 @@
+"""
+
+This script is based on this documentation:
+https://docs.github.com/en/rest/packages/packages?apiVersion=2022-11-28
+
+"""
+
+import os
+import requests
+import sys
+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!")
+
+    """
+    If there's no previous image to delete, we will stop the script (sys.exit()).
+    """
+    print_console(f"Looking for the previous tag...")
+    try:
+        previous_tag, version_id = find_previous_container_tag(response.json(), unique_tag)
+    except Exception as e:
+        print(e)
+        sys.exit() 
+    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!")

remove-previous-image/requirements.txt

@@ -0,0 +1 @@
+requests

setup.py

@@ -0,0 +1,21 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='devsecops-scripts',
+    version='1.0.0',
+    packages=find_packages(),
+    entry_points={
+        'console_scripts': [
+            'remove-previous-images=remove_previous_image.remove_previous_image:main',
+        ],
+    },
+    url='https://github.com/ai-cfia/devops.git',
+    author='ai-cfia',
+    author_email='[email protected]',
+    description='Every devops script used in dev, uat and production',
+    long_description=open('README.md').read(),
+    long_description_content_type='text/markdown',
+    install_requires=[
+        'requests'
+    ],
+)