"""Creates required DB fixtures."""

import mysql.connector


def db_royalty_accounting_connector():
    """Connector to the royalty_accounting db (docker)."""
    cnx = mysql.connector.connect(
        host='sync-contract-mysql',
        port='3306',
        database='royalty_accounting',
        user='db_user',
        password='db_pass',
    )
    return cnx


def db_art_relations_connector():
    """Connector to the art_relations db (docker)."""
    cnx = mysql.connector.connect(
        host='sync-contract-legacy-mysql',
        port='3306',
        database='art_relations',
        user='db_user',
        password='db_pass',
    )
    return cnx


cnx = db_royalty_accounting_connector()
cnx_art_rel = db_art_relations_connector()


cursor = cnx.cursor()
query = """
INSERT INTO royalty_accounting.account (
    account_id,
    account_name,
    created_by,
    created_at,
    last_modified_by,
    last_modified
) VALUES (
    25153,
    '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 royalty_accounting.abacus_event (
    abacus_event_id,
    event_name,
    target_type,
    target_id,
    event_date,
    previous_abacus_event_id,
    rolled_back_at
) VALUES (
    1,
    'legacy_sync_contract',
    'contract',
    1,
    '2021-04-12 02:38:45.000000',
    null,
    null
)
"""
cursor.execute(query)
cnx.commit()


cursor = cnx.cursor()
query = """
INSERT INTO royalty_accounting.accounting_run (
    accounting_run_id,
    accounting_period_id,
    run_controller_id,
    run_status,
    start_date,
    summary_export_url,
    created_by,
    created_at,
    last_modified_by,
    last_modified
) VALUES (
    1472,
    21,
    2687,
    'Committing',
    NOW(),
    "s3://test-royalties-sales-files/accounting-period/accounting-run/fake.tsv",
    '2107',
    '2021-04-12 02:40:58',
    '1952',
    '2021-04-12 02:41:06'
)
"""
cursor.execute(query)
cnx.commit()

cursor = cnx.cursor()
query = """
INSERT INTO royalty_accounting.accounting_period (
accounting_period_id,
statement_period_id,
accounting_period_name,
accounting_period_status,
contract_type,
closed_date,
created_by,
created_at,
last_modified_by,
last_modified)
VALUES (
21,
256,
'test',
'open',
'distribution',
null,
'vz',
'2022-02-16 05:35:39',
'vz',
'2022-02-16 05:35:42')"""
cursor.execute(query)
cnx.commit()

cursor = cnx.cursor()
query = """
INSERT INTO royalty_accounting.contract (
    contract_id,
    reference_signing_entity_id,
    contract_name,
    contract_type,
    term_start,
    term_end,
    created_by,
    created_at,
    last_modified_by,
    last_modified
) VALUES (
    1,
    1,
    'test',
    'distribution',
    '2022-06-06',
    '2022-06-09',
    'vz',
    '2022-06-06 03:43:02',
    'vz',
    '2022-06-06 03:43:05'
)
"""
cursor.execute(query)
cnx.commit()

cursor = cnx.cursor()
query = """INSERT INTO royalty_accounting.account_payment_term (
    account_payment_term_id,
    account_id,
    currency_code,
    payment_minimum,
    payment_entity_id,
    payment_schedule,
    created_by,
    created_at,
    last_modified_by,
    last_modified
) VALUES (
    1,
    25153,
    'USD',
    33.00,
    1,
    '30_days_after_month_end',
    'vz',
    '2022-06-06 04:12:24',
    'vz',
    '2022-06-06 04:12:28'
)"""
cursor.execute(query)
cnx.commit()

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

cursor = cnx.cursor()
insert_abacus_states = """
    INSERT INTO royalty_accounting.abacus_state(
        parent_table_id,
        parent_table_name,
        action_name,
        action_status,
        created_by,
        created_at,
        last_modified_by,
        last_modified
    ) VALUES (
        1,
        'contract',
        'legacy_sync',
        'init',
        'dana',
        NOW(),
        'dana',
        NOW()
    ), (
        1,
        'contract',
        'sap_sync',
        'init',
        'dana',
        NOW(),
        'dana',
        NOW()
    );
"""
cursor.execute(insert_abacus_states)
cnx.commit()


cnx.close()

print('Fixtures were created.')