@@ -218,7 +218,7 @@ def get_vendor_name(module: etree, model_data: ModuleData) -> ModuleData:
218218 if (vendor_name := module .find ("vendor_name" ).text ) is not None :
219219 model_data ["vendor_name" ] = vendor_name
220220 if "Plugwise" in vendor_name :
221- model_data ["vendor_name" ] = vendor_name .split (" " , 1 )[0 ]
221+ model_data ["vendor_name" ] = vendor_name .partition (" " )[0 ]
222222
223223 return model_data
224224
@@ -275,9 +275,9 @@ def power_data_peak_value(loc: Munch, legacy: bool) -> Munch:
275275 if not loc .found :
276276 return loc
277277
278- if (peak := loc .peak_select .split ("_" )[1 ]) == "offpeak" :
278+ if (peak := loc .peak_select .partition ("_" )[2 ]) == "offpeak" :
279279 peak = "off_peak"
280- log_found = loc .log_type .split ("_" )[0 ]
280+ log_found = loc .log_type .partition ("_" )[0 ]
281281 loc .key_string = f"{ loc .measurement } _{ peak } _{ log_found } "
282282 if "gas" in loc .measurement or loc .log_type == "point_meter" :
283283 loc .key_string = f"{ loc .measurement } _{ log_found } "
@@ -314,7 +314,7 @@ def skip_obsolete_measurements(xml: etree, measurement: str) -> bool:
314314 measurement in OBSOLETE_MEASUREMENTS
315315 and (updated_date_key := xml .find (locator )) is not None
316316 ):
317- updated_date = updated_date_key .text .split ("T" )[0 ]
317+ updated_date = updated_date_key .text .partition ("T" )[0 ]
318318 date_1 = dt .datetime .strptime (updated_date , "%Y-%m-%d" )
319319 date_2 = dt .datetime .now ()
320320 return int ((date_2 - date_1 ).days ) > 7
0 commit comments