31 lines
912 B
Python
31 lines
912 B
Python
"""Utility functions and classes for track backends"""
|
|
|
|
from datetime import datetime, date
|
|
import json
|
|
|
|
from pytz import UTC
|
|
|
|
|
|
class DateTimeJSONEncoder(json.JSONEncoder):
|
|
"""JSON encoder aware of datetime.datetime and datetime.date objects"""
|
|
|
|
def default(self, obj): # pylint: disable=method-hidden
|
|
"""
|
|
Serialize datetime and date objects of iso format.
|
|
|
|
datatime objects are converted to UTC.
|
|
"""
|
|
|
|
if isinstance(obj, datetime):
|
|
if obj.tzinfo is None:
|
|
# Localize to UTC naive datetime objects
|
|
obj = UTC.localize(obj)
|
|
else:
|
|
# Convert to UTC datetime objects from other timezones
|
|
obj = obj.astimezone(UTC)
|
|
return obj.isoformat()
|
|
elif isinstance(obj, date):
|
|
return obj.isoformat()
|
|
|
|
return super(DateTimeJSONEncoder, self).default(obj)
|