repocsv_multiple_prices.py

"""
Copy from csv repo files to db for multiple prices
"""

from sysdata.csv.csv_multiple_prices import csvFuturesMultiplePricesData
from sysproduction.data.prices import diagPrices


if __name__ == "__main__":
    input("Will overwrite existing prices are you sure?! CTL-C to abort")
    diag_prices = diagPrices()

    db_multiple_prices = diag_prices.db_futures_multiple_prices_data
    csv_multiple_prices = csvFuturesMultiplePricesData()

    instrument_code = input("Instrument code? <return for ALL instruments> ")
    if instrument_code == "":
        instrument_list = csv_multiple_prices.get_list_of_instruments()
    else:
        instrument_list = [instrument_code]

    for instrument_code in instrument_list:
        print(instrument_code)
        multiple_prices = csv_multiple_prices.get_multiple_prices(instrument_code)

        print(multiple_prices)

        db_multiple_prices.add_multiple_prices(
            instrument_code, multiple_prices, ignore_duplication=True
        )