unzip_parts.py

#!/bin/env python3
"""Module for unziping and merging split db zip file."""

import os
from zipfile import ZipFile


def unzip_parts(path):
    """Unzip and merge split zip file."""
    # unzip (needs to go into download function finally)
    # Set the name of the original file
    db_zip_file = os.path.join(path, "parts-fts5.db.zip")

    # Open the original file for writing
    with open(db_zip_file, "wb") as db:
        # Get a list of the split files in the split directory
        split_files = [
            f for f in os.listdir(path) if f.startswith("parts-fts5.db.zip.")
        ]

        # Sort the split files by their index
        split_files.sort(key=lambda f: int(f.split(".")[-1]))

        # Iterate over the split files and append their contents to the original file
        for split_file_name in split_files:
            split_path = os.path.join(path, split_file_name)
            # Open the split file
            with open(split_path, "rb") as split_file:
                # Read the file data
                while file_data := split_file.read(1024 * 1024):
                    # Append the file data to the original file
                    db.write(file_data)

            # Delete the split file
            os.unlink(split_path)

    with ZipFile(db_zip_file, "r") as zf:
        zf.extractall(path)

    os.unlink(db_zip_file)