mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
import re
|
|
from typing import Dict, Union
|
|
from epc_data.attributes.attribute_utils import extract_component_types
|
|
|
|
|
|
class RoofAttributes:
|
|
ROOF_TYPES = ['pitched', 'roof room', 'loft', 'flat', 'thatched', 'at rafters', 'assumed']
|
|
DWELLING_ABOVE = ["another dwelling above", "other premises above"]
|
|
|
|
def __init__(self, description: str):
|
|
"""
|
|
:param description: Description of the roof.
|
|
"""
|
|
|
|
self.description: str = description.lower()
|
|
|
|
if not description or not any(
|
|
rt in self.description for rt in self.ROOF_TYPES + self.DWELLING_ABOVE + ["average thermal transmittance"]
|
|
):
|
|
raise ValueError('Invalid description')
|
|
|
|
def process(self) -> Dict[str, Union[float, str, bool, None]]:
|
|
|
|
if not self.description:
|
|
raise ValueError("Description is empty")
|
|
|
|
result: Dict[str, Union[float, str, bool, None]] = {}
|
|
description = self.description
|
|
|
|
# thermal transmittance
|
|
match = re.search(r"average thermal transmittance (-?\d+\.\d+)\s(w/m-¦k)", description)
|
|
if match:
|
|
result['thermal_transmittance'] = float(match.group(1))
|
|
result['thermal_transmittance_unit'] = match.group(2)
|
|
# Remove the match from the description
|
|
description = re.sub(r"average thermal transmittance (-?\d+\.\d+)\s(w/m-¦k)", "", description)
|
|
else:
|
|
result['thermal_transmittance'] = None
|
|
result['thermal_transmittance_unit'] = None
|
|
|
|
# roof type
|
|
result, description = extract_component_types(result, description, list_of_components=self.ROOF_TYPES)
|
|
|
|
result["has_dwelling_above"] = (
|
|
"another dwelling above" in description or "other premises above" in description
|
|
)
|
|
for dwelling_above in self.DWELLING_ABOVE:
|
|
description = description.replace(dwelling_above, "")
|
|
|
|
result["is_valid"] = "invalid" not in description
|
|
description = description.replace("invalid", "")
|
|
|
|
# insulation thickness
|
|
thickness_map = {
|
|
"ceiling insulated": "average",
|
|
"insulated": "average",
|
|
"limited": "below average",
|
|
"no insulation": "none",
|
|
"limited insulation": "below average",
|
|
"additional insulation": "above average",
|
|
}
|
|
for key, value in thickness_map.items():
|
|
if key in description:
|
|
result['insulation_thickness'] = value
|
|
# Remove the match from the description
|
|
# description = description.replace(key, "")
|
|
break
|
|
else:
|
|
# Extract insulation thickness in mm, if present
|
|
match = re.search(r'(\d+\+?)\s*mm', description)
|
|
if match:
|
|
result['insulation_thickness'] = match.group(1)
|
|
|
|
if "insulation_thickness" not in result:
|
|
result['insulation_thickness'] = None
|
|
|
|
return result
|