from datetime import datetime, timezone

from sqlalchemy import (Boolean, Column, DateTime, Float, ForeignKey, Index,
                        Integer, String, Text, UniqueConstraint)
from sqlalchemy.orm import relationship

# Import Base from the central database module instead of creating a new one
from app.core.database import Base
from models.database.base import TimestampMixin


class MarketplaceImportJob(Base, TimestampMixin):
    __tablename__ = "marketplace_import_jobs"

    id = Column(Integer, primary_key=True, index=True)
    status = Column(
        String, nullable=False, default="pending"
    )  # pending, processing, completed, failed, completed_with_errors
    source_url = Column(String, nullable=False)
    marketplace = Column(
        String, nullable=False, index=True, default="Letzshop"
    )  # Index for marketplace filtering
    vendor_name = Column(String, nullable=False, index=True)  # Index for vendor filtering
    vendor_id = Column(
        Integer, ForeignKey("vendors.id"), nullable=False
    )  # Add proper foreign key
    user_id = Column(
        Integer, ForeignKey("users.id"), nullable=False
    )  # Foreign key to users table

    # Results
    imported_count = Column(Integer, default=0)
    updated_count = Column(Integer, default=0)
    error_count = Column(Integer, default=0)
    total_processed = Column(Integer, default=0)

    # Error handling
    error_message = Column(String)

    # Timestamps

    started_at = Column(DateTime)
    completed_at = Column(DateTime)

    # Relationship to user
    user = relationship("User", foreign_keys=[user_id])
    vendor = relationship("Vendor", back_populates="marketplace_import_jobs")

    # Additional indexes for marketplace import job queries
    __table_args__ = (
        Index(
            "idx_marketplace_import_user_marketplace", "user_id", "marketplace"
        ),  # User's marketplace imports
        Index("idx_marketplace_import_vendor_status", "status"),  # Vendor import status
        Index("idx_marketplace_import_vendor_id", "vendor_id"),
    )

    def __repr__(self):
        return (
            f"<MarketplaceImportJob(id={self.id}, marketplace='{self.marketplace}', vendor='{self.vendor_name}', "
            f"status='{self.status}', imported={self.imported_count})>"
        )