import powerstand
psm = powerstand.init()
# psm = powerstand.init_test()

# print("Сила ветра:", psm.wind.now)
print("Фактические контракты:")
for receipt in psm.exchange:
    print("Контрагент:", receipt.agent)
    # None = оператор, иначе это другой игрок
    print("Объём:", receipt.flux)
    # Плюс = покупка, минус = продажа
    print("Цена за МВт:", receipt.price)
    print("")

#psm.orders.line_on("m5", 1)

price_sell = 0.9
price_buy = 0.8


if psm.total_power.generated-psm.total_power.consumed < 0:
    psm.orders.discharge("M3", min(10, psm.total_power.generated-psm.total_power.consumed))

if psm.total_power.generated-psm.total_power.consumed < 0:
    if 4*(psm.total_power.consumed - psm.total_power.generated) > (psm.total_power.consumed-psm.total_power.generated)*(receipt.price + price_buy):
        psm.orders.buy(abs(psm.total_power.generated-psm.total_power.consumed), receipt.price + price_buy)
        if psm.total_power.generated-psm.total_power.consumed < 0:
            psm.orders.diesel("M3", min(5, abs(psm.total_power.generated-psm.total_power.consumed)))
    else:
        psm.orders.diesel("M3", min(5, abs(psm.total_power.generated-psm.total_power.consumed)))
        if psm.total_power.generated - psm.total_power.consumed < 0:
            psm.orders.buy(abs(psm.total_power.generated-psm.total_power.consumed), receipt.price + price_buy)
elif psm.total_power.generated-psm.total_power.consumed > 0:
    if psm.total_power.generated-psm.total_power.consumed > 10:
        psm.orders.charge("M3", 10)
        psm.orders.sell(psm.total_power.generated-psm.total_power.consumed, receipt.price - price_sell)
    else:
        temp = (psm.total_power.generated-psm.total_power.consumed)/2
        psm.orders.charge("M3", temp)
        psm.orders.sell(temp, receipt.price - price_sell)

if psm.total_power.generated-psm.total_power.consumed < 0:
    psm.orders.buy(abs(psm.total_power.generated-psm.total_power.consumed), receipt.price + price_buy)

psm.save_and_exit()
# print("\n".join(psm.orders.humanize()))