fedspendingtransparency/usaspending-api

from usaspending_api.etl.transaction_loaders.data_load_helpers import format_insert_or_update_column_sql


def insert_award(cursor, load_object):
    columns, values, pairs = format_insert_or_update_column_sql(cursor, load_object, "award")
    generate_matching_award_sql = "INSERT INTO awards {} VALUES {} RETURNING id".format(columns, values)
    cursor.execute(generate_matching_award_sql)
    return cursor.fetchall()[0][0]


def update_transaction_fpds(cursor, load_object):
    columns, values, pairs = format_insert_or_update_column_sql(cursor, load_object, "transaction_fpds")
    transaction_fpds_sql = "UPDATE transaction_search SET {} where detached_award_proc_unique = '{}'".format(
        pairs, load_object["transaction_fpds"]["detached_award_proc_unique"]
    )
    cursor.execute(transaction_fpds_sql)


def update_transaction_normalized(cursor, load_object):
    columns, values, pairs = format_insert_or_update_column_sql(cursor, load_object, "transaction_normalized")
    transaction_normalized_sql = "UPDATE transaction_search SET {} where transaction_id  = '{}'".format(
        pairs, load_object["transaction_fpds"]["transaction_id"]
    )
    cursor.execute(transaction_normalized_sql)


def insert_transaction_normalized(cursor, load_object):
    columns, values, pairs = format_insert_or_update_column_sql(cursor, load_object, "transaction_normalized")
    transaction_normalized_sql = "INSERT INTO transaction_search {} VALUES {} RETURNING transaction_id".format(
        columns, values
    )
    cursor.execute(transaction_normalized_sql)
    created_transaction_normalized = cursor.fetchall()
    transaction_normalized_id = created_transaction_normalized[0][0]
    return transaction_normalized_id


def insert_transaction_fpds(cursor, load_object):
    columns, values, pairs = format_insert_or_update_column_sql(cursor, load_object, "transaction_fpds")
    transaction_fpds_sql = "INSERT INTO transaction_search {} VALUES {} RETURNING transaction_id".format(
        columns, values
    )
    cursor.execute(transaction_fpds_sql)
    created_transaction_fpds = cursor.fetchall()
    return created_transaction_fpds