from pynamodb.attributes import NumberAttribute, UnicodeAttribute, UTCDateTimeAttribute
from pynamodb.models import Model

from slz_api_service.v1.enums import ColumnName, TableName


class DyContentStatus(Model):
    class Meta:
        table_name = TableName.ContentStatus.value

    completed_at = UTCDateTimeAttribute(attr_name=ColumnName.CompletedAt.value)
    content_name = UnicodeAttribute(attr_name=ColumnName.ContentName.value, range_key=True)
    content_status = UnicodeAttribute(attr_name=ColumnName.ContentStatus.value)
    context = UnicodeAttribute(attr_name=ColumnName.Context.value)
    created_at = UTCDateTimeAttribute(attr_name=ColumnName.CreatedAt.value)
    file_size = NumberAttribute(attr_name=ColumnName.FileSize.value)
    job_id = UnicodeAttribute(attr_name=ColumnName.JobID.value)
    last_checked_at = UTCDateTimeAttribute(attr_name=ColumnName.LastCheckedAt.value)
    original_content_name = UnicodeAttribute(attr_name=ColumnName.OriginalContentName.value,
                                             null=True)
    unit_of_work_id = UnicodeAttribute(attr_name=ColumnName.UnitOfWorkID.value, hash_key=True)