#!/usr/bin/env python

 __author__ = "Artur Duque de Souza"

 __author_email__ = "artur.souza@indt.org.br"

 __license__ = "GPL"

 __version__ = "0.1"

 import os

 import logging

 __all__ = ("Log", "log_structure")

 class log_structure(object):

     """Structure to hold log info."""

     def __init__(self, log=None):

         self.log = log

         self.history = []

     # __init__()

 # log_structure()

 class Log(object):

     """This class implements a log where we can store status of

     all transcoders (even from those that are not running any more)."""

     ## key = tid

     ## item = ls

     logs = {}

     def insert(self, tid, name):

         """Insert a given tid on the log structure"""

         if not self.logs.has_key(tid):

             self.logs[tid] = log_structure(logging.getLogger(name))

             return True

         else:

             return False

     # insert()

     def remove(self, tid=None):

         """Cleans up all log stored for a

         given tid or for all transcodes."""

         if not tid:

             self.logs = {}

         else:

             del(self.logs[tid])

         return True

     # clean()

     def get_status(self, tid=None, all=False):

         """Get the status of all transcoders or

         of just one of them if it's given an tid."""

         if not tid:

             ret = {}

             for tids, logs in self.logs.items():

                 if len(logs.history) > 0:

                     if not all:

                         ret[tids] = logs.history[-1]

                     else:

                         ret[tids] = logs.history

             return ret

         elif self.logs.has_key(tid) and len(self.logs[tid].history) > 0:

             if not all:

                 return self.logs[tid].history[-1]

             else:

                 return self.logs[tid].history

         return False

     # get_status()

     def _update_status(self, tid=None, msg=""):

         """Update the status of a given tid. Private method that

         is only called inside error/info/debug wrappers"""

         if msg != "":

             self.logs[tid].history.append(msg)

             return True

         else:

             return False

     # update_status()

     def error(self, tid, msg):

         """Python's Log.error wrapper"""

         if self.logs.has_key(tid):

             self.logs[tid].log.error("%s" % msg)

             return self._update_status(tid, msg)

         else:

             return False

     # error()

     def info(self, tid, msg):

         """Python's Log.info wrapper"""

         if self.logs.has_key(tid):

             self.logs[tid].log.info("%s" % msg)

             self._update_status(tid, msg)

             return True

         else:

             return False

     # info()

     def debug(self, tid, msg):

         """Python's Log.debug wrapper"""

         if self.logs.has_key(tid):

             self.logs[tid].log.debug("%s" % msg)

             self._update_status(tid, msg)

             return True

         else:

             return False

     # debug()