import datetime import django.utils.timezone as tz __all__ = ( "Redirect", "redirect", "cast_date", "date_or_default", "to_timedelta", "seconds_to_time", ) # FIXME: usage & why we don't use Django's class Redirect(Exception): """Redirect exception -- see `redirect()`.""" def __init__(self, url): self.url = url def redirect(url): """Raise a Redirect exception in order to response a redirection to client. AircoxMiddleware must be enabled. """ raise Redirect(url) def str_to_date(value, sep="/"): """Return a date from the provided `value` string, formated as "yyyy/mm/dd". :raises: ValueError for incorrect value format. """ value = value.split(sep)[:3] if len(value) < 3: raise ValueError("incorrect date format") return datetime.date(int(value[0]), int(value[1]), int(value[2])) def cast_date(date, into=datetime.date): """Cast a given date into the provided class' instance. Make datetime aware of timezone. """ date = into(date.year, date.month, date.day) return tz.make_aware(date) if issubclass(into, tz.datetime) else date def date_or_default(date, into=None): """Return date if not None, otherwise return now. Cast result into provided type if any. """ date = ( date if date is not None else datetime.date.today() if into is not None and issubclass(into, datetime.date) else tz.now() ) if into is not None: date = cast_date(date, into) if isinstance(date, tz.datetime) and not tz.is_aware(date): date = tz.make_aware(date) return date def to_timedelta(time): """Transform a datetime or a time instance to a timedelta, only using time info.""" return datetime.timedelta(hours=time.hour, minutes=time.minute, seconds=time.second) def to_seconds(time): """Return total seconds for provided time.""" return 3600 * time.hour + 60 * time.minute + time.second def seconds_to_time(seconds): """Seconds to datetime.time.""" seconds, microseconds = divmod(seconds, 1) minutes, seconds = divmod(seconds, 60) hours, minutes = divmod(minutes, 60) return datetime.time( hour=int(hours), minute=int(minutes), second=int(seconds), microsecond=int(microseconds * 100000), )