mirror of
https://github.com/lucaspalomodevelop/eventcally.git
synced 2026-03-13 00:07:22 +00:00
34 lines
936 B
Python
34 lines
936 B
Python
from marshmallow import ValidationError, fields
|
|
|
|
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 _serialize(self, value, attr, obj, **kwargs):
|
|
if value:
|
|
value = value.astimezone(berlin_tz)
|
|
|
|
return super()._serialize(value, attr, obj, **kwargs)
|
|
|
|
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
|