update_version.py

# 15.06.24

import os
import shutil
from io import BytesIO
from zipfile import ZipFile


# External library
import httpx
from rich.console import Console
from rich.prompt import Prompt


# Variable
console = Console()
local_path = os.path.join(".")
from Src.Upload.version import __author__, __title__


def move_content(source: str, destination: str):
    """
    Move all content from the source folder to the destination folder.

    Parameters:
        - source (str): The path to the source folder.
        - destination (str): The path to the destination folder.
    """

    os.makedirs(destination, exist_ok=True)

    # Iterate through all elements in the source folder
    for element in os.listdir(source):
        source_path = os.path.join(source, element)
        destination_path = os.path.join(destination, element)

        # If it's a directory, recursively call the function
        if os.path.isdir(source_path):
            move_content(source_path, destination_path)

        # Otherwise, move the file, replacing if it already exists    
        else:
            shutil.move(source_path, destination_path)


def keep_specific_items(directory: str, keep_folder: str, keep_file: str):
    """
    Delete all items in the directory except for the specified folder and file.

    Parameters:
        - directory (str): The path to the directory.
        - keep_folder (str): The name of the folder to keep.
        - keep_file (str): The name of the file to keep.
    """

    try:
        if not os.path.exists(directory) or not os.path.isdir(directory):
            raise ValueError(f"Error: '{directory}' is not a valid directory.")
        
        # Iterate through items in the directory
        for item in os.listdir(directory):
            item_path = os.path.join(directory, item)

            # Check if the item is the specified folder or file
            if os.path.isdir(item_path) and item != keep_folder:
                shutil.rmtree(item_path)

            elif os.path.isfile(item_path) and item != keep_file:
                os.remove(item_path)

    except PermissionError as pe:
        console.print(f"[red]PermissionError: {pe}. Check permissions and try again.")

    except Exception as e:
        console.print(f"[red]Error: {e}")


def download_and_extract_latest_commit():
    """
    Download and extract the latest commit from a GitHub repository.
    """

    try:

        # Get the latest commit information using GitHub API
        api_url = f'https://api.github.com/repos/{__author__}/{__title__}/commits?per_page=1'
        response = httpx.get(api_url, timeout=10)
        console.log("[green]Requesting latest commit from GitHub repository...")

        if response.status_code == 200:
            commit_info = response.json()[0]
            commit_sha = commit_info['sha']
            zipball_url = f'https://github.com/{__author__}/{__title__}/archive/{commit_sha}.zip'
            console.log("[green]Downloading latest commit zip file...")

            # Download the zipball
            response = httpx.get(zipball_url, follow_redirects=True, timeout=10)
            temp_path = os.path.join(os.path.dirname(os.getcwd()), 'temp_extracted')

            # Extract the content of the zipball into a temporary folder
            with ZipFile(BytesIO(response.content)) as zip_ref:
                zip_ref.extractall(temp_path)
            console.log("[green]Extracting files...")

            # Move files from the temporary folder to the current folder
            for item in os.listdir(temp_path):
                item_path = os.path.join(temp_path, item)
                destination_path = os.path.join(local_path, item)
                shutil.move(item_path, destination_path)

            # Remove the temporary folder
            shutil.rmtree(temp_path)

            # Move all folder to main folder
            new_folder_name = f"{__title__}-{commit_sha}"
            move_content(new_folder_name, ".")

            # Remove old temp folder
            shutil.rmtree(new_folder_name)
            console.log("[cyan]Latest commit downloaded and extracted successfully.")
        else:
            console.log(f"[red]Failed to fetch commit information. Status code: {response.status_code}")

    except httpx.RequestError as e:
        console.print(f"[red]Request failed: {e}")

    except Exception as e:
        console.print(f"[red]An unexpected error occurred: {e}")


def main_upload():
    """
    Main function to upload the latest commit of a GitHub repository.
    """
    cmd_insert = Prompt.ask("[bold red]Are you sure you want to delete all files? (Only 'Video' folder and 'update_version.py' will remain)", choices=['y', 'n'], default='y', show_choices=True)

    if cmd_insert.lower() == "yes":
        console.print("[red]Deleting all files except 'Video' folder and 'update_version.py'...")
        keep_specific_items(".", "Video", "upload.py")
        download_and_extract_latest_commit()
        
    else:
        console.print("[red]Operation cancelled.")


if __name__ == "__main__":
    main_upload()