Zip Zip Zip

import zipfile
import os

def unzip_recursive(zip_path, output_file, current_dir='.'):
    
    # Get the extracted folder name
    extracted_folder = os.path.splitext(os.path.basename(zip_path))[0]
    extracted_folder_path = os.path.join(current_dir, extracted_folder)
    print(f"Extracted folder: {extracted_folder_path}")
    # Extract the zip file
    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
        zip_ref.extractall(extracted_folder_path)
    
    # Traverse the folder
    for root, dirs, files in os.walk(extracted_folder_path):
        for file in files:
            if file.endswith('.txt'):
                # Read and record the content of .txt files
                with open(os.path.join(root, file), 'r') as txt_file:
                    content = txt_file.read()
                    with open(output_file, 'a') as out_file:
                        out_file.write(f"Content of {file}:\n")
                        out_file.write(content + "\n\n")
            elif file.endswith('.zip'):
                # Recursively unzip the next zip file
                next_zip = os.path.join(root, file)
                unzip_recursive(next_zip, output_file, root)

# Example usage
zip_path = './chunk_0.zip'  # The path to the initial zip file
output_file = './output.txt'  # The file to record the .txt contents
unzip_recursive(zip_path, output_file)

import zipfile
import os
import io

def unzip_iterative_in_memory(start_zip):
    # Stack to hold zip files that need to be processed
    zip_stack = [start_zip]

    while zip_stack:
        current_zip = zip_stack.pop()

        # Open the current zip file
        with zipfile.ZipFile(current_zip, 'r') as zip_ref:
            print(f"Number of files in the zip: {len(zip_ref.namelist())}")
            # Process each file in the zip
            for file in zip_ref.namelist():
                # print the number of files in the zip
                # If it's a .txt file, read and print the content
                if file.endswith('.txt'):
                    with zip_ref.open(file) as txt_file:
                        content = txt_file.read().decode('utf-8')
                        print(f"Content of {file}:\n{content}\n")
                        
                    # extract the number from the file name
                    with open(f'./result.txt', 'a') as f:
                        f.write(content)
                # If it's another .zip file, read it into memory and add to the stack
                elif file.endswith('.zip'):
                    with zip_ref.open(file) as nested_zip_file:
                        nested_zip_data = io.BytesIO(nested_zip_file.read())
                        zip_stack.append(nested_zip_data)