terom@50: """
terom@50:     Format LogLines into some other representation
terom@50: """
terom@50: 
terom@50: from qmsk.web import helpers
terom@50: 
terom@50: from log_line import LogTypes
terom@50: 
terom@50: class LogFormatter (object) :
terom@50:     """
terom@50:         Provides a method to format series of LogLines into various output formats, with varying themes
terom@50:     """
terom@50:     
terom@50:     def __init__ (self, tz, timestamp_fmt="%H:%M:%S") :
terom@50:         """
terom@50:             Initialize to format timestamps with the given timezone and timestamp
terom@50:         """
terom@50: 
terom@50:         self.tz = tz
terom@50:         self.timestamp_fmt = timestamp_fmt
terom@50:     
terom@50:     def _format_line_text (self, line, template_dict) :
terom@50:         """
terom@50:             Format the given line as text, using the given { type: string template } dict
terom@50:         """
terom@50:         
terom@50:         # look up the template
terom@50:         template = template_dict[line.type]
terom@50: 
terom@50:         # build timestamp
terom@50:         timestamp = line.timestamp.astimezone(self.tz).strftime(self.timestamp_fmt)
terom@50:         
terom@50:         # format with dict
terom@50:         return template % dict(
terom@50:             timestamp       = timestamp,
terom@50:             source          = line.source,
terom@50:             data            = line.data,
terom@50:         )
terom@50:     
terom@50:     def format_txt (self, lines) :
terom@50:         """
terom@50:             Format as plaintext
terom@50:         """
terom@50: 
terom@50:         abstract
terom@50: 
terom@50:     def format_html (self, lines) :
terom@50:         """
terom@50:             Format as HTML
terom@50:         """
terom@50: 
terom@50:         abstract
terom@50: 
terom@50: class IrssiTextFormatter (LogFormatter) :
terom@50:     """
terom@50:         Implements format_txt for irssi-style output
terom@50:     """
terom@50: 
terom@50:     # format definitions by type
terom@50:     __FMT = {
terom@50:         LogTypes.RAW        : "%(timestamp)s %(data)s",
terom@50:     }
terom@50: 
terom@50:     def format_txt (self, lines) :
terom@50:         # ...handle each line
terom@50:         for line in lines :
terom@50:             # using __TYPES
terom@50:             yield self._format_line_text(line, self.__FMT)
terom@50: 
terom@50: class IrssiFormatter (IrssiTextFormatter) :
terom@50:     """
terom@50:         Implements plain black-and-white irssi-style formatting
terom@50:     """
terom@50: 
terom@50:     def format_html (self, lines) :
terom@50:         """
terom@50:             Just uses format_txt, but wraps in <pre></pre>
terom@50:         """
terom@50:         
terom@50:         # open pre
terom@50:         yield "<pre>"
terom@50:         
terom@50:         # format using IrssiTextFormatter
terom@50:         for line in self.format_txt(lines) :
terom@50:             # escape HTML
terom@50:             yield helpers.escape(line)
terom@50: 
terom@50:         # close pre
terom@50:         yield "</pre>"
terom@50: 
terom@50: # define formatters by name
terom@50: FORMATTERS = {
terom@50:     'irssi':        IrssiFormatter,
terom@50: }
terom@50: 
terom@50: def by_name (name) :
terom@50:     """
terom@50:         Lookup and return a formatter by name
terom@50:         
terom@50:         XXX: uses default timezone/timefmt
terom@50:     """
terom@50: 
terom@50:     import pytz
terom@50: 
terom@50:     return FORMATTERS[name](pytz.utc)
terom@50: