Coverage for backend/django/idaes_factory/adapters/enum_adapter.py: 100%

1from ahuora_builder_types.unit_model_schema import ValueArgSchema 

2 

3from .property_info_adapter import ValueAdapter 

4from ..queryset_lookup import get_property, get_value_object 

5 

6 

7class EnumAdapter(ValueAdapter): 

8 """ 

9 Returns the enum value of the property. 

10 """ 

11 def __init__(self, toggle_key:str): 

12 self.toggle_key = toggle_key 

13 

14 def serialise(self, ctx, model) -> bool: 

15 

16 # get the property from the unit model 

17 property_set = model.properties 

18 prop = get_property(property_set, self.toggle_key) 

19 # serialise the property 

20 return ValueArgSchema(value=prop.values.first().value)