from datetime import datetime, timezone

from db_schema.schemas import exploration, slz


def uow_stub(**kwargs) -> slz.UnitOfWork:
    uow = slz.UnitOfWork(
        unit_of_work_code='1',
        licensor_id=1,
        report_id=1,
        report_date='2019-01-02',
        activity_status=slz.ActivityStatusEnum.NOT_IN_PROGRESS.value,
        completeness_status=slz.CompletenessStatusEnum.ACTIVE.value,
        is_force_complete=False,
        priority=slz.UnitOfWorkPriorityEnum.PRIORITY_5.value,
        next_run_at=datetime.now(tz=timezone.utc),
        created_at=datetime.now(tz=timezone.utc),
        last_updated_at=datetime.now(tz=timezone.utc),
    )
    if kwargs:
        for k, v in kwargs.items():
            if hasattr(uow, k):
                setattr(uow, k, v)
    return uow


def content_status_stub(**kwargs) -> slz.ContentStatus:
    cs = slz.ContentStatus(
        context='US',
        content_name='us.txt',
        content_status=slz.ContentStatusEnum.COMPLETE,
        created_at=datetime.now(tz=timezone.utc),
        failure_count=0,
        sub_content='{}',
        metadata_process_status=slz.ContentMetadataStatusEnum.NOT_QUEUED,
    )
    if kwargs:
        for k, v in kwargs.items():
            if hasattr(cs, k):
                setattr(cs, k, v)
    return cs


def disassemble_cs_stub(**kwargs) -> exploration.DisassembleContentStatus:
    cs = exploration.DisassembleContentStatus(
        content_name='us.txt',
        disassemble_status=exploration.DisassembleStatusEnum.QUEUED,
        created_at=datetime.now(tz=timezone.utc),
        last_updated_at=datetime.now(tz=timezone.utc),
    )
    if kwargs:
        for k, v in kwargs.items():
            if hasattr(cs, k):
                setattr(cs, k, v)
    return cs