"""
Functions for mathematical datetime operations
(C) 2011-2013 by the GRASS Development Team
This program is free software under the GNU General Public
License (>=v2). Read the file COPYING that comes with GRASS
for details.
:authors: Soeren Gebbert
"""
import sys
from datetime import datetime, timedelta
from .core import get_tgis_message_interface
import copy
try:
import dateutil.parser as parser
has_dateutil = True
except:
has_dateutil = False
if sys.version_info[0] >= 3:
unicode = str
DAY_IN_SECONDS = 86400
SECOND_AS_DAY = 1.1574074074074073e-05
###############################################################################
[docs]def relative_time_to_time_delta(value):
"""Convert the double value representing days
into a timedelta object.
"""
days = int(value)
seconds = value % 1
seconds = round(seconds * DAY_IN_SECONDS)
return timedelta(days, seconds)
###############################################################################
[docs]def time_delta_to_relative_time(delta):
"""Convert the time delta into a
double value, representing days.
"""
return float(delta.days) + float(delta.seconds * SECOND_AS_DAY)
###############################################################################
[docs]def relative_time_to_time_delta_seconds(value):
"""Convert the double value representing seconds
into a timedelta object.
"""
days = value / 86400
seconds = int(value % 86400)
return timedelta(days, seconds)
###############################################################################
[docs]def time_delta_to_relative_time_seconds(delta):
"""Convert the time delta into a
double value, representing seconds.
"""
return float(delta.days * DAY_IN_SECONDS) + float(delta.seconds)
###############################################################################
[docs]def decrement_datetime_by_string(mydate, increment, mult=1):
"""Return a new datetime object decremented with the provided
relative dates specified as string.
Additional a multiplier can be specified to multiply the increment
before adding to the provided datetime object.
Usage:
.. code-block:: python
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "31 days"
>>> decrement_datetime_by_string(dt, string)
datetime.datetime(2000, 12, 1, 0, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "1 month"
>>> decrement_datetime_by_string(dt, string)
datetime.datetime(2000, 12, 1, 0, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "2 month"
>>> decrement_datetime_by_string(dt, string)
datetime.datetime(2000, 11, 1, 0, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "24 months"
>>> decrement_datetime_by_string(dt, string)
datetime.datetime(1999, 1, 1, 0, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "48 months"
>>> decrement_datetime_by_string(dt, string)
datetime.datetime(1997, 1, 1, 0, 0)
>>> dt = datetime(2001, 6, 1, 0, 0, 0)
>>> string = "5 months"
>>> decrement_datetime_by_string(dt, string)
datetime.datetime(2001, 1, 1, 0, 0)
>>> dt = datetime(2001, 6, 1, 0, 0, 0)
>>> string = "7 months"
>>> decrement_datetime_by_string(dt, string)
datetime.datetime(2000, 11, 1, 0, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "1 year"
>>> decrement_datetime_by_string(dt, string)
datetime.datetime(2000, 1, 1, 0, 0)
:param mydate: A datetime object to incremented
:param increment: A string providing increment information:
The string may include comma separated values of type
seconds, minutes, hours, days, weeks, months and years
Example: Increment the datetime 2001-01-01 00:00:00
with "60 seconds, 4 minutes, 12 hours, 10 days,
1 weeks, 5 months, 1 years" will result in the
datetime 2003-02-18 12:05:00
:param mult: A multiplier, default is 1
:return: The new datetime object or none in case of an error
"""
return modify_datetime_by_string(mydate, increment, mult, sign=int(-1))
###############################################################################
[docs]def increment_datetime_by_string(mydate, increment, mult=1):
"""Return a new datetime object incremented with the provided
relative dates specified as string.
Additional a multiplier can be specified to multiply the increment
before adding to the provided datetime object.
Usage:
.. code-block:: python
>>> dt = datetime(2001, 9, 1, 0, 0, 0)
>>> string = "60 seconds, 4 minutes, 12 hours, 10 days, 1 weeks, 5 months, 1 years"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2003, 2, 18, 12, 5)
>>> dt = datetime(2001, 11, 1, 0, 0, 0)
>>> string = "1 months"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2001, 12, 1, 0, 0)
>>> dt = datetime(2001, 11, 1, 0, 0, 0)
>>> string = "13 months"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2002, 12, 1, 0, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "72 months"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2007, 1, 1, 0, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "72 months"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2007, 1, 1, 0, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "5 minutes"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2001, 1, 1, 0, 5)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "49 hours"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2001, 1, 3, 1, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "3600 seconds"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2001, 1, 1, 1, 0)
>>> dt = datetime(2001, 1, 1, 0, 0, 0)
>>> string = "30 days"
>>> increment_datetime_by_string(dt, string)
datetime.datetime(2001, 1, 31, 0, 0)
:param mydate: A datetime object to incremented
:param increment: A string providing increment information:
The string may include comma separated values of type
seconds, minutes, hours, days, weeks, months and years
Example: Increment the datetime 2001-01-01 00:00:00
with "60 seconds, 4 minutes, 12 hours, 10 days,
1 weeks, 5 months, 1 years" will result in the
datetime 2003-02-18 12:05:00
:param mult: A multiplier, default is 1
:return: The new datetime object or none in case of an error
"""
return modify_datetime_by_string(mydate, increment, mult, sign=int(1))
###############################################################################
[docs]def modify_datetime_by_string(mydate, increment, mult=1, sign=1):
"""Return a new datetime object incremented with the provided
relative dates specified as string.
Additional a multiplier can be specified to multiply the increment
before adding to the provided datetime object.
:param mydate: A datetime object to incremented
:param increment: A string providing increment information:
The string may include comma separated values of type
seconds, minutes, hours, days, weeks, months and years
Example: Increment the datetime 2001-01-01 00:00:00
with "60 seconds, 4 minutes, 12 hours, 10 days,
1 weeks, 5 months, 1 years" will result in the
datetime 2003-02-18 12:05:00
:param mult: A multiplier, default is 1
:param sign: Choose 1 for positive sign (incrementing) or -1 for
negative sign (decrementing).
:return: The new datetime object or none in case of an error
"""
sign = int(sign)
if sign != 1 and sign != -1:
return None
if increment:
seconds = 0
minutes = 0
hours = 0
days = 0
weeks = 0
months = 0
years = 0
inclist = []
# Split the increment string
incparts = increment.split(",")
for incpart in incparts:
inclist.append(incpart.strip().split(" "))
for inc in inclist:
msgr = get_tgis_message_interface()
if len(inc) < 2:
msgr.error(_("Wrong increment format: %s") % (increment))
return None
if inc[1].find("seconds") >= 0 or inc[1].find("second") >= 0:
seconds = sign * mult * int(inc[0])
elif inc[1].find("minutes") >= 0 or inc[1].find("minute") >= 0:
minutes = sign * mult * int(inc[0])
elif inc[1].find("hours") >= 0 or inc[1].find("hour") >= 0:
hours = sign * mult * int(inc[0])
elif inc[1].find("days") >= 0 or inc[1].find("day") >= 0:
days = sign * mult * int(inc[0])
elif inc[1].find("weeks") >= 0 or inc[1].find("week") >= 0:
weeks = sign * mult * int(inc[0])
elif inc[1].find("months") >= 0 or inc[1].find("month") >= 0:
months = sign * mult * int(inc[0])
elif inc[1].find("years") >= 0 or inc[1].find("year") >= 0:
years = sign * mult * int(inc[0])
else:
msgr.error(_("Wrong increment format: %s") % (increment))
return None
return modify_datetime(
mydate, years, months, weeks, days, hours, minutes, seconds
)
return mydate
###############################################################################
[docs]def modify_datetime(
mydate, years=0, months=0, weeks=0, days=0, hours=0, minutes=0, seconds=0
):
"""Return a new datetime object incremented with the provided
relative dates and times"""
tdelta_seconds = timedelta(seconds=seconds)
tdelta_minutes = timedelta(minutes=minutes)
tdelta_hours = timedelta(hours=hours)
tdelta_days = timedelta(days=days)
tdelta_weeks = timedelta(weeks=weeks)
tdelta_months = timedelta(0)
tdelta_years = timedelta(0)
if months > 0:
# Compute the actual number of days in the month to add as timedelta
year = mydate.year
month = mydate.month
all_months = int(months) + int(month)
years_to_add = int(all_months / 12.001)
residual_months = all_months - (years_to_add * 12)
# Make a deep copy of the datetime object
dt1 = copy.copy(mydate)
# Make sure the month starts with a 1
if residual_months == 0:
residual_months = 1
try:
dt1 = dt1.replace(year=year + years_to_add, month=residual_months)
except:
raise
tdelta_months = dt1 - mydate
elif months < 0:
# Compute the actual number of days in the month to add as timedelta
year = mydate.year
month = mydate.month
years_to_remove = 0
all_months = int(months) + int(month)
if all_months <= 0:
years_to_remove = abs(int(all_months / 12.001))
residual_months = all_months + (years_to_remove * 12)
years_to_remove += 1
else:
residual_months = all_months
# Make a deep copy of the datetime object
dt1 = copy.copy(mydate)
# Correct the months
if residual_months <= 0:
residual_months += 12
try:
dt1 = dt1.replace(year=year - years_to_remove, month=residual_months)
except:
raise
tdelta_months = dt1 - mydate
if years != 0:
# Make a deep copy of the datetime object
dt1 = copy.copy(mydate)
# Compute the number of days
dt1 = dt1.replace(year=mydate.year + int(years))
tdelta_years = dt1 - mydate
return (
mydate
+ tdelta_seconds
+ tdelta_minutes
+ tdelta_hours
+ tdelta_days
+ tdelta_weeks
+ tdelta_months
+ tdelta_years
)
###############################################################################
[docs]def adjust_datetime_to_granularity(mydate, granularity):
"""Modify the datetime object to fit the given granularity
- Years will start at the first of January
- Months will start at the first day of the month
- Days will start at the first Hour of the day
- Hours will start at the first minute of an hour
- Minutes will start at the first second of a minute
Usage:
.. code-block:: python
>>> dt = datetime(2001, 8, 8, 12,30,30)
>>> adjust_datetime_to_granularity(dt, "5 seconds")
datetime.datetime(2001, 8, 8, 12, 30, 30)
>>> adjust_datetime_to_granularity(dt, "20 minutes")
datetime.datetime(2001, 8, 8, 12, 30)
>>> adjust_datetime_to_granularity(dt, "20 minutes")
datetime.datetime(2001, 8, 8, 12, 30)
>>> adjust_datetime_to_granularity(dt, "3 hours")
datetime.datetime(2001, 8, 8, 12, 0)
>>> adjust_datetime_to_granularity(dt, "5 days")
datetime.datetime(2001, 8, 8, 0, 0)
>>> adjust_datetime_to_granularity(dt, "2 weeks")
datetime.datetime(2001, 8, 6, 0, 0)
>>> adjust_datetime_to_granularity(dt, "6 months")
datetime.datetime(2001, 8, 1, 0, 0)
>>> adjust_datetime_to_granularity(dt, "2 years")
datetime.datetime(2001, 1, 1, 0, 0)
>>> adjust_datetime_to_granularity(dt, "2 years, 3 months, 5 days, 3 hours, 3 minutes, 2 seconds")
datetime.datetime(2001, 8, 8, 12, 30, 30)
>>> adjust_datetime_to_granularity(dt, "3 months, 5 days, 3 minutes")
datetime.datetime(2001, 8, 8, 12, 30)
>>> adjust_datetime_to_granularity(dt, "3 weeks, 5 days")
datetime.datetime(2001, 8, 8, 0, 0)
"""
if granularity:
has_seconds = False
has_minutes = False
has_hours = False
has_days = False
has_weeks = False
has_months = False
has_years = False
seconds = mydate.second
minutes = mydate.minute
hours = mydate.hour
days = mydate.day
weekday = mydate.weekday()
months = mydate.month
years = mydate.year
granlist = []
# Split the increment string
granparts = granularity.split(",")
for granpart in granparts:
granlist.append(granpart.strip().split(" "))
for inc in granlist:
if inc[1].find("seconds") >= 0 or inc[1].find("second") >= 0:
has_seconds = True
elif inc[1].find("minutes") >= 0 or inc[1].find("minute") >= 0:
has_minutes = True
elif inc[1].find("hours") >= 0 or inc[1].find("hour") >= 0:
has_hours = True
elif inc[1].find("days") >= 0 or inc[1].find("day") >= 0:
has_days = True
elif inc[1].find("weeks") >= 0 or inc[1].find("week") >= 0:
has_weeks = True
elif inc[1].find("months") >= 0 or inc[1].find("month") >= 0:
has_months = True
elif inc[1].find("years") >= 0 or inc[1].find("year") >= 0:
has_years = True
else:
msgr = get_tgis_message_interface()
msgr.error(_("Wrong granularity format: %s") % (granularity))
return None
if has_seconds:
pass
elif has_minutes: # Start at 0 seconds
seconds = 0
elif has_hours: # Start at 0 minutes and seconds
seconds = 0
minutes = 0
elif has_days: # Start at 0 hours, minutes and seconds
seconds = 0
minutes = 0
hours = 0
elif has_weeks: # Start at the first day of the week (Monday) at 00:00:00
seconds = 0
minutes = 0
hours = 0
if days > weekday:
days = days - weekday # this needs to be fixed
else:
days = days + weekday # this needs to be fixed
elif has_months: # Start at the first day of the month at 00:00:00
seconds = 0
minutes = 0
hours = 0
days = 1
elif has_years: # Start at the first day of the first month at 00:00:00
seconds = 0
minutes = 0
hours = 0
days = 1
months = 1
dt = copy.copy(mydate)
return dt.replace(
year=years,
month=months,
day=days,
hour=hours,
minute=minutes,
second=seconds,
)
###############################################################################
[docs]def compute_datetime_delta(start, end):
"""Return a dictionary with the accumulated delta in year, month, day,
hour, minute and second
Usage:
.. code-block:: python
>>> start = datetime(2001, 1, 1, 00,00,00)
>>> end = datetime(2001, 1, 1, 00,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 0, 'second': 0, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 0}
>>> start = datetime(2001, 1, 1, 00,00,14)
>>> end = datetime(2001, 1, 1, 00,00,44)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 0, 'second': 30, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 0}
>>> start = datetime(2001, 1, 1, 00,00,44)
>>> end = datetime(2001, 1, 1, 00,01,14)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 0, 'second': 30, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 1}
>>> start = datetime(2001, 1, 1, 00,00,30)
>>> end = datetime(2001, 1, 1, 00,05,30)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 0, 'second': 300, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 5}
>>> start = datetime(2001, 1, 1, 00,00,00)
>>> end = datetime(2001, 1, 1, 00,01,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 0, 'second': 0, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 1}
>>> start = datetime(2011,10,31, 00,45,00)
>>> end = datetime(2011,10,31, 01,45,00)
>>> compute_datetime_delta(start, end)
{'hour': 1, 'second': 0, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 60}
>>> start = datetime(2011,10,31, 00,45,00)
>>> end = datetime(2011,10,31, 01,15,00)
>>> compute_datetime_delta(start, end)
{'hour': 1, 'second': 0, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 30}
>>> start = datetime(2011,10,31, 00,45,00)
>>> end = datetime(2011,10,31, 12,15,00)
>>> compute_datetime_delta(start, end)
{'hour': 12, 'second': 0, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 690}
>>> start = datetime(2011,10,31, 00,00,00)
>>> end = datetime(2011,10,31, 01,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 1, 'second': 0, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 0}
>>> start = datetime(2011,10,31, 00,00,00)
>>> end = datetime(2011,11,01, 01,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 25, 'second': 0, 'max_days': 1, 'year': 0, 'day': 1, 'minute': 0}
>>> start = datetime(2011,10,31, 12,00,00)
>>> end = datetime(2011,11,01, 06,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 18, 'second': 0, 'max_days': 0, 'year': 0, 'day': 0, 'minute': 0}
>>> start = datetime(2011,11,01, 00,00,00)
>>> end = datetime(2011,12,01, 01,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 721, 'month': 1, 'second': 0, 'max_days': 30, 'year': 0, 'day': 0, 'minute': 0}
>>> start = datetime(2011,11,01, 00,00,00)
>>> end = datetime(2011,11,05, 00,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'second': 0, 'max_days': 4, 'year': 0, 'day': 4, 'minute': 0}
>>> start = datetime(2011,10,06, 00,00,00)
>>> end = datetime(2011,11,05, 00,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'second': 0, 'max_days': 30, 'year': 0, 'day': 30, 'minute': 0}
>>> start = datetime(2011,12,02, 00,00,00)
>>> end = datetime(2012,01,01, 00,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'second': 0, 'max_days': 30, 'year': 1, 'day': 30, 'minute': 0}
>>> start = datetime(2011,01,01, 00,00,00)
>>> end = datetime(2011,02,01, 00,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 1, 'second': 0, 'max_days': 31, 'year': 0, 'day': 0, 'minute': 0}
>>> start = datetime(2011,12,01, 00,00,00)
>>> end = datetime(2012,01,01, 00,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 1, 'second': 0, 'max_days': 31, 'year': 1, 'day': 0, 'minute': 0}
>>> start = datetime(2011,12,01, 00,00,00)
>>> end = datetime(2012,06,01, 00,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 6, 'second': 0, 'max_days': 183, 'year': 1, 'day': 0, 'minute': 0}
>>> start = datetime(2011,06,01, 00,00,00)
>>> end = datetime(2021,06,01, 00,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 120, 'second': 0, 'max_days': 3653, 'year': 10, 'day': 0, 'minute': 0}
>>> start = datetime(2011,06,01, 00,00,00)
>>> end = datetime(2012,06,01, 12,00,00)
>>> compute_datetime_delta(start, end)
{'hour': 8796, 'month': 12, 'second': 0, 'max_days': 366, 'year': 1, 'day': 0, 'minute': 0}
>>> start = datetime(2011,06,01, 00,00,00)
>>> end = datetime(2012,06,01, 12,30,00)
>>> compute_datetime_delta(start, end)
{'hour': 8796, 'month': 12, 'second': 0, 'max_days': 366, 'year': 1, 'day': 0, 'minute': 527790}
>>> start = datetime(2011,06,01, 00,00,00)
>>> end = datetime(2012,06,01, 12,00,05)
>>> compute_datetime_delta(start, end)
{'hour': 8796, 'month': 12, 'second': 31665605, 'max_days': 366, 'year': 1, 'day': 0, 'minute': 0}
>>> start = datetime(2011,06,01, 00,00,00)
>>> end = datetime(2012,06,01, 00,30,00)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 12, 'second': 0, 'max_days': 366, 'year': 1, 'day': 0, 'minute': 527070}
>>> start = datetime(2011,06,01, 00,00,00)
>>> end = datetime(2012,06,01, 00,00,05)
>>> compute_datetime_delta(start, end)
{'hour': 0, 'month': 12, 'second': 31622405, 'max_days': 366, 'year': 1, 'day': 0, 'minute': 0}
:return: A dictionary with year, month, day, hour, minute and second as
keys()
"""
comp = {}
day_diff = (end - start).days
comp["max_days"] = day_diff
# Date
# Count full years
d = end.year - start.year
comp["year"] = d
# Count full months
if start.month == 1 and end.month == 1:
comp["month"] = 0
elif start.day == 1 and end.day == 1:
d = end.month - start.month
if d < 0:
d = d + 12 * comp["year"]
elif d == 0:
d = 12 * comp["year"]
comp["month"] = d
# Count full days
if start.day == 1 and end.day == 1:
comp["day"] = 0
else:
comp["day"] = day_diff
# Time
# Hours
if start.hour == 0 and end.hour == 0:
comp["hour"] = 0
else:
d = end.hour - start.hour
if d < 0:
d = d + 24 + 24 * day_diff
else:
d = d + 24 * day_diff
comp["hour"] = d
# Minutes
if start.minute == 0 and end.minute == 0:
comp["minute"] = 0
else:
d = end.minute - start.minute
if d != 0:
if comp["hour"]:
d = d + 60 * comp["hour"]
else:
d = d + 24 * 60 * day_diff
elif d == 0:
if comp["hour"]:
d = 60 * comp["hour"]
else:
d = 24 * 60 * day_diff
comp["minute"] = d
# Seconds
if start.second == 0 and end.second == 0:
comp["second"] = 0
else:
d = end.second - start.second
if d != 0:
if comp["minute"]:
d = d + 60 * comp["minute"]
elif comp["hour"]:
d = d + 3600 * comp["hour"]
else:
d = d + 24 * 60 * 60 * day_diff
elif d == 0:
if comp["minute"]:
d = 60 * comp["minute"]
elif comp["hour"]:
d = 3600 * comp["hour"]
else:
d = 24 * 60 * 60 * day_diff
comp["second"] = d
return comp
###############################################################################
[docs]def check_datetime_string(time_string, use_dateutil=True):
"""Check if a string can be converted into a datetime object and return the object
In case datutil is not installed the supported ISO string formats are:
- YYYY-mm-dd
- YYYY-mm-dd HH:MM:SS
- YYYY-mm-ddTHH:MM:SS
- YYYY-mm-dd HH:MM:SS.s
- YYYY-mm-ddTHH:MM:SS.s
Time zones are not supported
If dateutil is installed, all string formats of the dateutil module
are supported, as well as time zones
Time zones are not supported
:param time_string: The time string to be checked for conversion
:param use_dateutil: Use dateutil if available for datetime string parsing
:return: datetime: object or an error message string in case of an error
>>> s = "2000-01-01"
>>> check_datetime_string(s)
datetime.datetime(2000, 1, 1, 0, 0)
>>> s = "2000-01-01T10:00:00"
>>> check_datetime_string(s)
datetime.datetime(2000, 1, 1, 10, 0)
>>> s = "2000-01-01 10:00:00"
>>> check_datetime_string(s)
datetime.datetime(2000, 1, 1, 10, 0)
>>> s = "2000-01-01T10:00:00.000001"
>>> check_datetime_string(s)
datetime.datetime(2000, 1, 1, 10, 0, 0, 1)
>>> s = "2000-01-01 10:00:00.000001"
>>> check_datetime_string(s)
datetime.datetime(2000, 1, 1, 10, 0, 0, 1)
# using native implementation, ignoring dateutil
>>> s = "2000-01-01"
>>> check_datetime_string(s, False)
datetime.datetime(2000, 1, 1, 0, 0)
>>> s = "2000-01-01T10:00:00"
>>> check_datetime_string(s, False)
datetime.datetime(2000, 1, 1, 10, 0)
>>> s = "2000-01-01 10:00:00"
>>> check_datetime_string(s, False)
datetime.datetime(2000, 1, 1, 10, 0)
>>> s = "2000-01-01T10:00:00.000001"
>>> check_datetime_string(s, False)
datetime.datetime(2000, 1, 1, 10, 0, 0, 1)
>>> s = "2000-01-01 10:00:00.000001"
>>> check_datetime_string(s, False)
datetime.datetime(2000, 1, 1, 10, 0, 0, 1)
"""
global has_dateutil
if has_dateutil and use_dateutil is True:
# First check if there is only a single number, which specifies
# relative time. dateutil will interpret a single number as a valid
# time string, so we have to catch this case beforehand
try:
value = int(time_string)
return _("Time string seems to specify relative time")
except ValueError:
pass
try:
time_object = parser.parse(time_string)
except Exception as inst:
time_object = str(inst)
return time_object
# BC is not supported
if "bc" in time_string > 0:
return _("Dates Before Christ (BC) are not supported")
# BC is not supported
if "+" in time_string:
return _("Time zones are not supported")
if ":" in time_string or "T" in time_string:
# Check for microseconds
if "." in time_string:
if "T" in time_string:
time_format = "%Y-%m-%dT%H:%M:%S.%f"
else:
time_format = "%Y-%m-%d %H:%M:%S.%f"
else:
if "T" in time_string:
time_format = "%Y-%m-%dT%H:%M:%S"
else:
time_format = "%Y-%m-%d %H:%M:%S"
else:
time_format = "%Y-%m-%d"
try:
return datetime.strptime(time_string, time_format)
except:
return _("Unable to parse time string: %s" % time_string)
###############################################################################
[docs]def string_to_datetime(time_string):
"""Convert a string into a datetime object
In case datutil is not installed the supported ISO string formats are:
- YYYY-mm-dd
- YYYY-mm-dd HH:MM:SS
- YYYY-mm-ddTHH:MM:SS
- YYYY-mm-dd HH:MM:SS.s
- YYYY-mm-ddTHH:MM:SS.s
Time zones are not supported
If dateutil is installed, all string formats of the dateutil module
are supported, as well as time zones
:param time_string: The time string to convert
:return: datetime object or None in case the string
could not be converted
"""
if not isinstance(time_string, unicode) and not isinstance(time_string, str):
return None
time_object = check_datetime_string(time_string)
if not isinstance(time_object, datetime):
msgr = get_tgis_message_interface()
msgr.error(str(time_object))
return None
return time_object
###############################################################################
[docs]def datetime_to_grass_datetime_string(dt):
"""Convert a python datetime object into a GRASS datetime string
.. code-block:: python
>>> import grass.temporal as tgis
>>> import dateutil.parser as parser
>>> dt = parser.parse("2011-01-01 10:00:00 +01:30")
>>> tgis.datetime_to_grass_datetime_string(dt)
'01 jan 2011 10:00:00 +0090'
>>> dt = parser.parse("2011-01-01 10:00:00 +02:30")
>>> tgis.datetime_to_grass_datetime_string(dt)
'01 jan 2011 10:00:00 +0150'
>>> dt = parser.parse("2011-01-01 10:00:00 +12:00")
>>> tgis.datetime_to_grass_datetime_string(dt)
'01 jan 2011 10:00:00 +0720'
>>> dt = parser.parse("2011-01-01 10:00:00 -01:30")
>>> tgis.datetime_to_grass_datetime_string(dt)
'01 jan 2011 10:00:00 -0090'
"""
# GRASS datetime month names
month_names = [
"",
"jan",
"feb",
"mar",
"apr",
"may",
"jun",
"jul",
"aug",
"sep",
"oct",
"nov",
"dec",
]
if dt is None:
raise Exception("Empty datetime object in datetime_to_grass_datetime_string")
# Check for time zone info in the datetime object
if dt.tzinfo is not None:
tz = dt.tzinfo.utcoffset(0)
if tz.seconds > 86400 / 2:
tz = (tz.seconds - 86400) / 60
else:
tz = tz.seconds / 60
string = "%.2i %s %.2i %.2i:%.2i:%.2i %+.4i" % (
dt.day,
month_names[dt.month],
dt.year,
dt.hour,
dt.minute,
dt.second,
tz,
)
else:
string = "%.2i %s %.4i %.2i:%.2i:%.2i" % (
dt.day,
month_names[dt.month],
dt.year,
dt.hour,
dt.minute,
dt.second,
)
return string
###############################################################################
suffix_units = {
"years": "%Y",
"year": "%Y",
"months": "%Y_%m",
"month": "%Y_%m",
"weeks": "%Y_%m_%d",
"week": "%Y_%m_%d",
"days": "%Y_%m_%d",
"day": "%Y_%m_%d",
"hours": "%Y_%m_%d_%H",
"hour": "%Y_%m_%d_%H",
"minutes": "%Y_%m_%d_%H_%M",
"minute": "%Y_%m_%d_%H_%M",
}
[docs]def create_suffix_from_datetime(start_time, granularity):
"""Create a datetime string based on a datetime object and a provided
granularity that can be used as suffix for map names.
dateteime=2001-01-01 00:00:00, granularity="1 month" returns "2001_01"
:param start_time: The datetime object
:param granularity: The granularity for example "1 month" or "100 seconds"
:return: A string
"""
global suffix_units
return start_time.strftime(suffix_units[granularity.split(" ")[1]])
[docs]def create_time_suffix(mapp, end=False):
"""Create a datetime string based on a map datetime object
:param mapp: a temporal map dataset
:param end: True if you want add also end time to the suffix
"""
start = mapp.temporal_extent.get_start_time()
sstring = start.isoformat().replace(":", "_").replace("-", "_")
if end:
end = mapp.temporal_extent.get_end_time()
estring = end.isoformat().replace(":", "_").replace("-", "_")
return "{st}_{en}".format(st=sstring, en=estring)
return sstring
[docs]def create_numeric_suffix(base, count, zeros):
"""Create a string based on count and number of zeros decided by zeros
:param base: the basename for new map
:param count: a number
:param zeros: a string containing the expected number, coming from suffix option like "%05"
"""
spli = zeros.split("%")
if len(spli) == 2:
suff = spli[1]
if suff.isdigit():
if int(suff[0]) == 0:
zero = suff
else:
zero = "0{nu}".format(nu=suff)
else:
zero = "05"
else:
zero = "05"
s = "{ba}_{i:" + zero + "d}"
return s.format(ba=base, i=count)
if __name__ == "__main__":
import doctest
doctest.testmod()
