|
@@ -1,10 +1,12 @@
|
|
|
from flask_restful import fields
|
|
|
|
|
|
-from core.app.segments import SecretVariable, Variable
|
|
|
+from core.app.segments import SecretVariable, SegmentType, Variable
|
|
|
from core.helper import encrypter
|
|
|
from fields.member_fields import simple_account_fields
|
|
|
from libs.helper import TimestampField
|
|
|
|
|
|
+ENVIRONMENT_VARIABLE_SUPPORTED_TYPES = (SegmentType.STRING, SegmentType.NUMBER, SegmentType.SECRET)
|
|
|
+
|
|
|
|
|
|
class EnvironmentVariableField(fields.Raw):
|
|
|
def format(self, value):
|
|
@@ -16,14 +18,18 @@ class EnvironmentVariableField(fields.Raw):
|
|
|
'value': encrypter.obfuscated_token(value.value),
|
|
|
'value_type': value.value_type.value,
|
|
|
}
|
|
|
- elif isinstance(value, Variable):
|
|
|
+ if isinstance(value, Variable):
|
|
|
return {
|
|
|
'id': value.id,
|
|
|
'name': value.name,
|
|
|
'value': value.value,
|
|
|
'value_type': value.value_type.value,
|
|
|
}
|
|
|
- return value
|
|
|
+ if isinstance(value, dict):
|
|
|
+ value_type = value.get('value_type')
|
|
|
+ if value_type not in ENVIRONMENT_VARIABLE_SUPPORTED_TYPES:
|
|
|
+ raise ValueError(f'Unsupported environment variable value type: {value_type}')
|
|
|
+ return value
|
|
|
|
|
|
|
|
|
environment_variable_fields = {
|