import datetime
from sqlalchemy import Column, DateTime, Integer, String, Enum
from bulkperformancerights.connectors import mysql


class JobStatus(mysql.BaseModel):
    __tablename__ = 'job_status'

    job_id = Column(Integer, primary_key=True)
    input_file_path = Column(String)
    user_id = Column(Integer)
    user_type = Column(Enum('subaccount', 'vendor'))
    job_status = Column(Enum('uploaded', 'updating', 'converting',
                             'validating', 'processed', 'errored'))
    total_rows = Column(Integer)
    processed_rows = Column(Integer)
    error_rows = Column(Integer)
    error_message = Column(String)
    error_file_path = Column(String)
    start_time = Column(DateTime, default=datetime.datetime.now)
    end_time = Column(DateTime)