理论上的测试001

1

import os
import shutil
import zipfile

def process_files(target_dir):
    # Iterate over each item in the target directory
    for item in os.listdir(target_dir):
        item_path = os.path.join(target_dir, item)
        new_folder_name = f"{item}_unzip"
        new_folder_path = os.path.join(target_dir, new_folder_name)
        
        # Create a new folder for the item
        os.makedirs(new_folder_path, exist_ok=True)
        
        if os.path.isdir(item_path):
            # If item is a directory, copy it to the new folder
            shutil.copytree(item_path, os.path.join(new_folder_path, item))
        elif item.lower().endswith('.zip'):
            # If item is a zip file, extract it to the new folder
            with zipfile.ZipFile(item_path, 'r') as zip_ref:
                zip_ref.extractall(new_folder_path)

# Replace 'your_target_directory' with the path to your directory
process_files('your_target_directory')