eventcally/project/api/fields.py
2021-02-09 11:12:15 +01:00

27 lines
751 B
Python

from marshmallow import fields, ValidationError
from project.dateutils import berlin_tz
class NumericStr(fields.String):
def _serialize(self, value, attr, obj, **kwargs):
if value is None:
return None
return str(value)
def _deserialize(self, value, attr, data, **kwargs):
try:
return float(value)
except ValueError as error:
raise ValidationError("Must be a numeric value.") from error
class CustomDateTimeField(fields.DateTime):
def deserialize(self, value, attr, data, **kwargs):
result = super().deserialize(value, attr, data, **kwargs)
if result and result.tzinfo is None:
result = berlin_tz.localize(result)
return result