157 lines
5.8 KiB
Python
157 lines
5.8 KiB
Python
"""
|
|
Product Asset Models
|
|
Database models for storing product-specific assets (separate from campaign assets).
|
|
These models are for the Product Marketing Suite (product asset creation).
|
|
"""
|
|
|
|
from sqlalchemy import Column, Integer, String, DateTime, Float, Boolean, JSON, Text, ForeignKey, Index
|
|
from sqlalchemy.orm import relationship
|
|
from datetime import datetime
|
|
import enum
|
|
|
|
from models.subscription_models import Base
|
|
|
|
|
|
class ProductAssetType(enum.Enum):
|
|
"""Product asset type enum."""
|
|
IMAGE = "image"
|
|
VIDEO = "video"
|
|
AUDIO = "audio"
|
|
ANIMATION = "animation"
|
|
|
|
|
|
class ProductImageStyle(enum.Enum):
|
|
"""Product image style enum."""
|
|
STUDIO = "studio"
|
|
LIFESTYLE = "lifestyle"
|
|
OUTDOOR = "outdoor"
|
|
MINIMALIST = "minimalist"
|
|
LUXURY = "luxury"
|
|
TECHNICAL = "technical"
|
|
|
|
|
|
class ProductAsset(Base):
|
|
"""
|
|
Product asset model.
|
|
Stores product-specific assets (images, videos, audio) generated for product marketing.
|
|
"""
|
|
|
|
__tablename__ = "product_assets"
|
|
|
|
# Primary fields
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
product_id = Column(String(255), nullable=False, index=True) # User-defined product ID
|
|
user_id = Column(String(255), nullable=False, index=True) # Clerk user ID
|
|
|
|
# Product information
|
|
product_name = Column(String(500), nullable=False)
|
|
product_description = Column(Text, nullable=True)
|
|
|
|
# Asset details
|
|
asset_type = Column(String(50), nullable=False, index=True) # image, video, audio, animation
|
|
variant = Column(String(100), nullable=True) # color, size, angle, etc.
|
|
style = Column(String(50), nullable=True) # studio, lifestyle, minimalist, etc.
|
|
environment = Column(String(50), nullable=True) # studio, lifestyle, outdoor, etc.
|
|
|
|
# Link to ContentAsset (unified asset library)
|
|
content_asset_id = Column(Integer, ForeignKey('content_assets.id', ondelete='SET NULL'), nullable=True, index=True)
|
|
|
|
# Generation details
|
|
provider = Column(String(100), nullable=True)
|
|
model = Column(String(100), nullable=True)
|
|
cost = Column(Float, default=0.0)
|
|
generation_time = Column(Float, nullable=True)
|
|
prompt_used = Column(Text, nullable=True)
|
|
|
|
# E-commerce integration
|
|
ecommerce_exported = Column(Boolean, default=False)
|
|
exported_to = Column(JSON, nullable=True) # Array of platform names
|
|
|
|
# Status
|
|
status = Column(String(50), default="completed", nullable=False) # completed, processing, failed
|
|
|
|
# Metadata
|
|
created_at = Column(DateTime, default=datetime.utcnow, nullable=False, index=True)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
# Additional metadata (renamed from 'metadata' to avoid SQLAlchemy reserved name conflict)
|
|
# Using 'product_metadata' as column name in DB to avoid conflict with SQLAlchemy's reserved 'metadata' attribute
|
|
product_metadata = Column('product_metadata', JSON, nullable=True) # Additional product-specific metadata
|
|
|
|
# Composite indexes
|
|
__table_args__ = (
|
|
Index('idx_user_product', 'user_id', 'product_id'),
|
|
Index('idx_user_type', 'user_id', 'asset_type'),
|
|
Index('idx_product_type', 'product_id', 'asset_type'),
|
|
)
|
|
|
|
|
|
class ProductStyleTemplate(Base):
|
|
"""
|
|
Brand style template for products.
|
|
Stores reusable brand style configurations for product asset generation.
|
|
"""
|
|
|
|
__tablename__ = "product_style_templates"
|
|
|
|
# Primary fields
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
user_id = Column(String(255), nullable=False, index=True)
|
|
template_name = Column(String(255), nullable=False)
|
|
|
|
# Style configuration
|
|
color_palette = Column(JSON, nullable=True) # Array of brand colors
|
|
background_style = Column(String(50), nullable=True) # white, transparent, lifestyle, branded
|
|
lighting_preset = Column(String(50), nullable=True) # natural, studio, dramatic, soft
|
|
preferred_style = Column(String(50), nullable=True) # photorealistic, minimalist, luxury, technical
|
|
preferred_environment = Column(String(50), nullable=True) # studio, lifestyle, outdoor
|
|
|
|
# Brand integration
|
|
use_brand_colors = Column(Boolean, default=True)
|
|
use_brand_logo = Column(Boolean, default=False)
|
|
|
|
# Metadata
|
|
is_default = Column(Boolean, default=False) # Default template for user
|
|
created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
# Composite indexes
|
|
__table_args__ = (
|
|
Index('idx_user_template', 'user_id', 'template_name'),
|
|
)
|
|
|
|
|
|
class EcommerceExport(Base):
|
|
"""
|
|
E-commerce platform export tracking.
|
|
Tracks product asset exports to e-commerce platforms.
|
|
"""
|
|
|
|
__tablename__ = "product_ecommerce_exports"
|
|
|
|
# Primary fields
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
user_id = Column(String(255), nullable=False, index=True)
|
|
product_id = Column(String(255), nullable=False, index=True)
|
|
|
|
# Platform information
|
|
platform = Column(String(50), nullable=False) # shopify, amazon, woocommerce
|
|
platform_product_id = Column(String(255), nullable=True) # Product ID on the platform
|
|
|
|
# Export details
|
|
exported_assets = Column(JSON, nullable=False) # Array of asset IDs exported
|
|
export_status = Column(String(50), default="pending", nullable=False) # pending, completed, failed
|
|
error_message = Column(Text, nullable=True)
|
|
|
|
# Metadata
|
|
created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
exported_at = Column(DateTime, nullable=True)
|
|
|
|
# Composite indexes
|
|
__table_args__ = (
|
|
Index('idx_user_platform', 'user_id', 'platform'),
|
|
Index('idx_product_platform', 'product_id', 'platform'),
|
|
)
|
|
|