import os
import shutil
from datetime import datetime

# Define source and destination root folders
source_folder = r"P:\o\iPhone"
destination_root = r"V:\o\iShot\ReQuin"

# Ensure root folders exist
for fmt in ["mp4", "mov"]:
    os.makedirs(os.path.join(destination_root, fmt), exist_ok=True)

# Walk through the source folder
for root, _, files in os.walk(source_folder):
    for file in files:
        file_ext = file.lower().rsplit(".", 1)[-1]  # Get file extension
        if file_ext in ["mp4", "mov"]:
            file_path = os.path.join(root, file)
            mod_time = os.path.getmtime(file_path)
            year = datetime.fromtimestamp(mod_time).strftime("%Y")  # Get year (e.g., 2024)
            month = datetime.fromtimestamp(mod_time).strftime("%m")  # Get month (01-12)

            # Define target folder (e.g., C:\path\to\destination\mp4\2024\03)
            target_folder = os.path.join(destination_root, file_ext, year, month)
            os.makedirs(target_folder, exist_ok=True)

            # Move file
            shutil.move(file_path, os.path.join(target_folder, file))
            print(f"Moved: {file} → {target_folder}")

print("File transfer completed.")