chellow 1725528079.0.0__py3-none-any.whl → 1725628177.0.0__py3-none-any.whl

chellow/e/views.py

@@ -5898,11 +5898,23 @@ def supply_months_get(supply_id):
 @e.route("/supplies/<int:supply_id>/edit")
 def supply_edit_get(supply_id):
     supply = Supply.get_by_id(g.sess, supply_id)
-    sources = g.sess.query(Source).order_by(Source.code)
-    generator_types = g.sess.query(GeneratorType).order_by(GeneratorType.code)
-    gsp_groups = g.sess.query(GspGroup).order_by(GspGroup.code)
-    eras = (
-        g.sess.query(Era).filter(Era.supply == supply).order_by(Era.start_date.desc())
+    sources = g.sess.scalars(select(Source).order_by(Source.code))
+    generator_types = g.sess.scalars(select(GeneratorType).order_by(GeneratorType.code))
+
+    dno_contract = Contract.get_dno_by_name(g.sess, supply.dno.dno_code)
+    dno_rate_script = dno_contract.find_rate_script_at(
+        g.sess, supply.eras[-1].start_date
+    )
+    dno_props = dno_rate_script.make_script()
+    allowed_gsp_group_codes = list(dno_props.keys())
+
+    gsp_groups = g.sess.scalars(
+        select(GspGroup)
+        .where(GspGroup.code.in_(allowed_gsp_group_codes))
+        .order_by(GspGroup.code)
+    )
+    eras = g.sess.scalars(
+        select(Era).where(Era.supply == supply).order_by(Era.start_date.desc())
     )
     return render_template(
         "supply_edit.html",
@@ -5914,50 +5926,127 @@ def supply_edit_get(supply_id):
     )
 
 
-@e.route("/supplies/<int:supply_id>/edit", methods=["POST"])
-def supply_edit_post(supply_id):
+@e.route("/supplies/<int:supply_id>/edit", methods=["DELETE"])
+def supply_edit_delete(supply_id):
     try:
         supply = Supply.get_by_id(g.sess, supply_id)
 
-        if "delete" in request.form:
-            site_id = None
-            for site_era in supply.eras[-1].site_eras:
-                if site_era.is_physical:
-                    site_id = site_era.site.id
-                    break
+        site_id = None
+        for site_era in supply.eras[-1].site_eras:
+            if site_era.is_physical:
+                site_id = site_era.site.id
+                break
 
-            supply.delete(g.sess)
-            g.sess.commit()
-            return credirect(f"/sites/{site_id}", 303)
-        elif "insert_era" in request.form:
-            start_date = req_date("start")
-            supply.insert_era_at(g.sess, start_date)
-            g.sess.commit()
-            return chellow_redirect(f"/supplies/{supply.id}", 303)
+        supply.delete(g.sess)
+        g.sess.commit()
+        return hx_redirect(f"/sites/{site_id}", 303)
+    except BadRequest as e:
+        g.sess.rollback()
+        flash(e.description)
+        sources = g.sess.scalars(select(Source).order_by(Source.code))
+        generator_types = g.sess.scalars(
+            select(GeneratorType).order_by(GeneratorType.code)
+        )
+        dno_contract = Contract.get_dno_by_name(g.sess, supply.dno.dno_code)
+        dno_rate_script = dno_contract.find_rate_script_at(
+            g.sess, supply.eras[-1].start_date
+        )
+        dno_props = dno_rate_script.make_script()
+        allowed_gsp_group_codes = list(dno_props.keys())
+
+        gsp_groups = g.sess.scalars(
+            select(GspGroup)
+            .where(GspGroup.code.in_(allowed_gsp_group_codes))
+            .order_by(GspGroup.code)
+        )
+        eras = g.sess.scalars(
+            select(Era).where(Era.supply == supply).order_by(Era.start_date.desc())
+        )
+        return make_response(
+            render_template(
+                "supply_edit.html",
+                supply=supply,
+                sources=sources,
+                generator_types=generator_types,
+                gsp_groups=gsp_groups,
+                eras=eras,
+            ),
+            400,
+        )
+
+
+@e.route("/supplies/<int:supply_id>/edit", methods=["PATCH"])
+def supply_edit_patch(supply_id):
+    try:
+        supply = Supply.get_by_id(g.sess, supply_id)
+
+        name = req_str("name")
+        source_id = req_int("source_id")
+        gsp_group_id = req_int("gsp_group_id")
+        source = Source.get_by_id(g.sess, source_id)
+        if source.code in ("gen", "gen-grid"):
+            generator_type_id = req_int("generator_type_id")
+            generator_type = GeneratorType.get_by_id(g.sess, generator_type_id)
         else:
-            name = req_str("name")
-            source_id = req_int("source_id")
-            gsp_group_id = req_int("gsp_group_id")
-            source = Source.get_by_id(g.sess, source_id)
-            if source.code in ("gen", "gen-grid"):
-                generator_type_id = req_int("generator_type_id")
-                generator_type = GeneratorType.get_by_id(g.sess, generator_type_id)
-            else:
-                generator_type = None
-            gsp_group = GspGroup.get_by_id(g.sess, gsp_group_id)
-            supply.update(name, source, generator_type, gsp_group, supply.dno)
-            g.sess.commit()
-            return chellow_redirect(f"/supplies/{supply.id}", 303)
+            generator_type = None
+        gsp_group = GspGroup.get_by_id(g.sess, gsp_group_id)
+        supply.update(name, source, generator_type, gsp_group, supply.dno)
+        g.sess.commit()
+        return hx_redirect(f"/supplies/{supply.id}", 303)
     except BadRequest as e:
         g.sess.rollback()
         flash(e.description)
-        sources = g.sess.query(Source).order_by(Source.code)
-        generator_types = g.sess.query(GeneratorType).order_by(GeneratorType.code)
-        gsp_groups = g.sess.query(GspGroup).order_by(GspGroup.code)
-        eras = (
-            g.sess.query(Era)
-            .filter(Era.supply == supply)
-            .order_by(Era.start_date.desc())
+        sources = g.sess.scalars(select(Source).order_by(Source.code))
+        generator_types = g.sess.scalars(
+            select(GeneratorType).order_by(GeneratorType.code)
+        )
+        dno_contract = Contract.get_dno_by_name(g.sess, supply.dno.dno_code)
+        dno_rate_script = dno_contract.find_rate_script_at(
+            g.sess, supply.eras[-1].start_date
+        )
+        dno_props = dno_rate_script.make_script()
+        allowed_gsp_group_codes = list(dno_props.keys())
+
+        gsp_groups = g.sess.scalars(
+            select(GspGroup)
+            .where(GspGroup.code.in_(allowed_gsp_group_codes))
+            .order_by(GspGroup.code)
+        )
+        eras = g.sess.scalars(
+            select(Era).where(Era.supply == supply).order_by(Era.start_date.desc())
+        )
+        return make_response(
+            render_template(
+                "supply_edit.html",
+                supply=supply,
+                sources=sources,
+                generator_types=generator_types,
+                gsp_groups=gsp_groups,
+                eras=eras,
+            ),
+            400,
+        )
+
+
+@e.route("/supplies/<int:supply_id>/edit", methods=["POST"])
+def supply_edit_post(supply_id):
+    try:
+        supply = Supply.get_by_id(g.sess, supply_id)
+
+        start_date = req_date("start")
+        supply.insert_era_at(g.sess, start_date)
+        g.sess.commit()
+        return chellow_redirect(f"/supplies/{supply.id}", 303)
+    except BadRequest as e:
+        g.sess.rollback()
+        flash(e.description)
+        sources = g.sess.scalars(select(Source).order_by(Source.code))
+        generator_types = g.sess.scalars(
+            select(GeneratorType).order_by(GeneratorType.code)
+        )
+        gsp_groups = g.sess.scalars(select(GspGroup).order_by(GspGroup.code))
+        eras = g.sess.scalars(
+            select(Era).where(Era.supply == supply).order_by(Era.start_date.desc())
         )
         return make_response(
             render_template(

chellow/templates/e/supply_edit.html

@@ -11,95 +11,80 @@
 
 {% block content %}
 
-	{% if request.method == 'GET' and request.values.delete %}	
+	<form hx-patch="/e/supplies/{{supply.id}}/edit">
+		<fieldset>
+			<legend>Update this supply</legend>
+			<label>Name</label> {{input_text('name', supply.name)}}
+			<label>Source</label>
+			<select name="source_id">
+				{% for source in sources %}
+					{{input_option('source_id', source.id, source.code, supply.source.id)}}
+				{% endfor %}
+			</select>
+			<label>Generator Type (if source is 'gen' or 'gen-net')</label>
+			<select name="generator_type_id">
+				{% for generator_type in generator_types %}
+					{{input_option(
+						'generator_type_id', generator_type.id, generator_type.code,
+						supply.generator_type.id)}}
+				{% endfor %}
+			</select>
+			<label>GSP Group</label>
+			<select name="gsp_group_id">
+				{% for gsp_group in gsp_groups %}
+					{{input_option(
+						'gsp_group_id', gsp_group.id,
+						gsp_group.code + ' '  + gsp_group.description, supply.gsp_group.id)}}
+				{% endfor %}
+			</select>
+			<input type="submit" value="Update">
+		</fieldset>
+	</form>
 
-		<form method="post">
-			<fieldset>						
-				<legend>Are you sure you want to delete this supply?</legend>
-				<input type="submit" name="delete" value="Delete">
-				<a class="btn" href="/e/supplies/{{supply.id}}/edit">Cancel</a>
-			</fieldset>
-		</form>
+	<form hx-delete="/e/supplies/{{supply.id}}/edit"
+		hx-confirm="Are you sure you want to delete this supply?">
+		<fieldset>
+			<legend>Delete this supply</legend>
+			<input type="submit" name="delete" value="Delete">
+		</fieldset>
+	</form>					
 
-	{% else %}
+	<form method="post" action="/e/supplies/{{supply.id}}/edit">
+		<fieldset>
+			<legend>Insert a new era</legend>
+			<label>Start date</label> {{input_date('start', None)}}
+			<input type="submit" name="insert_era" value="Insert">
+		</fieldset>
+	</form>
 
-		<form method="post" action="/e/supplies/{{supply.id}}/edit">
-			<fieldset>
-				<legend>Update this supply</legend>
-				<label>Name</label> {{input_text('name', supply.name)}}
-				<label>Source</label>
-				<select name="source_id">
-					{% for source in sources %}
-						{{input_option('source_id', source.id, source.code, supply.source.id)}}
-					{% endfor %}
-				</select>
-				<label>Generator Type (if source is 'gen' or 'gen-net')</label>
-				<select name="generator_type_id">
-					{% for generator_type in generator_types %}
-						{{input_option(
-							'generator_type_id', generator_type.id, generator_type.code,
-							supply.generator_type.id)}}
-					{% endfor %}
-				</select>
-					</div>
-				</div>
-				<label>GSP Group</label>
-				<select name="gsp_group_id">
-					{% for gsp_group in gsp_groups %}
-						{{input_option(
-							'gsp_group_id', gsp_group.id,
-							gsp_group.code + ' '  + gsp_group.description, supply.gsp_group.id)}}
-					{% endfor %}
-				</select>
-				<input type="submit" value="Update">
-			</fieldset>
-		</form>
-
-		<form action="/e/supplies/{{supply.id}}/edit">
-			<fieldset>
-				<legend>Delete this supply</legend>
-				<input type="submit" name="delete" value="Delete">
-			</fieldset>
-		</form>					
-
-		<form method="post" action="/e/supplies/{{supply.id}}/edit">
-			<fieldset>
-				<legend>Insert a new era</legend>
-				<label>Start date</label> {{input_date('start', None)}}
-				<input type="submit" name="insert_era" value="Insert">
-			</fieldset>
-		</form>
-
-		<table>
-			<caption>Existing Eras</caption>
-			<thead>
-				<tr>
-					<th>Start date</th>
-					<th>Finish date</th>
-					<th>Import Mpan Core</th>
-					<th>Export Mpan Core</th>
-				</tr>
-			</thead>
-			<tbody>
-			{% for era in eras %}
-				<tr>
-					<td>{{era.start_date|hh_format}}</td>
-					<td>{{era.finish_date|hh_format}}</td>
-					<td>
-						{% if era.imp_mpan_core %}
-							{{era.imp_mpan_core}}
-						{% endif %}
-					</td>
-					<td>
-						{% if era.exp_mpan_core %}
-							{{era.exp_mpan_core}}
-						{% endif %}
-					</td>
-				</tr>
-			{% endfor %}
-			</tbody>
-		</table>
-
-	{% endif %}
+	<table>
+		<caption>Existing Eras</caption>
+		<thead>
+			<tr>
+				<th>Start date</th>
+				<th>Finish date</th>
+				<th>Import Mpan Core</th>
+				<th>Export Mpan Core</th>
+			</tr>
+		</thead>
+		<tbody>
+		{% for era in eras %}
+			<tr>
+				<td>{{era.start_date|hh_format}}</td>
+				<td>{{era.finish_date|hh_format}}</td>
+				<td>
+					{% if era.imp_mpan_core %}
+						{{era.imp_mpan_core}}
+					{% endif %}
+				</td>
+				<td>
+					{% if era.exp_mpan_core %}
+						{{era.exp_mpan_core}}
+					{% endif %}
+				</td>
+			</tr>
+		{% endfor %}
+		</tbody>
+	</table>
 
 {% endblock %}

{chellow-1725528079.0.0.dist-info → chellow-1725628177.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: chellow
-Version: 1725528079.0.0
+Version: 1725628177.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-1725528079.0.0.dist-info → chellow-1725628177.0.0.dist-info}/RECORD

@@ -38,7 +38,7 @@ chellow/e/system_price.py,sha256=6w5J7bzwFAZubE2zdOFRiS8IIrVP8hkoIOaG2yCt-Ic,623
 chellow/e/tlms.py,sha256=M33D6YpMixu2KkwSCzDRM3kThLgShg8exp63Obo75l8,8905
 chellow/e/tnuos.py,sha256=XseYztPUsQXNKuBmystO2kzzwAG9ehCZgpGBTdgSk-A,4313
 chellow/e/triad.py,sha256=lIQj7EdUrcFwEqleuHZXYU_bfzIwNOqUVVxB3NPQt4A,13710
-chellow/e/views.py,sha256=Ue11tdJSjXni4sFobzlvnrTI5evSja_qUuTQ_JusPdU,216671
+chellow/e/views.py,sha256=NwDgs0iZ6nmfX4qF5sSwFlecSsqDBMRsf46vxuIkgVg,219605
 chellow/e/bill_parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chellow/e/bill_parsers/activity_mop_stark_xlsx.py,sha256=UgWXDPzQkQghyj_lfgBqoSJpHB-t-qOdSaB8qY6GLog,4071
 chellow/e/bill_parsers/annual_mop_stark_xlsx.py,sha256=-HMoIfa_utXYKA44RuC0Xqv3vd2HLeQU_4P0iBUd3WA,4219
@@ -310,7 +310,7 @@ chellow/templates/e/supplier_rate_script.html,sha256=tjWeCUAgNip3VLHzbXqe19Msias
 chellow/templates/e/supplier_rate_script_add.html,sha256=Yf2LZEIHbL7qT6oxBCtPf0ZX7vJsSo_ZeOKJhJoVh3o,690
 chellow/templates/e/supplier_rate_script_edit.html,sha256=VaYJt8nxHdnuP-zEAuBJC-ibEpjDU1b80hXtdBQH1dg,1968
 chellow/templates/e/supply.html,sha256=No26xivLm4hA8kRufVvPjrEOE3of7pmZaW_jgTcKSY0,8511
-chellow/templates/e/supply_edit.html,sha256=Ef4ivmSvj9NAb1Uy43S3TCwheCffV457bdpaoD8XZ_s,2723
+chellow/templates/e/supply_edit.html,sha256=2BRGU35nb0ZpUHKCi_fgAPMU6SlHI7ve3_Ne8LXHtms,2333
 chellow/templates/e/supply_eras.html,sha256=5lB7_oC-sTWRQoASuPgBWj5XSZa9ajR0s6V6xKHvf6E,19996
 chellow/templates/e/supply_hh_data.html,sha256=d9ho4Tq3ZkR5GmxSZr_zHGwCU68PH3aSc4OwFG9i9zE,3587
 chellow/templates/e/supply_months.html,sha256=2EUF_SWCcjG9GiAw86gco8xR6LXaqCPXfQE5A33TqqI,1927
@@ -365,6 +365,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-1725528079.0.0.dist-info/METADATA,sha256=T1o-2gNx1BoRjojAEb0dP7MAPEAnrewUDUWM_s0ol1A,12204
-chellow-1725528079.0.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-chellow-1725528079.0.0.dist-info/RECORD,,
+chellow-1725628177.0.0.dist-info/METADATA,sha256=wHjj_Ouyy8wLs6Dv3TYkeiDOzc0SCVRSPCsUOjCnSoM,12204
+chellow-1725628177.0.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+chellow-1725628177.0.0.dist-info/RECORD,,