major refactoring adding vendor and customer features

models/database/marketplace_import_job.py

@@ -1,10 +1,8 @@
 from datetime import datetime, timezone
 
-from sqlalchemy import (Boolean, Column, DateTime, Float, ForeignKey, Index,
-                        Integer, String, Text, UniqueConstraint)
+from sqlalchemy import Column, DateTime, ForeignKey, Index, Integer, String, Text
 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
 
@@ -13,20 +11,17 @@ class MarketplaceImportJob(Base, TimestampMixin):
     __tablename__ = "marketplace_import_jobs"
 
     id = Column(Integer, primary_key=True, index=True)
+    vendor_id = Column(Integer, ForeignKey("vendors.id"), nullable=False, index=True)
+    user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
+
+    # Import configuration
+    marketplace = Column(String, nullable=False, index=True, default="Letzshop")
+    source_url = Column(String, nullable=False)
+
+    # Status tracking
     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)
@@ -35,28 +30,26 @@ class MarketplaceImportJob(Base, TimestampMixin):
     total_processed = Column(Integer, default=0)
 
     # Error handling
-    error_message = Column(String)
+    error_message = Column(Text)
 
     # Timestamps
+    started_at = Column(DateTime(timezone=True))
+    completed_at = Column(DateTime(timezone=True))
 
-    started_at = Column(DateTime)
-    completed_at = Column(DateTime)
-
-    # Relationship to user
-    user = relationship("User", foreign_keys=[user_id])
+    # Relationships
     vendor = relationship("Vendor", back_populates="marketplace_import_jobs")
+    user = relationship("User", foreign_keys=[user_id])
 
-    # Additional indexes for marketplace import job queries
+    # Indexes for performance
     __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"),
+        Index("idx_import_vendor_status", "vendor_id", "status"),
+        Index("idx_import_vendor_created", "vendor_id", "created_at"),
+        Index("idx_import_user_marketplace", "user_id", "marketplace"),
     )
 
     def __repr__(self):
         return (
-            f"<MarketplaceImportJob(id={self.id}, marketplace='{self.marketplace}', vendor='{self.vendor_name}', "
-            f"status='{self.status}', imported={self.imported_count})>"
+            f"<MarketplaceImportJob(id={self.id}, vendor_id={self.vendor_id}, "
+            f"marketplace='{self.marketplace}', status='{self.status}', "
+            f"imported={self.imported_count})>"
         )