chellow 1754662596.0.0__py3-none-any.whl → 1755011219.0.0__py3-none-any.whl

chellow/e/bill_parsers/drax_edi.py

@@ -30,6 +30,8 @@ ELEMENT_MAP = {
     "ADH": {
         "ADHOC": {
             "020330": ("eii-gbp", None, None),
+            "064305": ("fit-gbp", None, None),
+            "590346": ("cfd-operational-gbp", None, None),
             "637050": ("meter-rental-gbp", "meter-rental-rate", "meter-rental-days"),
         },
     },
@@ -88,6 +90,8 @@ ELEMENT_MAP = {
             "209269": ("tnuos-gbp", "tnuos-rate", "tnuos-days"),
         },
         "DUS002": {
+            "185913": ("duos-yellow-gbp", "duos-yellow-rate", "duos-yellow-kwh"),
+            "517270": ("duos-black-gbp", "duos-black-rate", "duos-black-kwh"),
             "797790": (
                 "duos-reactive-gbp",
                 "duos-reactive-rate",
@@ -147,13 +151,6 @@ ELEMENT_MAP = {
             "012069": ("triad-gbp", "triad-rate", "triad-kw"),
         },
     },
-    "064305": ("fit-gbp", None, None),
-    "590346": ("cfd-operational-gbp", None, None),
-    "439724": ("eii-gbp", None, None),
-    "247610": ("eii-gbp", None, None),
-    "930504": ("eii-gbp", None, None),
-    "331201": ("eii-gbp", None, None),
-    "307253": ("eii-gbp", "eii-rate", "eii-kwh"),
 }
 
 TPR_LOOKUP = {

chellow/e/dno_rate_parser.py

@@ -145,6 +145,204 @@ BAND_WEEKEND = {
     "notes": None,
 }
 
+PERIODS = {
+    "monday to friday (including bank holidays) april to march inclusive": [
+        {
+            "weekend": False,
+            "start-month": 4,
+            "start-day": 1,
+            "finish-month": 12,
+            "finish-day": 31,
+        },
+        {
+            "weekend": False,
+            "start-month": 1,
+            "start-day": 1,
+            "finish-month": 3,
+            "finish-day": 31,
+        },
+    ],
+    "monday to friday (including bank holidays) april to october inclusive and march": [
+        {
+            "weekend": False,
+            "start-month": 4,
+            "start-day": 1,
+            "finish-month": 10,
+            "finish-day": 31,
+        },
+        {
+            "weekend": False,
+            "start-month": 3,
+            "start-day": 1,
+            "finish-month": 3,
+            "finish-day": 31,
+        },
+    ],
+    "monday to friday (including bank holidays) june to august inclusive": [
+        {
+            "weekend": False,
+            "start-month": 6,
+            "start-day": 1,
+            "finish-month": 8,
+            "finish-day": 31,
+        },
+    ],
+    "monday to friday (including bank holidays) november to february inclusive": [
+        {
+            "weekend": False,
+            "start-month": 11,
+            "start-day": 1,
+            "finish-month": 12,
+            "finish-day": 31,
+        },
+        {
+            "weekend": False,
+            "start-month": 1,
+            "start-day": 1,
+            "finish-month": 2,
+            "finish-day": "last",
+        },
+    ],
+    "monday to friday (including bank holidays) november to february inclusive "
+    "(excluding 22nd dec to 4th jan inclusive)": [
+        {
+            "weekend": False,
+            "start-month": 11,
+            "start-day": 1,
+            "finish-month": 12,
+            "finish-day": 21,
+        },
+        {
+            "weekend": False,
+            "start-month": 1,
+            "start-day": 5,
+            "finish-month": 2,
+            "finish-day": "last",
+        },
+    ],
+    "monday to friday (including bank holidays) march to october inclusive": [
+        {
+            "weekend": False,
+            "start-month": 3,
+            "start-day": 1,
+            "finish-month": 10,
+            "finish-day": 31,
+        },
+    ],
+    "monday to friday (including bank holidays) mar to oct inclusive": [
+        {
+            "weekend": False,
+            "start-month": 3,
+            "start-day": 1,
+            "finish-month": 10,
+            "finish-day": 31,
+        },
+    ],
+    "monday to friday (including bank holidays) nov to feb inclusive": [
+        {
+            "weekend": False,
+            "start-month": 11,
+            "start-day": 1,
+            "finish-month": 12,
+            "finish-day": 31,
+        },
+        {
+            "weekend": False,
+            "start-month": 1,
+            "start-day": 1,
+            "finish-month": 2,
+            "finish-day": "last",
+        },
+    ],
+    (
+        "monday to friday nov to feb (excluding 22nd dec to 4th jan inclusive)"
+        "monday to friday (including bank holidays) nov to feb inclusive "
+        "(excluding 22nd dec to 4th jan inclusive)"
+    ): [
+        {
+            "weekend": False,
+            "start-month": 11,
+            "start-day": 1,
+            "finish-month": 12,
+            "finish-day": 21,
+        },
+        {
+            "weekend": False,
+            "start-month": 1,
+            "start-day": 5,
+            "finish-month": 2,
+            "finish-day": "last",
+        },
+    ],
+    (
+        "monday to friday (including bank holidays) "
+        "mar to oct inclusive (plus 22nd dec to 4th jan inclusive)"
+    ): [
+        {
+            "weekend": False,
+            "start-month": 3,
+            "start-day": 1,
+            "finish-month": 10,
+            "finish-day": 31,
+        },
+        {
+            "weekend": False,
+            "start-month": 12,
+            "start-day": 22,
+            "finish-month": 12,
+            "finish-day": 31,
+        },
+        {
+            "weekend": False,
+            "start-month": 1,
+            "start-day": 1,
+            "finish-month": 1,
+            "finish-day": 4,
+        },
+    ],
+    (
+        "monday to friday "
+        "(including bank holidays) "
+        "nov to feb inclusive (excluding 22nd dec to 4th jan inclusive)"
+    ): [
+        {
+            "weekend": False,
+            "start-month": 11,
+            "start-day": 1,
+            "finish-month": 12,
+            "finish-day": 21,
+        },
+        {
+            "weekend": False,
+            "start-month": 1,
+            "start-day": 4,
+            "finish-month": 2,
+            "finish-day": "last",
+        },
+    ],
+    "saturday and sunday all year": [
+        {
+            "weekend": True,
+            "start-month": 1,
+            "start-day": 1,
+            "finish-month": 12,
+            "finish-day": 31,
+        }
+    ],
+    (
+        "monday to friday (including bank holidays) november to february "
+        "monday to friday nov to feb"
+    ): [
+        {
+            "weekend": False,
+            "start-month": 11,
+            "start-day": 1,
+            "finish-month": 2,
+            "finish-day": "last",
+        }
+    ],
+}
+
 
 def str_to_hr(hr_str):
     for sep in (":", "."):
@@ -209,6 +407,7 @@ def tab_lv_hv(sheet, gsp_rates):
         tariffs = gsp_rates["tariffs"] = {}
 
     bands = gsp_rates["bands"] = []
+    ums_bands = gsp_rates["ums_bands"] = []
 
     class State(Enum):
         OUTSIDE = auto()
@@ -236,15 +435,31 @@ def tab_lv_hv(sheet, gsp_rates):
 
             if is_weekend is None:
                 state = State.OUTSIDE
-            else:
-                for i, band_name in enumerate(("red", "amber")):
-                    for slot in val_to_slots(get_value(row, i + 1)):
-                        bands.append(
+                continue
+
+            for i, band_name in enumerate(("red", "amber")):
+                for slot in val_to_slots(get_value(row, i + 1)):
+                    bands.append(
+                        {
+                            "weekend": is_weekend,
+                            "start": slot["start"],
+                            "finish": slot["finish"],
+                            "band": band_name,
+                        }
+                    )
+
+            val_6 = " ".join(get_value(row, 6).split()).lower()
+            periods = PERIODS[val_6]
+
+            for i, band_name in enumerate(("black", "yellow")):
+                for slot in val_to_slots(get_value(row, i + 8)):
+                    for period in periods:
+                        ums_bands.append(
                             {
-                                "weekend": is_weekend,
-                                "start": slot["start"],
-                                "finish": slot["finish"],
                                 "band": band_name,
+                                **period,
+                                "start-decimal-hour": slot["start"],
+                                "finish-decimal-hour": slot["finish"],
                             }
                         )
 
@@ -255,6 +470,9 @@ def tab_lv_hv(sheet, gsp_rates):
                 llfcs_str = ",".join(
                     chain(to_llfcs(get_value(row, 1)), to_llfcs(get_value(row, 10)))
                 )
+                is_ums = "unmetered supplies" in val_0.lower()
+                high_band = "black" if is_ums else "red"
+                medium_band = "yellow" if is_ums else "amber"
                 pcs_str = ",".join(to_pcs(row, 2))
                 tariffs[llfcs_str + "_" + pcs_str] = {
                     "description": val_0,
@@ -267,8 +485,10 @@ def tab_lv_hv(sheet, gsp_rates):
                     "excess-gbp-per-kva-per-day": get_zero_rate(
                         row, col_match(title_row, "exce")
                     ),
-                    "red-gbp-per-kwh": get_rag_rate(row, col_match(title_row, "red")),
-                    "amber-gbp-per-kwh": get_rag_rate(
+                    f"{high_band}-gbp-per-kwh": get_rag_rate(
+                        row, col_match(title_row, "red")
+                    ),
+                    f"{medium_band}-gbp-per-kwh": get_rag_rate(
                         row, col_match(title_row, "amber")
                     ),
                     "green-gbp-per-kwh": get_rag_rate(
@@ -344,54 +564,15 @@ def tab_ehv(sheet, gsp_rates):
                 state = EHV_BLANK
             else:
                 period_str = " ".join(get_value(row, 0).split()).lower()
-                periods = []
-
-                if period_str in (
-                    "monday to friday nov to feb (excluding "
-                    "22nd dec to 4th jan inclusive)",
-                    "monday to friday (including bank holidays) nov to feb inclusive "
-                    "(excluding 22nd dec to 4th jan inclusive)",
-                ):
-                    periods.append(
-                        {
-                            "weekend": False,
-                            "start-month": 11,
-                            "start-day": 1,
-                            "finish-month": 12,
-                            "finish-day": 21,
-                        }
-                    )
-                    periods.append(
-                        {
-                            "weekend": False,
-                            "start-month": 1,
-                            "start-day": 5,
-                            "finish-month": 2,
-                            "finish-day": "last",
-                        }
-                    )
-
-                elif period_str in (
-                    "monday to friday (including bank holidays) november to february",
-                    "monday to friday nov to feb",
-                ):
-                    periods.append(
-                        {
-                            "weekend": False,
-                            "start-month": 11,
-                            "start-day": 1,
-                            "finish-month": 2,
-                            "finish-day": "last",
-                        }
-                    )
+                periods = PERIODS[period_str]
 
                 for slot in val_to_slots(get_value(row, 3)):
                     for period in periods:
                         bands.append(
                             {
                                 **period,
-                                "start-hour": slot["start"],
-                                "finish-hour": slot["finish"],
+                                "start-decimal-hour": slot["start"],
+                                "finish-decimal-hour": slot["finish"],
                             }
                         )
 

chellow/e/duos.py

@@ -16,20 +16,17 @@ from chellow.utils import (
 )
 
 
-BANDS = ("super-red", "red", "amber", "green")
-
-KEYS = dict(
-    (
-        band,
-        {
-            "kwh": f"duos-{band}-kwh",
-            "tariff-rate": f"{band}-gbp-per-kwh",
-            "bill-rate": f"duos-{band}-rate",
-            "gbp": f"duos-{band}-gbp",
-        },
-    )
+BANDS = ("black", "super-red", "red", "amber", "yellow", "green")
+
+KEYS = {
+    band: {
+        "kwh": f"duos-{band}-kwh",
+        "tariff-rate": f"{band}-gbp-per-kwh",
+        "bill-rate": f"duos-{band}-rate",
+        "gbp": f"duos-{band}-gbp",
+    }
     for band in BANDS
-)
+}
 
 VL_LOOKUP = {
     "LV": {True: "lv-sub", False: "lv-net"},
@@ -706,6 +703,58 @@ def datum_2012_02_23(ds, hh):
                         band = "super-red"
                         break
 
+                bands_cache[start_date] = band
+    elif ds.dtc_meter_type_code is None:  # unmetered
+        try:
+            band = gsp_group_cache["ums-bands"][start_date]
+        except KeyError:
+            try:
+                bands_cache = gsp_group_cache["ums-bands"]
+            except KeyError:
+                bands_cache = gsp_group_cache["ums-bands"] = {}
+
+            try:
+                band = bands_cache[start_date]
+            except KeyError:
+                band = "green"
+                ct_hr = hh["ct-decimal-hour"]
+                ct_day = hh["ct-day"]
+                ct_month = hh["ct-month"]
+                weekend = hh["ct-day-of-week"] > 4
+                try:
+                    slots = ds.hh_rate(ds.dno_contract.id, start_date)[
+                        ds.gsp_group_code
+                    ]["ums_bands"]
+                except KeyError as e:
+                    raise BadRequest(str(e))
+
+                for slot in slots:
+                    if (
+                        slot["weekend"] == weekend
+                        and slot["start-decimal-hour"]
+                        <= ct_hr
+                        < slot["finish-decimal-hour"]
+                        and (
+                            ct_month > slot["start-month"]
+                            or (
+                                ct_month == slot["start-month"]
+                                and ct_day >= slot["start-day"]
+                            )
+                        )
+                        and (
+                            ct_month < slot["finish-month"]
+                            or (
+                                ct_month == slot["finish-month"]
+                                and (
+                                    slot["finish-day"] == "last"
+                                    or ct_day <= slot["finish-day"]
+                                )
+                            )
+                        )
+                    ):
+                        band = slot["band"]
+                        break
+
                 bands_cache[start_date] = band
     else:
 

chellow/reports/report_111.py

@@ -321,9 +321,9 @@ def _process_supply(
     while len(bill_ids) > 0:
         bill_id = list(sorted(bill_ids))[0]
         bill_ids.remove(bill_id)
-        bill = (
-            sess.query(Bill)
-            .filter(Bill.id == bill_id)
+        bill = sess.scalar(
+            select(Bill)
+            .where(Bill.id == bill_id)
             .options(
                 joinedload(Bill.batch),
                 joinedload(Bill.bill_type),
@@ -332,7 +332,6 @@ def _process_supply(
                 joinedload(Bill.reads).joinedload(RegisterRead.present_type),
                 joinedload(Bill.reads).joinedload(RegisterRead.previous_type),
             )
-            .one()
         )
         virtual_bill = {"problem": ""}
         supply = bill.supply

{chellow-1754662596.0.0.dist-info → chellow-1755011219.0.0.dist-info}/METADATA

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

@@ -22,8 +22,8 @@ chellow/e/bsuos.py,sha256=2L7pgiGbV4k1Udfzjj4B2Qye02sV_FhNjwh7_sIxXFg,15525
 chellow/e/ccl.py,sha256=30dh_SvlgzsTQPPAJNZWILaMvbeDsv9-P-S1JxS5_SQ,3184
 chellow/e/cfd.py,sha256=CWLdYeNjgqT6Ro8YRf4vhwXIAJ2aV4Wi6HLNClVSeaQ,14260
 chellow/e/computer.py,sha256=lt4Zlr8EOClL1acuuvKA-tNPCI5RRHnAZbsaGQnB9j4,68835
-chellow/e/dno_rate_parser.py,sha256=EO3uIYD0uiT8uumI5wnODMsFwYGrdB5HexuOu-2aFkw,21875
-chellow/e/duos.py,sha256=RHrn93I1ASO2uYkuF18qlhG4p-jpuJhd_g3o69wtP4U,31004
+chellow/e/dno_rate_parser.py,sha256=PRJnRHsoMdNQTzXfHet8PptXOcNUURKhVpwd9Dzt2Q4,26503
+chellow/e/duos.py,sha256=3jOOnRP633xfzC7OhqWWW48H34T-NZosss9CNW6HQQs,33004
 chellow/e/elexon.py,sha256=eOmX_wpxj6Ty6eaEgGAgXX-MgyJkYQSAT6DtJeP1rUE,5544
 chellow/e/energy_management.py,sha256=tRQJ-7_caG91oKnh5qIuK9qE_Nr3BmZ3ddaj5GTHIKQ,10487
 chellow/e/hh_importer.py,sha256=JBVCNSpM-DMu9_io516KojvujwZEQq6upDSsMbtu_sI,21511
@@ -50,7 +50,7 @@ chellow/e/bill_parsers/activity_mop_stark_xlsx.py,sha256=opjXRrqrgBTbSKzL0JfTLP0
 chellow/e/bill_parsers/annual_mop_stark_xlsx.py,sha256=-HMoIfa_utXYKA44RuC0Xqv3vd2HLeQU_4P0iBUd3WA,4219
 chellow/e/bill_parsers/bgb_edi.py,sha256=GuwHeYbAGk7BVg5n19FcTANFDyKI-y0z3f9niQaPSSw,4828
 chellow/e/bill_parsers/csv.py,sha256=U5zcIaZ6B5QTTpFDAcBnk4G2r8B3j5kJhDPL4AJNkEk,5640
-chellow/e/bill_parsers/drax_edi.py,sha256=mzYaPYMVj2dFb1HIGV-HFtqYjhZ2g9V3xl_w8Z1ECfc,15400
+chellow/e/bill_parsers/drax_edi.py,sha256=TPfFAM3z4CUsgIKT4B7Tc87urVzidklEFiZjQ3UgaVM,15371
 chellow/e/bill_parsers/drax_element_edi.py,sha256=kxjg1KA4UheMa8doGp9skXtQYrNK8Eisy9ksafR5TQo,13661
 chellow/e/bill_parsers/edf_export_xlsx.py,sha256=J4lY8epiSTIePZ6D1SGD76TXRoev35KrUC2sjHkNqlE,6632
 chellow/e/bill_parsers/engie_edi.py,sha256=yC5Tz5xFIcJmKZbvK_xcl0s-1fZsse1z4HEbwpC7fxU,15834
@@ -79,7 +79,7 @@ chellow/gas/transportation.py,sha256=Bkg8TWOs-v0ES-4qqwbleiOhqbE_t2KauUx9JYMZELM
 chellow/gas/views.py,sha256=GeCvi6BGTUN7bu7sVkypNckwG3Crl6AbUcRob9qMi0E,59733
 chellow/reports/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chellow/reports/report_109.py,sha256=Exb-FQ5f70-ier_h15CgHGysQ7vJ7k3gFZ1001zM3iM,11171
-chellow/reports/report_111.py,sha256=_ipuMuvhjWLBVjK4wV08tbDG6s8cspQ_bzRZu1mPuXE,30224
+chellow/reports/report_111.py,sha256=LAWmCLVE5yhPTVV_agZtlTyXHyNmJVes2zL7CS9eITo,30211
 chellow/reports/report_169.py,sha256=wortJyvgXTrnjjXhLzqwaksKKu63K5IgdzU5f4SkHMI,9246
 chellow/reports/report_181.py,sha256=ypfYWYVNC1X2fMlJyDzwNiNTTRrpPW49DirXE_xDKa0,4958
 chellow/reports/report_183.py,sha256=fGONpKEIXTqIT_3dE5jZKBIesWPgLq-chHO6X22dIv0,8768
@@ -386,6 +386,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-1754662596.0.0.dist-info/METADATA,sha256=aX-vVmKqbNRf1rplpPv_hkiUpx6Bt0bFszWCFmaBh5k,12519
-chellow-1754662596.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-chellow-1754662596.0.0.dist-info/RECORD,,
+chellow-1755011219.0.0.dist-info/METADATA,sha256=LR04CnNjbSXoLGxeZWwX2C97Zj3DBdM8lNiUR4t21pg,12519
+chellow-1755011219.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+chellow-1755011219.0.0.dist-info/RECORD,,