"""Creates required DB fixtures."""

import mysql.connector


def mysqlclient_docker():
    """Initialize mysql client for docker image."""
    return mysql.connector.connect(
        host='correction-apply-mysql',
        port='3306',
        database='royalty_accounting',
        user='db_user',
        password='db_pass'
    )


cnx = mysqlclient_docker()

cursor = cnx.cursor()
query = """
INSERT INTO royalty_accounting.account (
    account_id,
    account_name,
    created_by,
    created_at,
    last_modified_by,
    last_modified
) VALUES (
    1042518,
    'TEST ACCOUNT',
    1472,
    '2021-04-12 02:38:45.000000',
    1472,
    '2021-04-12 02:38:45.000000'
)
"""
cursor.execute(query)
cnx.commit()


cursor = cnx.cursor()
query = """
        INSERT INTO contract (
            `contract_id`,
            `reference_signing_entity_id`,
            `contract_name`,
            `term_start`,
            `term_end`,
            `created_by`,
            `created_at`,
            `last_modified_by`,
            `last_modified`
        ) VALUES (
            1,
            1,
            'test',
            '2021-05-06',
            '2021-05-27',
            'vz',
            NOW(),
            'vz',
            NOW()
        );
    """
cursor.execute(query)
cnx.commit()

cursor = cnx.cursor()
query = """
        INSERT INTO account_contract(
            `account_contract_id`,
            `account_id`,
            `contract_id`
        ) VALUES (
            1,
            1042518,
            1
        );
    """
cursor.execute(query)
cnx.commit()


cursor = cnx.cursor()
query = """
INSERT INTO royalty_accounting.abacus_event (
    abacus_event_id,
    event_name,
    target_type,
    target_id,
    event_date,
    previous_abacus_event_id,
    rolled_back_at
) VALUES (
    1067,
    'apply_royalty_reversal',
    'statement_period',
    1,
    '2021-04-12 02:38:45.000000',
    null,
    null
)
"""
cursor.execute(query)
cnx.commit()


cursor = cnx.cursor()
query = """
    INSERT INTO royalty_accounting.statement_period(
        statement_period_id,
        statement_period_name,
        statement_period_status,
        closed_date,
        closed_by
    ) VALUES
        (1, 'Month 2', 'current', NULL, NULL)
"""
cursor.execute(query)
cnx.commit()

cursor = cnx.cursor()
query = """
INSERT INTO royalty_accounting.worksheet_correction (
    account_id,
    contract_id,
    statement_period_id,
    correction_statement_period_id,
    correction_type,
    currency_code,
    gross_revenue,
    distribution_fee,
    net_revenue,
    note,
    created_at,
    created_by,
    last_modified,
    last_modified_by,
    deleted_at,
    deleted_by
) VALUES (
    1042518,
    1,
    1,
    1,
    'royalty_reversal',
    'USD',
    200.000000000000000000,
    0.000000000000000000,
    200.000000000000000000,
    'test',
    '2023-04-26 12:31:07',
    'vz',
    '2023-04-26 10:31:16',
    'vz',
    null,
    null
);"""
cursor.execute(query)
cnx.commit()

cnx.close()

print('Fixtures were created.')