class TemplateDateNode(TemplateAttribNode):
def __init__(self, date_format=None, locale=None):
TemplateAttribNode.__init__(self)
if date_format is None:
self._format = TemplateWordNode("%c")
else:
self._format = TemplateWordNode(date_format)
self._locale = None
if locale is not None:
self._locale = TemplateWordNode(locale)
@property
def date_format(self):
return self._format
@date_format.setter
def date_format(self, fmt):
if isinstance(fmt, TemplateNode):
self._format = fmt
else:
self._format = TemplateWordNode(str(fmt))
@property
def locale(self):
return self._locale
@locale.setter
def locale(self, lcl):
if isinstance(lcl, TemplateNode):
self._locale = lcl
else:
self._locale = TemplateWordNode(str(lcl))
def _set_locate(self, client_context):
local_locale = self._locale.resolve_to_string(client_context)
locale.setlocale(locale.LC_TIME, local_locale)
return local_locale
def resolve_to_string(self, client_context):
time_now = datetime.datetime.now()
local_time = None
if self._locale is not None:
local_time = locale.setlocale(locale.LC_TIME)
local_locale = ""
resolved = ""
try:
if self._locale is not None:
local_locale = self._set_locate(client_context)
resolved_format = self._format.resolve_to_string(client_context)
resolved = time_now.strftime(resolved_format)
except Exception as exep:
YLogger.exception(self, "Failed to set locale to [%s]", exep, local_locale)
finally:
if local_time is not None:
locale.setlocale(locale.LC_TIME, local_time)
YLogger.debug(client_context, "[%s] resolved to [%s]", self.to_string(), resolved)
return resolved
def to_string(self):
if self._locale is None:
return "[DATE format=%s]" % self._format.to_string()
else:
return "[DATE format=%s locale=%s]" % (self._format.to_string(), self._locale.to_string())
def set_attrib(self, attrib_name, attrib_value):
if attrib_name == 'format':
if isinstance(attrib_value, TemplateNode):
self._format = attrib_value
else:
self._format = TemplateWordNode(attrib_value)
elif attrib_name == 'locale':
if isinstance(attrib_value, TemplateNode):
self._locale = attrib_value
else:
self._locale = TemplateWordNode(attrib_value)
else:
raise ParserException("Invalid attribute name %s for this node" % (attrib_name))
def to_xml(self, client_context):
if self._locale is None:
xml = '<date format="%s" >' % self._format.to_xml(client_context)
else:
xml = '<date format="%s" locale="%s">' % (self._format.to_xml(client_context),
self._locale.to_xml(client_context))
xml += self.children_to_xml(client_context)
xml += "</date>"
return xml
#######################################################################################################
# DATE_ATTRIBUTES ::== (format="LISP_DATE_FORMAT") | (jformat="JAVA DATE FORMAT")
# DATE_ATTRIBUTE_TAG ::== <format>TEMPLATE_EXPRESSION</format> | <jformat>TEMPLATE_EXPRESSION</jformat>
# DATE_EXPRESSION ::== <date( DATE_ATTRIBUTES)*/> | <date>(DATE_ATTRIBUTE_TAG)</date>
# Pandorabots supports three extension attributes to the date element in templates:
# locale
# format
# timezone
def parse_expression(self, graph, expression):
self._parse_node_with_attribs(graph, expression, [["format", "%c"], ["locale", None]])
class TemplateLogNode(TemplateAttribNode):
def __init__(self):
TemplateAttribNode.__init__(self)
self._level = TemplateWordNode("debug")
self._output = TemplateWordNode("logging")
@property
def level(self):
return self._level
@level.setter
def level(self, level):
self._level = level
def resolve_to_string(self, client_context):
resolved = self.resolve_children_to_string(client_context)
output = self._output.resolve_to_string(client_context)
level = self._level.resolve_to_string(client_context)
if output == "logging":
YLogger.debug(client_context, "[%s] resolved to [%s]", self.to_string(), resolved)
if level == "debug":
YLogger.debug(client_context, resolved)
elif level == "warning":
YLogger.warning(client_context, resolved)
elif level == "error":
YLogger.error(client_context, resolved)
elif level == "info":
YLogger.info(client_context, resolved)
else:
YLogger.info(client_context, resolved)
else:
print(resolved)
return ""
def to_string(self):
return "[LOG level=%s]" % (self._level.to_string())
def set_attrib(self, attrib_name, attrib_value):
if attrib_name != 'level' and attrib_name != 'output':
raise ParserException("Invalid attribute name %s for this node", attrib_name)
if attrib_name == 'level':
if isinstance(attrib_value, TemplateWordNode):
self._level = attrib_value
else:
self._level = TemplateWordNode(attrib_value)
if attrib_name == 'output':
if isinstance(attrib_value, TemplateWordNode):
self._output = attrib_value
else:
self._output = TemplateWordNode(attrib_value)
def to_xml(self, client_context):
xml = "<log"
if self._level is not None:
xml += ' level="%s"' % self._level.to_xml(client_context)
xml += ">"
xml += self.children_to_xml(client_context)
xml += "</log>"
return xml
#######################################################################################################
# LOG_EXPRESSION ::== <log>Message</log>
# <log level="error|warning|debug|info">Message</log>
#
def parse_expression(self, graph, expression):
self._parse_node_with_attribs(graph, expression, [["level", "debug"],["output", "logging"]])
def test_resolve_to_string_no_word(self):
node = TemplateWordNode(None)
self.assertEquals("", node.resolve_to_string(self._client_context))
