#!/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()