chellow 1759823276.0.0__py3-none-any.whl → 1760028799.0.0__py3-none-any.whl

chellow/e/lcc.py

@@ -46,7 +46,7 @@ def run_import(sess, log, set_progress):
     s = requests.Session()
     s.verify = False
 
-    for mod_name in ("chellow.e.cfd",):
+    for mod_name in ("chellow.e.cfd", "chellow.e.rab"):
         mod = import_module(mod_name)
         mod.lcc_import(sess, log, set_progress, s)
 

chellow/e/rab.py

@@ -0,0 +1,75 @@
+from datetime import datetime as Datetime
+from decimal import Decimal
+
+
+from sqlalchemy import select
+
+from chellow.e.lcc import api_records
+from chellow.models import Contract, RateScript
+from chellow.utils import ct_datetime, to_ct, to_utc
+
+
+def hh(data_source):
+    try:
+        rab_cache = data_source.caches["rab"]
+    except KeyError:
+        rab_cache = data_source.caches["rab"] = {}
+
+    for h in data_source.hh_data:
+        try:
+            h["rab"] = rab_cache[h["start-date"]]
+        except KeyError:
+            h_start = h["start-date"]
+            rate_str = data_source.non_core_rate("rab_forecast_ilr_tra", h_start)[
+                "record"
+            ]["Interim_Levy_Rate_GBP_MWh"]
+            if rate_str == "":
+                base_rate_dec = Decimal("0")
+            else:
+                base_rate_dec = Decimal(rate_str) / Decimal(1000)
+
+            base_rate = float(base_rate_dec)
+
+            h["rab"] = rab_cache[h_start] = {
+                "interim": base_rate,
+            }
+
+
+def lcc_import(sess, log, set_progress, s):
+    import_forecast_ilr_tra(sess, log, set_progress, s)
+
+
+def _parse_date(date_str):
+    return to_utc(to_ct(Datetime.strptime(date_str[:10], "%Y-%m-%d")))
+
+
+def import_forecast_ilr_tra(sess, log, set_progress, s):
+    log("Starting to check for new LCC RAB Forecast ILR TRA")
+
+    contract_name = "rab_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, {}
+        )
+
+    for record in api_records(log, s, "1231fbb3-93ee-4a33-87a9-f15bb377346d"):
+        period_start_str = record["Month"]
+        if len(period_start_str) == 0:
+            continue
+        period_start = _parse_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 RAB Forecast ILR TRA")

{chellow-1759823276.0.0.dist-info → chellow-1760028799.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chellow
-Version: 1759823276.0.0
+Version: 1760028799.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-1759823276.0.0.dist-info → chellow-1760028799.0.0.dist-info}/RECORD

@@ -35,8 +35,9 @@ chellow/e/hh_parser_schneider_xlsx.py,sha256=Vtq0TNz-oojoKJm4PeH4ZwBp2I-mjArB9-F
 chellow/e/hh_parser_simple_csv.py,sha256=lJx9tw9BWFSoBmns1Cws_vY-OIn90LPt2yvIN_CFcTE,2177
 chellow/e/hh_parser_vital_xlsx.py,sha256=g9-CElfH1PPfwpuUcVvD6WQpBlNxCo8j9pq_0Yza0ZM,4125
 chellow/e/lafs.py,sha256=SUUFtvn_IQQTrZue1zefCYgzuscA0FVX2ySZ9u8BDPw,7776
-chellow/e/lcc.py,sha256=OkpynN8_iAdHRlu-yyU6BhRUqYYOZsUnl0HbHULYo_4,4670
+chellow/e/lcc.py,sha256=VqwKBE6N6wXv_xu0o-1CRtJripoFGCZ8Q8e1hvce_34,4686
 chellow/e/mdd_importer.py,sha256=NugJr2JhuzkPTsEMl_5UdQuw5K2p8lVJ-hyz4MK6Hfg,35762
+chellow/e/rab.py,sha256=8buWWqo7CPzAqenI8zEnKDMVxzXZzQZUFAWY9sdGH3s,2320
 chellow/e/rcrc.py,sha256=92CA1uIotIHd1epQ_jEPdJKzXqDFV-AoJOJeRO6MEyA,4274
 chellow/e/ro.py,sha256=cpeJQMY_6SpVxL1nySSj9we58A6E2KwJEAb6fK7cZKY,626
 chellow/e/scenario.py,sha256=FLgh03r_SgXx0hMWFbAvwsz2ScDL8LUwYWSWVv2rQlg,24973
@@ -399,6 +400,6 @@ chellow/templates/g/supply_note_edit.html,sha256=b8mB6_ucBwoljp03iy6AgVaZUhGw3-1
 chellow/templates/g/supply_notes.html,sha256=6epNmZ3NKdXZz27fvmRUGeffg_oc1kmwuBeyRzQe3Rg,854
 chellow/templates/g/unit.html,sha256=KouNVU0-i84afANkLQ_heJ0uDfJ9H5A05PuLqb8iCN8,438
 chellow/templates/g/units.html,sha256=p5Nd-lAIboKPEOO6N451hx1bcKxMg4BDODnZ-43MmJc,441
-chellow-1759823276.0.0.dist-info/METADATA,sha256=FByd2AG_PHGsgs27uxDhVTHa9U52X14rX8HGh4X7aVo,12500
-chellow-1759823276.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-chellow-1759823276.0.0.dist-info/RECORD,,
+chellow-1760028799.0.0.dist-info/METADATA,sha256=zm1x1B_QH5ZI_HiJ4vFBX0RbyndLd_v2iw77jNS4QqU,12500
+chellow-1760028799.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+chellow-1760028799.0.0.dist-info/RECORD,,