shinken/log.py

   1 #!/usr/bin/env python
   2 #Copyright (C) 2009-2010 :
   3 #    Gabes Jean, naparuba@gmail.com
   4 #    Gerhard Lausser, Gerhard.Lausser@consol.de
   5 #    Gregory Starck, g.starck@gmail.com
   6 #    Hartmut Goebel, h.goebel@goebel-consult.de
   7 #
   8 #This file is part of Shinken.
   9 #
  10 #Shinken is free software: you can redistribute it and/or modify
  11 #it under the terms of the GNU Affero General Public License as published by
  12 #the Free Software Foundation, either version 3 of the License, or
  13 #(at your option) any later version.
  14 #
  15 #Shinken is distributed in the hope that it will be useful,
  16 #but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 #GNU Affero General Public License for more details.
  19 #
  20 #You should have received a copy of the GNU Affero General Public License
  21 #along with Shinken.  If not, see <http://www.gnu.org/licenses/>.
  22
  23 import time
  24
  25 from brok import Brok
  26
  27 obj = None
  28 name = None
  29 local_log = None
  30
  31 class Log:
  32     # We load the object where we will put log broks
  33     # with the 'add' method
  34     def load_obj(self, object, name_ = None):
  35         global obj
  36         global name
  37         obj = object
  38         name = name_
  39
  40     # We enter a log message, we format it, and we add the log brok
  41     def log(self, message, format = None):
  42         global obj
  43         global name
  44         global local_log
  45
  46         print message
  47         if format is None:
  48             if name is None:
  49             # We format the log in UTF-8
  50                 s = u'[%d] %s\n' % (int(time.time()), message.decode('UTF-8', 'replace'))
  51             else:
  52                 s = u'[%d] [%s] %s\n' % (int(time.time()), name, message.decode('UTF-8', 'replace'))
  53         else:
  54             s = format % message
  55
  56         # We create and add the brok
  57         b = Brok('log', {'log' : s})
  58         obj.add(b)
  59
  60         # If we want a local log write, do it
  61         if local_log is not None:
  62             local_log.write(s)
  63             local_log.flush()
  64
  65
  66     # The log can also write to a local file if need
  67     def register_local_log(self, path):
  68         global local_log
  69         local_log = open(path, 'a')
  70
  71
  72     # Clsoe the local log file at program exit
  73     def quit(self):
  74         global local_log
  75         if local_log:
  76             local_log.close()
  77
  78
  79 logger = Log()