forked from DevCycleHQ/python-server-sdk
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathvariable.py
More file actions
65 lines (56 loc) · 1.67 KB
/
Copy pathvariable.py
File metadata and controls
65 lines (56 loc) · 1.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# ruff: noqa: N815
from dataclasses import dataclass
from typing import Optional, Any
class TypeEnum:
BOOLEAN = "Boolean"
STRING = "String"
NUMBER = "Number"
JSON = "JSON"
def determine_variable_type(value: Any) -> str:
if isinstance(value, bool):
return TypeEnum.BOOLEAN
elif isinstance(value, str):
return TypeEnum.STRING
elif isinstance(value, (int, float)):
return TypeEnum.NUMBER
elif isinstance(value, dict):
return TypeEnum.JSON
else:
raise TypeError(f"Unsupported type: {type(value)}")
@dataclass(order=False)
class Variable:
_id: Optional[str]
key: str
type: str
value: Any = None
isDefaulted: Optional[bool] = False
defaultValue: Any = None
evalReason: Optional[str] = None
def to_json(self):
return {
key: getattr(self, key)
for key in self.__dataclass_fields__
if getattr(self, key) is not None
}
@classmethod
def from_json(cls, data: dict) -> "Variable":
return cls(
_id=data["_id"],
key=data["key"],
type=data["type"],
value=data["value"],
isDefaulted=data.get("isDefaulted", None),
defaultValue=data.get("defaultValue"),
evalReason=data.get("evalReason"),
)
@staticmethod
def create_default_variable(key: str, default_value: Any) -> "Variable":
var_type = determine_variable_type(default_value)
return Variable(
_id=None,
key=key,
type=var_type,
value=default_value,
defaultValue=default_value,
isDefaulted=True,
)