From 4bcd17596e9931508c1df3cfc13b16d440b36980 Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Tue, 2 Jul 2024 13:11:23 +0100 Subject: [PATCH] handle missing bills data --- etl/bill_savings/data_collection.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/etl/bill_savings/data_collection.py b/etl/bill_savings/data_collection.py index 2632c296..1f787d48 100644 --- a/etl/bill_savings/data_collection.py +++ b/etl/bill_savings/data_collection.py @@ -98,8 +98,11 @@ def retrieve_find_my_epc_data(uprn: int, postcode: str, address: str, expected_e # Retrieve the energy consumption bills = address_res.find('div', {'id': 'bills-affected'}) - heating_text = bills.find_all('li')[0].text - hot_water_text = bills.find_all('li')[1].text + bills_list = bills.find_all('li') + if not bills_list: + return None + heating_text = bills_list[0].text + hot_water_text = bills_list[1].text resulting_data = { 'extracted_uprn': uprn,