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


class ExportStatus(mysql.BaseModel):
    __tablename__ = 'export_status'

    job_id = Column(Integer, primary_key=True)
    input_file_path = Column(String)
    output_file_path = Column(String)
    user_id = Column(Integer)
    user_type = Column(Enum('subaccount', 'vendor'))
    job_status = Column(Enum('querying', 'jsonify', 'xlsxify', 'exported',
                             'errored'))
    row_count = Column(Integer)
    error_message = Column(String)
    start_time = Column(DateTime, default=datetime.datetime.now)
    end_time = Column(DateTime)