radionica3d/backend/seed_portfolio.py

import db

def seed():
    # Disable foreign key checks to make it easier to seed orders without users
    db.execute_commit("SET FOREIGN_KEY_CHECKS = 0;")
    
    # 1. Clear existing portfolio data for clean test
    db.execute_commit("DELETE FROM order_photos;")
    
    # 2. Insert Orders with consent
    orders_data = [
        ('Nikola', 'Tesla', '+38267123456', 'nikola@tesla.me', 'Podgorica, Montenegro', 'SLA Resin', True),
        ('John', 'Doe', '+38267000111', 'john@gmail.com', 'Budva, Montenegro', 'PLA Plastic', True),
        ('Alice', 'Smith', '+38268333444', 'alice@matrix.me', 'Kotor, Montenegro', 'Resin', True),
        ('Bob', 'Ross', '+38269555666', 'bob@art.me', 'Herceg Novi, Montenegro', 'ABS Plastic', True),
        ('Tony', 'Stark', '+38267999888', 'tony@stark.me', 'Tivat, Montenegro', 'Engineering Plastic', True),
    ]
    
    order_ids = []
    for o in orders_data:
        qid = db.execute_commit(
            "INSERT INTO orders (first_name, last_name, phone, email, shipping_address, material_name, allow_portfolio, status) VALUES (%s, %s, %s, %s, %s, %s, %s, 'completed')",
            o
        )
        order_ids.append(qid)
        
    # 3. Insert Photos
    photos = [
        (order_ids[0], 'uploads/portfolio_gear.png', True),
        (order_ids[1], 'uploads/portfolio_arch.png', True),
        (order_ids[2], 'uploads/portfolio_voronoi.png', True),
        (order_ids[3], 'uploads/portfolio_prosthetic.png', True),
        (order_ids[4], 'uploads/portfolio_minifigs.png', True),
    ]
    
    for p in photos:
        db.execute_commit(
            "INSERT INTO order_photos (order_id, file_path, is_public) VALUES (%s, %s, %s)",
            p
        )
    
    db.execute_commit("SET FOREIGN_KEY_CHECKS = 1;")
    print("Portfolio seeded successfully with 5 items.")

if __name__ == "__main__":
    seed()