Coverage for backend/django/idaes_factory/build_hooks.py: 100%

1from __future__ import annotations 

2 

3from dataclasses import dataclass 

4from typing import TYPE_CHECKING 

5 

6if TYPE_CHECKING: 

7 from core.auxiliary.models.Scenario import Scenario 

8 from core.auxiliary.models.Flowsheet import Flowsheet 

9 

10 

11@dataclass(frozen=True) 

12class IdaesBuildHookContext: 

13 """Context passed to domain packages before IDAES loads the factory context.""" 

14 

15 flowsheet: "Flowsheet" 

16 scenario: "Scenario | None" 

17 group_id: int 

18 solve_index: int | None 

19 

20 

21def run_before_context_load_hooks(context: IdaesBuildHookContext) -> None: 

22 """Run domain updates before IDAES eagerly loads flowsheet data.""" 

23 

24 from Economics.idaes_build_hooks import refresh_economics_formulas_before_context_load 

25 

26 refresh_economics_formulas_before_context_load(context)