Coverage for backend/django/core/auxiliary/methods/custom_property_packages/create_property_package.py: 100%
7 statements
« prev ^ index » next coverage.py v7.10.7, created at 2026-07-22 05:22 +0000
1from ...models.FlowsheetState import FlowsheetState
2from ...models.CustomPropertyPackage import CustomPropertyPackage, CustomPropertyPackageProperty, CustomPropertyPackagePropertyEnum
4def add_property_package(flowsheet_state: FlowsheetState, name: str) -> CustomPropertyPackage:
5 """
6 Creates the custom property package, and adds all the properties the package needs.
7 """
8 package = CustomPropertyPackage.objects.create(
9 flowsheet_state=flowsheet_state,
10 name=name,
11 )
13 # add the properties for this package.
14 properties = [
15 CustomPropertyPackageProperty(
16 flowsheet_state=flowsheet_state,
17 package=package,
18 package_property_key=key,
19 )
20 for key in CustomPropertyPackagePropertyEnum.values
21 ]
22 CustomPropertyPackageProperty.objects.bulk_create(properties)
23 return package