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 import Flowsheet 

4 

5def update_diagnostics_results(flowsheet: Flowsheet, diagnostics: list[UnitDiagnosticsFinding]): 

6 

7 # Clear existing diagnostic results for the flowsheet 

8 DiagnosticsResult.objects.filter(flowsheet=flowsheet).delete() 

9 print(diagnostics) 

10 

11 # Create new diagnostic results based on the provided diagnostics 

12 new_results = [] 

13 for finding in diagnostics: 

14 new_result = DiagnosticsResult( 

15 flowsheet=flowsheet, 

16 propertyValue_id=finding.property_value_id, 

17 severity=finding.severity, 

18 message=finding.message 

19 ) 

20 new_results.append(new_result) 

21 

22 DiagnosticsResult.objects.bulk_create(new_results)