"""Delete foreign keys from tables."""

import mysql.connector


cnx = mysql.connector.connect(
    host='abacus-account-mysql',
    port='3306',
    database='royalty_accounting',
    user='royalties',
    password='1234'
)

fks = ["""ALTER TABLE account_payee
            DROP FOREIGN KEY fk_account_payee_account;""",
       """ALTER TABLE account_tax_info
           DROP FOREIGN KEY fk_account_tax_info_account_id;""",
       """ALTER TABLE account_payment_term
           DROP FOREIGN KEY fk_account_payment_term_account_id;""",
       """ALTER TABLE ledger_account_contract
           DROP FOREIGN KEY fk_ledger_account_contract_account;""",
       """ALTER TABLE ledger_account_contract
           DROP FOREIGN KEY fk_ledger_account_contract_contract;""",
       """ALTER TABLE ledger_account_contract
           DROP FOREIGN KEY fk_ledger_account_contract_abacus_event;""",
       """ALTER TABLE account_payment_term_history
           DROP FOREIGN KEY fk_account_payment_term_history_account;""",
       """ALTER TABLE ledger_account_contract_current_balance
           DROP FOREIGN KEY fk_ledger_account_contract_current_balance_account;""",
       """ALTER TABLE ledger_account_contract_current_balance
           DROP FOREIGN KEY fk_ledger_account_contract_current_balance_contract;""",
       """ALTER TABLE payment_group_payment_account
           DROP FOREIGN KEY fk_group_account_group_payment;""",
       """ALTER TABLE payment_group_payment_account
           DROP FOREIGN KEY fk_group_account_account;""",
       """ALTER TABLE payment_group_payment
           DROP FOREIGN KEY fk_payment_group_id;""",
       """ALTER TABLE payment_hold
           DROP FOREIGN KEY fk_account_payment_hold;""",
       """ALTER TABLE payment_group_payment_account
           DROP FOREIGN KEY fk_payment_account_reference_payoneer_program;""",
       """ALTER TABLE account_payee
           DROP FOREIGN KEY fk_payoneer_program_id;""",
       ]

for i in fks:
    try:
        cursor = cnx.cursor()
        cursor.execute(i)
    except Exception as ex:
        print('FK was already deleted: {}'.format(ex))

cnx.close()