Coverage for backend/django/diagnostics/methods/update_diagnostic_result.py: 100%

1from diagnostics.models.DiagnosticsResult import DiagnosticsResult 

2from ahuora_builder_types.flowsheet_schema import UnitDiagnosticsFinding 

3from core.auxiliary.models.FlowsheetState import FlowsheetState 

4 

5def update_diagnostics_results( 

6 flowsheet_state: FlowsheetState, 

7 diagnostics: list[UnitDiagnosticsFinding], 

8): 

9 """Replace generated diagnostics for one captured flowsheet state.""" 

10 

11 # Clear existing diagnostic results for the flowsheet 

12 DiagnosticsResult.objects.filter(flowsheet_state=flowsheet_state).delete() 

13 print(diagnostics) 

14 

15 # Create new diagnostic results based on the provided diagnostics 

16 new_results = [] 

17 for finding in diagnostics: 

18 new_result = DiagnosticsResult( 

19 flowsheet_state=flowsheet_state, 

20 propertyValue_id=finding.property_value_id, 

21 severity=finding.severity, 

22 message=finding.message 

23 ) 

24 new_results.append(new_result) 

25 

26 DiagnosticsResult.objects.bulk_create(new_results)