31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
"""Collection Task model"""
|
|
|
|
from sqlalchemy import BigInteger, Column, DateTime, Integer, String, Text, Float
|
|
from sqlalchemy.sql import func
|
|
|
|
from app.db.session import Base
|
|
|
|
|
|
class CollectionTask(Base):
|
|
__tablename__ = "collection_tasks"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
datasource_id = Column(Integer, nullable=False, index=True)
|
|
status = Column(String(20), nullable=False) # pending, running, success, failed, cancelled
|
|
phase = Column(String(30), default="queued")
|
|
phase_progress = Column(Float)
|
|
phase_message = Column(String(255))
|
|
phase_current = Column(BigInteger)
|
|
phase_total = Column(BigInteger)
|
|
phase_unit = Column(String(30))
|
|
started_at = Column(DateTime(timezone=True))
|
|
completed_at = Column(DateTime(timezone=True))
|
|
records_processed = Column(Integer, default=0)
|
|
total_records = Column(Integer, default=0) # Total records to process
|
|
progress = Column(Float, default=0.0) # Progress percentage (0-100)
|
|
error_message = Column(Text)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
def __repr__(self):
|
|
return f"<CollectionTask {self.id}: {self.status}>"
|