9a7b8acd5a
* Centralize timezone validation into one spot + add serializer validation for schedules and oncall shifts (both public and internal API) * add engine-manage make command
30 lines
970 B
Python
30 lines
970 B
Python
import pytz
|
|
from rest_framework import serializers
|
|
|
|
from common.api_helpers.exceptions import BadRequest
|
|
|
|
|
|
def is_valid_timezone(timezone: str):
|
|
try:
|
|
return pytz.timezone(timezone)
|
|
except pytz.UnknownTimeZoneError:
|
|
return False
|
|
|
|
|
|
def raise_exception_if_not_valid_timezone(timezone, exception_class=BadRequest):
|
|
"""
|
|
Like `is_valid_timezone` but throws specified "exception_class" class
|
|
(default `common.api_helpers.exceptions.BadRequest`) if not a valid timezone.
|
|
|
|
**NOTE**: if `exception_class` is provided, it should take a `detail` kwarg in its constructor
|
|
"""
|
|
if not is_valid_timezone(timezone):
|
|
raise exception_class(detail="Invalid timezone")
|
|
|
|
|
|
class TimeZoneField(serializers.CharField):
|
|
def _validator(self, value: str):
|
|
raise_exception_if_not_valid_timezone(value, serializers.ValidationError)
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(validators=[self._validator], **kwargs)
|