chellow 1705585731.0.0__tar.gz → 1706020835.0.0__tar.gz

{chellow-1705585731.0.0 → chellow-1706020835.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chellow
-Version: 1705585731.0.0
+Version: 1706020835.0.0
 Summary: Web Application for checking UK energy bills.
 Project-URL: Homepage, https://github.com/WessexWater/chellow
 Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)

chellow-1706020835.0.0/VERSION

@@ -0,0 +1 @@
+__version__ = "1706020835.0.0"

{chellow-1705585731.0.0 → chellow-1706020835.0.0}/chellow/e/cfd.py

@@ -1,14 +1,17 @@
 from datetime import datetime as Datetime
 
 from dateutil.relativedelta import relativedelta
+
 from sqlalchemy import null, or_, select
 
+from werkzeug.exceptions import BadRequest
+
 from chellow.e.lcc import api_search, api_sql
 from chellow.models import Contract, RateScript
 from chellow.utils import ct_datetime, hh_format, to_ct, to_utc
 
 
-def hh(data_source):
+def hh(data_source, use_bill_check=False):
     try:
         cfd_cache = data_source.caches["cfd"]
     except KeyError:
@@ -19,47 +22,60 @@ def hh(data_source):
             h["cfd-rate"] = cfd_cache[h["start-date"]]
         except KeyError:
             h_start = h["start-date"]
-            daily_contract = Contract.get_non_core_by_name(
-                data_source.sess, "cfd_reconciled_daily_levy_rates"
-            )
-            daily_rs = data_source.sess.scalars(
-                select(RateScript).where(
-                    RateScript.contract == daily_contract,
-                    RateScript.start_date <= h_start,
-                    or_(
-                        RateScript.finish_date == null(),
-                        RateScript.finish_date >= h_start,
-                    ),
+            if use_bill_check:
+                base_rate = (
+                    float(
+                        data_source.non_core_rate("cfd_forecast_ilr_tra", h_start)[
+                            "record"
+                        ]["Interim_Levy_Rate_GBP_Per_MWh"]
+                    )
+                    / 1000
                 )
-            ).one_or_none()
-            if daily_rs.finish_date is None and h_start >= to_utc(
-                to_ct(daily_rs.start_date) + relativedelta(months=3)
-            ):
-                base_rate = data_source.non_core_rate(
-                    "cfd_in_period_tracking", h_start
-                )["rate_gbp_per_kwh"]
             else:
-                base_rate = data_source.non_core_rate(
-                    "cfd_reconciled_daily_levy_rates", h_start
-                )["rate_gbp_per_kwh"]
+                daily_contract = Contract.get_non_core_by_name(
+                    data_source.sess, "cfd_reconciled_daily_levy_rates"
+                )
+                daily_rs = data_source.sess.scalars(
+                    select(RateScript).where(
+                        RateScript.contract == daily_contract,
+                        RateScript.start_date <= h_start,
+                        or_(
+                            RateScript.finish_date == null(),
+                            RateScript.finish_date >= h_start,
+                        ),
+                    )
+                ).one_or_none()
+                if daily_rs.finish_date is None and h_start >= to_utc(
+                    to_ct(daily_rs.start_date) + relativedelta(months=3)
+                ):
+                    base_rate_dec = data_source.non_core_rate(
+                        "cfd_in_period_tracking", h_start
+                    )["rate_gbp_per_kwh"]
+                else:
+                    base_rate_dec = data_source.non_core_rate(
+                        "cfd_reconciled_daily_levy_rates", h_start
+                    )["rate_gbp_per_kwh"]
+                base_rate = float(base_rate_dec)
 
             effective_ocl_rate = data_source.non_core_rate(
                 "cfd_operational_costs_levy", h["start-date"]
             )["record"]["Effective_OCL_Rate_GBP_Per_MWh"]
             if effective_ocl_rate == "":
                 levy_rate_str = data_source.non_core_rate(
-                    "cfd_operational_costs_levy", h["start-date"]
+                    "cfd_operational_costs_levy", h_start
                 )["record"]["OCL_Rate_GBP_Per_MWh"]
             else:
                 levy_rate_str = effective_ocl_rate
             levy_rate = float(levy_rate_str) / 1000
-            h["cfd-rate"] = cfd_cache[h["start-date"]] = float(base_rate) + levy_rate
+
+            h["cfd-rate"] = cfd_cache[h_start] = base_rate + levy_rate
 
 
 def lcc_import(sess, log, set_progress, s):
     import_in_period_tracking(sess, log, set_progress, s)
     import_operational_costs_levy(sess, log, set_progress, s)
     import_reconciled_daily_levy_rates(sess, log, set_progress, s)
+    import_forecast_ilr_tra(sess, log, set_progress, s)
 
 
 def _quarters(s):
@@ -93,6 +109,16 @@ def _parse_date(date_str):
     return to_utc(to_ct(Datetime.strptime(date_str[:10], "%Y-%m-%d")))
 
 
+def _parse_varying_date(date_str):
+    if "-" in date_str:
+        pattern = "%Y-%m-%d"
+    elif "/" in date_str:
+        pattern = "%d/%m/%Y"
+    else:
+        raise BadRequest(f"The date {date_str} is not recognized.")
+    return Datetime.strptime(date_str, pattern)
+
+
 def import_in_period_tracking(sess, log, set_progress, s):
     log("Starting to check for new LCC CfD In-Period Tracking")
 
@@ -303,3 +329,34 @@ def import_reconciled_daily_levy_rates(sess, log, set_progress, s):
 
     log("Finished LCC CfD Reconciled Daily Levy Rates")
     sess.commit()
+
+
+def import_forecast_ilr_tra(sess, log, set_progress, s):
+    log("Starting to check for new LCC CfD Forecast ILR TRA")
+
+    contract_name = "cfd_forecast_ilr_tra"
+    contract = Contract.find_non_core_by_name(sess, contract_name)
+    if contract is None:
+        contract = Contract.insert_non_core(
+            sess, contract_name, "", {}, to_utc(ct_datetime(1996, 4, 1)), None, {}
+        )
+
+    res_j = api_search(s, "fbece4ce-7cfc-42b7-8fb2-387cf59a3c32", sort="Period_Start")
+    for record in res_j["result"]["records"]:
+        period_start_str = record["Period_Start"]
+        period_start = to_utc(to_ct(_parse_varying_date(period_start_str)))
+
+        rs = sess.execute(
+            select(RateScript).where(
+                RateScript.contract == contract,
+                RateScript.start_date == period_start,
+            )
+        ).scalar_one_or_none()
+        if rs is None:
+            rs = contract.insert_rate_script(sess, period_start, {})
+
+        rs_script = rs.make_script()
+        rs_script["record"] = record
+        rs.update(rs_script)
+        sess.commit()
+    log("Finished LCC CfD Forecast ILR TRA")

{chellow-1705585731.0.0 → chellow-1706020835.0.0}/chellow/e/triad.py

@@ -8,6 +8,8 @@ from pypdf import PdfReader
 
 from sqlalchemy import null, or_, select
 
+from werkzeug.exceptions import BadRequest
+
 import chellow.e.duos
 from chellow.models import Contract, RateScript
 from chellow.national_grid import api_get
@@ -276,6 +278,16 @@ GSP_LOOKUP = {
 }
 
 
+def _parse_varying_date(date_str):
+    if "-" in date_str:
+        pattern = "%Y-%m-%d"
+    elif "/" in date_str:
+        pattern = "%d/%m/%Y"
+    else:
+        raise BadRequest(f"The date {date_str} is not recognized.")
+    return Datetime.strptime(date_str, pattern)
+
+
 def national_grid_import(sess, log, set_progress, s):
     log("Starting to check for new TNUoS TRIAD Tariffs")
 
@@ -317,10 +329,13 @@ def national_grid_import(sess, log, set_progress, s):
                 rates = tr[polarity] = {}
 
             record_key = GSP_LOOKUP[record["Zone_No"]]
-            record_published_date = record["Published_Date"]
+            record_published_date = _parse_varying_date(record["Published_Date"])
 
             rate = rates.get(record_key)
-            if rate is None or rate["Published_Date"] > record_published_date:
+            if (
+                rate is None
+                or _parse_varying_date(rate["Published_Date"]) < record_published_date
+            ):
                 rates[record_key] = record
                 rs.update(rs_script)
                 sess.commit()

chellow-1705585731.0.0/VERSION

@@ -1 +0,0 @@
-__version__ = "1705585731.0.0"