class TemplateIntervalNode(TemplateNode):
def __init__(self, date_format="%c", style="days"):
TemplateNode.__init__(self)
self._interval_from = None
self._interval_to = None
if isinstance(date_format, str):
self._format = TemplateWordNode(date_format)
else:
self._format = date_format
if isinstance(style, str):
self._style = TemplateWordNode(style)
else:
self._style = style
@property
def format(self):
return self._format
@format.setter
def format(self, format):
self._format = format
@property
def interval_from(self):
return self._interval_from
@interval_from.setter
def interval_from(self, interval_from):
self._interval_from = interval_from
@property
def interval_to(self):
return self._interval_to
@interval_to.setter
def interval_to(self, interval_to):
self._interval_to = interval_to
@property
def style(self):
return self._style
@style.setter
def style(self, style):
self._style = style
def resolve(self, bot, clientid):
try:
format_str = self._format.resolve(bot, clientid)
from_str = self.interval_from.resolve(bot, clientid)
from_time = datetime.datetime.strptime(from_str, format_str)
to_str = self.interval_to.resolve(bot, clientid)
to_time = datetime.datetime.strptime(to_str, format_str)
style = self._style.resolve(bot, clientid)
diff = to_time - from_time
difference = relativedelta(to_time, from_time)
if style == "years":
resolved = str(difference.years)
elif style == "months":
resolved = str(difference.months)
elif style == "weeks":
resolved = str(difference.weeks)
elif style == "days":
resolved = str(diff.days)
elif style == "hours":
resolved = str(difference.hours)
elif style == "minutes":
resolved = str(difference.minutes)
elif style == "seconds":
resolved = str(difference.seconds)
elif style == "microseconds":
resolved = str(difference.microseconds)
elif style == "ymd":
resolved = "%d years, %d months, %d days" % \
(difference.years, difference.months, difference.days)
elif style == "hms":
resolved = "%d hours, %d minutes, %d seconds" % \
(difference.hours, difference.minutes, difference.seconds)
elif style == "ymdhms":
resolved = "%d years, %d months, %d days, %d hours, %d minutes, %d seconds" % \
(difference.years, difference.months, difference.days,
difference.hours, difference.minutes, difference.seconds)
else:
logging.error("Unknown interval style [%s]", style)
resolved = ""
logging.debug("[INTERVAL] resolved to [%s]", resolved)
return resolved
except Exception as excep:
logging.exception(excep)
return ""
def to_string(self):
return "[INTERVAL]"
def to_xml(self, bot, clientid):
xml = '<interval'
xml += ' format="%s"' % self._format.to_xml(bot, clientid)
xml += ' style="%s"' % self._style.to_xml(bot, clientid)
xml += '>'
xml += '<from>'
xml += self._interval_from.to_xml(bot, clientid)
xml += '</from>'
xml += '<to>'
xml += self._interval_to.to_xml(bot, clientid)
xml += '</to>'
xml += '</interval>'
return xml
class TemplateIntervalNode(TemplateNode):
def __init__(self, date_format="%c", style="days"):
TemplateNode.__init__(self)
self._interval_from = None
self._interval_to = None
if isinstance(date_format, str):
self._format = TemplateWordNode(date_format)
else:
self._format = date_format
if isinstance(style, str):
self._style = TemplateWordNode(style)
else:
self._style = style
@property
def format(self):
return self._format
@format.setter
def format(self, format):
self._format = format
@property
def interval_from(self):
return self._interval_from
@interval_from.setter
def interval_from(self, interval_from):
self._interval_from = interval_from
@property
def interval_to(self):
return self._interval_to
@interval_to.setter
def interval_to(self, interval_to):
self._interval_to = interval_to
@property
def style(self):
return self._style
@style.setter
def style(self, style):
self._style = style
def resolve(self, bot, clientid):
try:
format_str = self._format.resolve(bot, clientid)
from_str = self.interval_from.resolve(bot, clientid)
from_time = datetime.datetime.strptime(from_str, format_str)
to_str = self.interval_to.resolve(bot, clientid)
to_time = datetime.datetime.strptime(to_str, format_str)
style = self._style.resolve(bot, clientid)
diff = to_time - from_time
difference = relativedelta(to_time, from_time)
if style == "years":
resolved = str(difference.years)
elif style == "months":
resolved = str(difference.months)
elif style == "weeks":
resolved = str(difference.weeks)
elif style == "days":
resolved = str(diff.days)
elif style == "hours":
resolved = str(difference.hours)
elif style == "minutes":
resolved = str(difference.minutes)
elif style == "seconds":
resolved = str(difference.seconds)
elif style == "microseconds":
resolved = str(difference.microseconds)
elif style == "ymd":
resolved = "%d years, %d months, %d days" % \
(difference.years, difference.months, difference.days)
elif style == "hms":
resolved = "%d hours, %d minutes, %d seconds" % \
(difference.hours, difference.minutes, difference.seconds)
elif style == "ymdhms":
resolved = "%d years, %d months, %d days, %d hours, %d minutes, %d seconds" % \
(difference.years, difference.months, difference.days,
difference.hours, difference.minutes, difference.seconds)
else:
logging.error("Unknown interval style [%s]", style)
resolved = ""
logging.debug("[INTERVAL] resolved to [%s]", resolved)
return resolved
except Exception as excep:
logging.exception(excep)
return ""
def to_string(self):
return "[INTERVAL]"
def to_xml(self, bot, clientid):
xml = '<interval'
xml += ' format="%s"' % self._format.to_xml(bot, clientid)
xml += ' style="%s"' % self._style.to_xml(bot, clientid)
xml += '>'
xml += '<from>'
xml += self._interval_from.to_xml(bot, clientid)
xml += '</from>'
xml += '<to>'
xml += self._interval_to.to_xml(bot, clientid)
xml += '</to>'
xml += '</interval>'
return xml
def test_resolve_no_word(self):
node = TemplateWordNode(None)
self.assertEquals("", node.resolve(self._client_context))