| import os |
| import requests |
| import shutil |
| import uuid |
| from sqlmodel import Session, select |
| from models import User, WardrobeItem |
| from database import engine |
| from datetime import datetime |
|
|
| |
| SAMPLE_ITEMS = [ |
| { |
| "category": "top", |
| "url": "https://images.unsplash.com/photo-1521572163474-6864f9cf17ab?w=600&q=80", |
| "description": "White Cotton T-Shirt" |
| }, |
| { |
| "category": "top", |
| "url": "https://images.unsplash.com/photo-1618354691373-d851c5c3a990?w=600&q=80", |
| "description": "Black Graphic Tee" |
| }, |
| { |
| "category": "top", |
| "url": "https://images.unsplash.com/photo-1551028919-ac66e6a46121?w=600&q=80", |
| "description": "Blue Denim Jacket" |
| }, |
| { |
| "category": "bottom", |
| "url": "https://images.unsplash.com/photo-1542272617-08f086303294?w=600&q=80", |
| "description": "Patterned Summer Skirt" |
| }, |
| { |
| "category": "bottom", |
| "url": "https://images.unsplash.com/photo-1584370848010-d7cc637703ef?w=600&q=80", |
| "description": "Grey Trousers" |
| }, |
| { |
| "category": "bottom", |
| "url": "https://images.unsplash.com/photo-1541099649105-f69ad21f3246?w=600&q=80", |
| "description": "Blue Jeans" |
| }, |
| { |
| "category": "shoes", |
| "url": "https://images.unsplash.com/photo-1549298916-b41d501d3772?w=600&q=80", |
| "description": "Running Shoes" |
| }, |
| { |
| "category": "shoes", |
| "url": "https://images.unsplash.com/photo-1543163521-1bf539c55dd2?w=600&q=80", |
| "description": "High Heels" |
| } |
| ] |
|
|
| UPLOAD_DIR = "static/uploads" |
| os.makedirs(UPLOAD_DIR, exist_ok=True) |
|
|
| def download_image(url): |
| try: |
| response = requests.get(url, stream=True) |
| if response.status_code == 200: |
| file_ext = "jpg" |
| filename = f"{uuid.uuid4()}.{file_ext}" |
| file_path = os.path.join(UPLOAD_DIR, filename) |
| with open(file_path, 'wb') as f: |
| response.raw.decode_content = True |
| shutil.copyfileobj(response.raw, f) |
| return f"/static/uploads/{filename}" |
| except Exception as e: |
| print(f"Failed to download {url}: {e}") |
| return None |
|
|
| def seed(): |
| print("Starting data seeding...") |
| with Session(engine) as session: |
| |
| user = session.exec(select(User).where(User.email == "demo@bsure.app")).first() |
| if not user: |
| print("Creating demo user...") |
| |
| user = User( |
| email="demo@bsure.app", |
| password_hash="$pbkdf2-sha256$29000$dummyhash$dummy", |
| full_name="Demo User" |
| ) |
| session.add(user) |
| session.commit() |
| session.refresh(user) |
| |
| print(f"Seeding wardrobe for user: {user.email}") |
| |
| count = 0 |
| for item_data in SAMPLE_ITEMS: |
| print(f"Downloading {item_data['description']}...") |
| local_url = download_image(item_data['url']) |
| if local_url: |
| db_item = WardrobeItem( |
| owner_id=user.id, |
| image_url=local_url, |
| category=item_data['category'], |
| tags="demo-data" |
| ) |
| session.add(db_item) |
| count += 1 |
| |
| session.commit() |
| print(f"Successfully seeded {count} items!") |
|
|
| if __name__ == "__main__": |
| seed() |
|
|
