# -*- coding: utf-8 -*-
###########################################################################
# Copyright (c), The AiiDA team. All rights reserved. #
# This file is part of the AiiDA code. #
# #
# The code is hosted on GitHub at https://github.com/aiidateam/aiida_core #
# For further information on the license, see the LICENSE.txt file #
# For further information please visit http://www.aiida.net #
###########################################################################
import pytz
from datetime import datetime
from aiida import settings
# All the timezone part here is taken from Django.
# TODO SP: check license terms ?
# TODO SP: docstring
utc = pytz.utc
[docs]def get_current_timezone():
from tzlocal import get_localzone
return get_localzone()
[docs]def now():
if getattr(settings, "USE_TZ", None):
return datetime.utcnow().replace(tzinfo=utc)
else:
return datetime.now()
[docs]def is_naive(value):
return value.utcoffset() is None
[docs]def is_aware(value):
return value.utcoffset() is not None
[docs]def make_aware(value, timezone=None, is_dst=None):
"""
:param value: The datetime to make aware
:type value: :class:`!datetime.datetime`
:param timezone:
:param is_dst:
:return:
"""
if timezone is None:
timezone = get_current_timezone()
if hasattr(timezone, 'localize'):
return timezone.localize(value, is_dst=is_dst)
else:
if is_aware(value):
raise ValueError(
"make_aware expects a naive datetime, got %s" % value)
# This may be wrong around DST changes!
return value.replace(tzinfo=timezone)
[docs]def localtime(value, timezone=None):
"""
Converts an aware datetime.datetime to local time.
Local time is defined by the current time zone, unless another time zone
is specified.
"""
if timezone is None:
timezone = get_current_timezone()
# If `value` is naive, astimezone() will raise a ValueError,
# so we don't need to perform a redundant check.
value = value.astimezone(timezone)
if hasattr(timezone, 'normalize'):
# This method is available for pytz time zones.
value = timezone.normalize(value)
return value
[docs]def delta(from_time, to_time=None):
if to_time is None:
to_time = now()
try:
x0 = make_aware(from_time)
except ValueError:
x0 = from_time
try:
x1 = make_aware(to_time)
except ValueError:
x1 = to_time
return x1 - x0