Coverage for /opt/conda/envs/apienv/lib/python3.10/site-packages/daiquiri/core/hardware/rtschemamixin.py: 0%

1#!/usr/bin/env python 

2# -*- coding: utf-8 -*- 

3from marshmallow import fields, validate 

4 

5from daiquiri.core.hardware.abstract import HardwareProperty 

6from daiquiri.core.hardware.abstract.generic import ( 

7 GenericPropertiesSchema, 

8 GenericCallablesSchema, 

9) 

10 

11 

12class RTSchemaMixin: 

13 """RunTime Schema Mixin 

14 

15 Allows the validation schema for the object to be generated at 

16 run time from the hardware.yml config 

17 """ 

18 

19 def __init__(self, *args, **kwargs): 

20 super().__init__(*args, **kwargs) 

21 

22 attrs, ui = self._map_to_schema(kwargs.get("attributes", {})) 

23 

24 class Meta: 

25 uischema = ui 

26 

27 attrs["Meta"] = Meta 

28 

29 properties_schema = type( 

30 self.id() + "PropertiesSchema", (GenericPropertiesSchema,), attrs 

31 ) 

32 self._properties = properties_schema() 

33 

34 callables_schema = type( 

35 self.id() + "CallablesSchema", (GenericCallablesSchema,), attrs 

36 ) 

37 self._callables = callables_schema() 

38 

39 def schema_name(self): 

40 return self.id() 

41 

42 def _map_to_schema(self, attributes): 

43 attr_map = { 

44 "int": fields.Int, 

45 "float": fields.Float, 

46 "bool": fields.Bool, 

47 "str": fields.Str, 

48 } 

49 

50 attrs = {} 

51 ui = {} 

52 for a in attributes: 

53 if a.get("type") in attr_map: 

54 field = attr_map[a["type"]] 

55 val = None 

56 if a.get("min") is not None and a.get("max") is not None: 

57 val = (validate.Range(min=a["min"], max=a["max"]),) 

58 

59 extra_kws = {} 

60 if a.get("step"): 

61 extra_kws["multipleOf"] = a["step"] 

62 

63 attrs[a["id"]] = field( 

64 validate=val, metadata={"title": a.get("name", None)}, **extra_kws 

65 ) 

66 

67 if a.get("ui_schema"): 

68 ui[a.get("id")] = a.get("ui_schema") 

69 

70 self._property_map[a["id"]] = HardwareProperty(a["id"]) 

71 

72 return attrs, ui