"""
Script for checking results of daemon work.
Compare results for test DB and result log DB.
Compare by pair <ISRC, correlation_id>.

Need to set <Environment> and <DB_CONNECTION_STRING> in env
variables of OS or in configuration of runner.

For load testing.
"""

from scripts.fill_load_test_db import YouTubeApiLogTest

from ytownership.connectors.mysql import session_scope
from ytownership.models.apilog import YouTubeApiLog


def check_results():

    with session_scope() as db_session:
        for instance in db_session.query(YouTubeApiLogTest):
            finished_from_log = (
            db_session
            .query(YouTubeApiLog)
            .filter(YouTubeApiLog.correlation_id ==
            instance.correlation_id)
            .filter(YouTubeApiLog.isrc == instance.isrc)
            .first())
            if (finished_from_log is None or
                finished_from_log.code != 200):
                print(
                    instance.isrc,
                    ' ',
                    instance.correlation_id,
                    ' not found.')


if __name__ == '__main__':
    print('Start working')
    check_results()
    print('Finish working')