89 lines
4.6 KiB
Python
89 lines
4.6 KiB
Python
from dataclasses import dataclass
|
|
import datetime
|
|
from typing import List
|
|
|
|
from fastapi import HTTPException
|
|
|
|
from calculator.miner import PriceNotFound, get_energy_prices, get_eur_czk_ratio
|
|
from calculator.schema import BatteryChargingInfo, Price, SpotPrices
|
|
|
|
|
|
def get_spot_prices(date: datetime.date, hour:int, kwh_fees_low:float, kwh_fees_high:float, sell_fees:float, VAT:float, low_tariff_hours:List[int], no_cache: bool = False) -> SpotPrices:
|
|
is_today = datetime.date.today() == date
|
|
|
|
spot_hours = {}
|
|
spot_hours_for_sell = {}
|
|
spot_hours_total = {}
|
|
|
|
spot_data = get_energy_prices(date, no_cache=no_cache)
|
|
currency_ratio = get_eur_czk_ratio(date, no_cache=no_cache)
|
|
|
|
for key, value in spot_data.items():
|
|
kwh_fees = kwh_fees_low if int(key) in low_tariff_hours else kwh_fees_high
|
|
|
|
spot_hours[key] = value * currency_ratio / 1000
|
|
spot_hours_total[key] = (value * currency_ratio / 1000 + kwh_fees) * VAT
|
|
spot_hours_for_sell[key] = value * currency_ratio / 1000 - sell_fees
|
|
|
|
spot = Price(hours=spot_hours, now=spot_hours[str(hour)] if is_today else None)
|
|
|
|
spot_hours_total_sorted = {k: v for k, v in sorted(spot_hours_total.items(), key=lambda item: item[1])}
|
|
spot_total = Price(hours=spot_hours_total_sorted, now=spot_hours_total[str(hour)] if is_today else None)
|
|
spot_for_sell = Price(hours=spot_hours_for_sell, now=spot_hours_for_sell[str(hour)] if is_today else None)
|
|
|
|
return SpotPrices(
|
|
spot=spot,
|
|
spot_hours_total_sorted=Price(hours=spot_hours_total_sorted, now=spot_hours_total[str(hour)] if is_today else None),
|
|
spot_total=spot_total,
|
|
spot_for_sell=spot_for_sell,
|
|
)
|
|
|
|
|
|
def battery_charging_info(kwh_fees_low:float, kwh_fees_high:float, sell_fees:float, VAT:float, low_tariff_hours:List[int], no_cache: bool = False, battery_kwh_price:float=2.5) -> BatteryChargingInfo:
|
|
today = datetime.date.today()
|
|
hour = datetime.datetime.now().hour
|
|
tomorrow = today + datetime.timedelta(days=1)
|
|
|
|
spot_prices_today:SpotPrices = get_spot_prices(today, hour, kwh_fees_low, kwh_fees_high, sell_fees, VAT, low_tariff_hours, no_cache)
|
|
|
|
# average4hours = sum(list(spot_prices_today.spot_hours_total_sorted.hours.values())[0:4]) / 4
|
|
max_cheapest_hour = max(list(spot_prices_today.spot_hours_total_sorted.hours.values())[0:4])
|
|
average4expensive_hours = sum(list(spot_prices_today.spot_hours_total_sorted.hours.values())[20:24]) / 4
|
|
max_most_expensive_hour = max(
|
|
[x[1] for x in list(spot_prices_today.spot_hours_total_sorted.hours.items())[0:20]]
|
|
) if spot_prices_today.spot_hours_total_sorted.hours else 0
|
|
diff = max_most_expensive_hour - max_cheapest_hour
|
|
|
|
charging_hours = [int(k) for k, v in spot_prices_today.spot_hours_total_sorted.hours.items()][0:4]
|
|
discharging_hours = [int(k) for k, v in spot_prices_today.spot_hours_total_sorted.hours.items() if v > (max_cheapest_hour + battery_kwh_price)]
|
|
|
|
# Add charging hours if the price is just 10% above the most expensive charging hour
|
|
if charging_hours:
|
|
for h in range(5,24):
|
|
value = spot_prices_today.spot_hours_total_sorted.hours[str(h)]
|
|
if value <= max_cheapest_hour*1.1:
|
|
charging_hours.append(h)
|
|
|
|
# We remove end of the day hours from charging hours if the average of the last 4 hours is higher than the average of the first 4 hours of the next day
|
|
try:
|
|
spot_prices_tomorrow:SpotPrices = get_spot_prices(tomorrow, hour, kwh_fees_low, kwh_fees_high, sell_fees, VAT, low_tariff_hours, no_cache)
|
|
average_last_4hours_today = sum(list(spot_prices_today.spot_hours_total_sorted.hours.values())[20:24]) / 4
|
|
average_first_4hours_tomorrow = sum(list(spot_prices_tomorrow.spot_hours_total_sorted.hours.values())[0:4]) / 4
|
|
if average_last_4hours_today > average_first_4hours_tomorrow:
|
|
for h in range(20,24):
|
|
if h in charging_hours:
|
|
charging_hours.remove(h)
|
|
except PriceNotFound:
|
|
pass
|
|
|
|
is_viable = len(discharging_hours) > 0
|
|
|
|
return BatteryChargingInfo(
|
|
diff=diff,
|
|
is_viable=is_viable,
|
|
charging_hours=sorted(charging_hours) if len(charging_hours) > 0 else [],
|
|
is_charging_hour=hour in charging_hours if len(charging_hours) > 0 and is_viable else False,
|
|
discharging_hours=sorted(discharging_hours) if len(discharging_hours) > 0 else [],
|
|
is_discharging_hour=hour in discharging_hours if len(discharging_hours) > 0 and is_viable else False,
|
|
total_price=spot_prices_today.spot_hours_total_sorted
|
|
)
|