PricePower/calculator/calc.py

90 lines
4.6 KiB
Python
Raw Permalink Normal View History

2025-01-16 14:52:15 +00:00
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
)