# pylint: disable=redefined-outer-name
import logging
import os
from datetime import datetime, timedelta, timezone
from time import sleep

import pytest
from smelog.entities import LoggerConfig
from smelog.factory import LoggerFactory

from db_schema.common import ActivityStatusEnum, CompletenessStatusEnum
from db_schema.postgres import connection
from db_schema.schemas import slz


def creds_loader():
    return dict(
        active_endpoint=os.environ.get('PG_HOST', '0.0.0.0'),
        port=os.environ.get('PG_PORT', 5432),
        database=os.environ.get('PG_DB', 'slz'),
        username=os.environ.get('PG_USER', 'admin'),
        password=os.environ.get('PG_PASSWORD', 'admin'),
    )

@pytest.fixture(scope='function')
def db() -> connection.Connection:
    conn = connection.Connection(
        credentials_loader=creds_loader, name='db_schema_test', version='v1'
    )

    # Simple hack for waiting migrations to be completed
    while True:
        result = conn.session.execute(  # pylint: disable=no-member
            'SELECT id FROM databasechangelog ORDER BY dateexecuted DESC LIMIT 1'
        )
        if result.first():
            break
        sleep(1)

    yield conn

    for model in [
        slz.UnitOfWorkMigrationLog,
        slz.ContentStatusMigrationLog,
        slz.ContentFailureLog,
        slz.ContentStatus,
        slz.UnitOfWork,
    ]:
        conn.session.query(model).delete()
        conn.session.commit()
    conn.disconnect()


@pytest.fixture(scope='function')
def db_licensors(db):
    return {item.licensor_name: item for item in db.session.query(slz.Licensor).all()}


@pytest.fixture(scope='function')
def db_reports(db):
    return {item.report_name: item for item in db.session.query(slz.Report).all()}


@pytest.fixture(scope='function')
def db_data_sources(db):
    return {item.data_source_name: item for item in db.session.query(slz.DataSource).all()}


@pytest.fixture(scope='function')
def unit_of_work_stubs(db_licensors, db_reports):
    now = datetime.now().astimezone(timezone.utc)
    return {
        1:
            dict(
                unit_of_work_code='apple-20191124-theorchard-amEvent-v1_2',
                reprocess_id='1',
                licensor=db_licensors['theorchard'],
                report=db_reports['amEvent'],
                report_date='2019-11-24',
                version='v1_2',
                activity_status=ActivityStatusEnum.NOT_IN_PROGRESS.value,
                completeness_status=CompletenessStatusEnum.COMPLETE.value,
                is_force_complete=False,
                priority=slz.UnitOfWorkPriorityEnum.DEFAULT.value,
                next_run_at=now + timedelta(minutes=15),
                created_at=now,
                last_updated_at=now,
            ),
        2:
            dict(
                unit_of_work_code='apple-20191125-theorchard-amEvent-v1_2',
                reprocess_id='1',
                licensor=db_licensors['theorchard'],
                report=db_reports['amEvent'],
                report_date='2019-11-24',
                version='v1_2',
                activity_status=ActivityStatusEnum.NOT_IN_PROGRESS.value,
                completeness_status=CompletenessStatusEnum.COMPLETE.value,
                is_force_complete=False,
                priority=slz.UnitOfWorkPriorityEnum.DEFAULT.value,
                next_run_at=now + timedelta(minutes=15),
                created_at=now,
                last_updated_at=now,
            ),
    }

@pytest.fixture(scope='function')
def logger_test():
    log_config = LoggerConfig(
        name='TEST',
        version='1',
        level=logging.DEBUG,
        environment='test',
        is_local=True,
    )
    return LoggerFactory(log_config).get_logger('TEST')