#!/usr/bin/env python3
"""
Premakne DLC vsebino nazaj v glavne kategorije
"""

import os
import shutil
from pathlib import Path

BASE_DIR = Path("/Users/davidkotnik/Desktop/novafarma/Slike_za_Tiled")

def move_dlc_content():
    """Premakne DLC content v glavne kategorije"""
    print("🔄 Premikami DLC vsebino v glavne kategorije...\n")
    
    dlc_dir = BASE_DIR / "07_dlc_content"
    
    if not dlc_dir.exists():
        print("❌ DLC mapa ne obstaja!")
        return
    
    moved_count = 0
    
    # Amazon -> creatures/monsters
    for file in (dlc_dir / "amazon").glob("*.png"):
        shutil.move(str(file), str(BASE_DIR / "02_creatures/monsters" / file.name))
        moved_count += 1
    
    # Atlantis -> buildings/structures
    for file in (dlc_dir / "atlantis").glob("*.png"):
        shutil.move(str(file), str(BASE_DIR / "04_buildings/structures" / file.name))
        moved_count += 1
    
    # Catacombs -> creatures/monsters
    for file in (dlc_dir / "catacombs").glob("*.png"):
        shutil.move(str(file), str(BASE_DIR / "02_creatures/monsters" / file.name))
        moved_count += 1
    
    # Chernobyl -> creatures/monsters
    for file in (dlc_dir / "chernobyl").glob("*.png"):
        shutil.move(str(file), str(BASE_DIR / "02_creatures/monsters" / file.name))
        moved_count += 1
    
    # Desert/Egypt -> buildings/structures
    for file in (dlc_dir / "desert_egypt").glob("*.png"):
        shutil.move(str(file), str(BASE_DIR / "04_buildings/structures" / file.name))
        moved_count += 1
    
    # Lochness -> creatures/monsters
    for file in (dlc_dir / "lochness").glob("*.png"):
        shutil.move(str(file), str(BASE_DIR / "02_creatures/monsters" / file.name))
        moved_count += 1
    
    # Mythical -> creatures/monsters
    for file in (dlc_dir / "mythical").glob("*.png"):
        shutil.move(str(file), str(BASE_DIR / "02_creatures/monsters" / file.name))
        moved_count += 1
    
    # Pobriši prazne DLC mape
    shutil.rmtree(dlc_dir)
    
    print(f"✅ Premaknjenih {moved_count} DLC datotek")
    print("✅ Mapa 07_dlc_content izbrisana\n")

if __name__ == "__main__":
    print("=" * 60)
    print("  REORGANIZACIJA DLC VSEBINE")
    print("=" * 60)
    print()
    
    move_dlc_content()
    
    print("=" * 60)
    print("  KONČANO!")
    print("=" * 60)