diff --git a/etl/transform/conditionReportTypes.py b/etl/transform/conditionReportTypes.py index 358ceae..9935af5 100644 --- a/etl/transform/conditionReportTypes.py +++ b/etl/transform/conditionReportTypes.py @@ -1,10 +1,23 @@ from etl.transform.preSiteNoteTypes import BaseModel, Optional, List -class ConditionReport(BaseModel): - property_reference_code: str + +class AssessorInformationFromConditionReport(BaseModel): + assessor_name: str + assessor_id: str + +class InspectionAndProject(BaseModel): + inspection_date: str + risk_assessment_pathway: str + +class TheProperty(BaseModel): + property_address: str + is_the_building_classed_as_protected: bool + type_of_proerty: str class GeneralInformation(BaseModel): - pass + assessor: AssessorInformationFromConditionReport + inspection_and_project: InspectionAndProject + property: TheProperty class PropertyAccess(BaseModel): are_there_any_road_restrictions_in_the_locality: bool @@ -209,4 +222,13 @@ class OccupancyAssessment(BaseModel): class VentilationChecklist(BaseModel): intermittent_extract_ventilation: bool continous_mechanical_extract_ventilation: bool - mechanical_ventilation_with_heat_recovery: bool \ No newline at end of file + mechanical_ventilation_with_heat_recovery: bool + +class ConditionReport(BaseModel): + property_reference_code: str + general_information: GeneralInformation + access_and_elevation: AccessAndElevations + rooms: Rooms + heating_system: HeatingSystem + occupancy_assessment: OccupancyAssessment + ventilation_checklist: VentilationChecklist \ No newline at end of file