Coverage for backend/django/core/auxiliary/methods/custom_property_packages/create_property_package.py: 100%

1from ...models.Flowsheet import Flowsheet 

2from ...models.CustomPropertyPackage import CustomPropertyPackage, CustomPropertyPackageProperty, CustomPropertyPackagePropertyEnum 

3 

4def add_property_package(flowsheet: Flowsheet, name: str) -> CustomPropertyPackage: 

5 """ 

6 Creates the custom property package, and adds all the properties the package needs. 

7 """ 

8 package = CustomPropertyPackage.objects.create(flowsheet=flowsheet, name=name) 

9 

10 # add the properties for this package. 

11 properties = [ 

12 CustomPropertyPackageProperty( 

13 flowsheet=flowsheet, 

14 package=package, 

15 package_property_key=key, 

16 ) 

17 for key in CustomPropertyPackagePropertyEnum.values 

18 ] 

19 CustomPropertyPackageProperty.objects.bulk_create(properties) 

20 return package