Agile Price Batpred yaml

agile_prices.py

from OctopusAgile import Agile
import datetime
import yaml

today = datetime.date.today()

agile = Agile('B') # N is the region code for Southern Scotland
date_from = today + datetime.timedelta(days=1)
date_to = today + datetime.timedelta(days=2)
rates = agile.get_rates(date_from, date_to)

date = (today + datetime.timedelta(days=1)).strftime("%Y-%m-%d")
rates_formatted = []
for time, rate in rates['date_rates'].items():
    time = datetime.datetime.strptime(time, "%Y-%m-%dT%H:%M:%SZ")
    start =  time.strftime("%H:%M:%S")
    end = (time + datetime.timedelta(minutes=30)).strftime("%H:%M:%S")
    rates_formatted.append({"date":date, "start":start, "end":end, "rate":rate})


output = { "rates_import_override":rates_formatted }
print(yaml.dump(output))