logger.py
No OneTemporary
Actions

Authored By

Unknown

Size

6 KB

Referenced Files

None

Subscribers

None

logger.py
View Options

	# -- coding: utf-8 --
	# Copyright 2010-2013 Kolab Systems AG (http://www.kolabsys.com)
	#
	# Jeroen van Meeuwen (Kolab Systems) <vanmeeuwen a kolabsys.com>
	#
	# This program is free software: you can redistribute it and/or modify
	# it under the terms of the GNU General Public License as published by
	# the Free Software Foundation, either version 3 of the License, or
	# (at your option) any later version.

	# This program is distributed in the hope that it will be useful,
	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	# GNU General Public License for more details.

	# You should have received a copy of the GNU General Public License
	# along with this program. If not, see <http://www.gnu.org/licenses/>.
	#

	import grp
	import logging
	import logging.handlers
	import os
	import pwd
	import sys
	import time

	from pykolab.translate import _

	class Logger(logging.Logger):
	"""
	The PyKolab version of a logger.

	This class wraps the Python native logging library, adding to the
	loglevel capabilities, a debuglevel capability.
	"""
	debuglevel = 0
	fork = False
	loglevel = logging.CRITICAL
	process_username = 'kolab'
	process_groupname = 'kolab-n'

	if hasattr(sys, 'argv'):
	for arg in sys.argv:
	if debuglevel == -1:
	try:
	debuglevel = int(arg)
	except ValueError, errmsg:
	continue

	loglevel = logging.DEBUG
	break

	if '-d' == arg:
	debuglevel = -1
	continue

	if '-l' == arg:
	loglevel = -1
	continue

	if '--fork' == arg:
	fork = True

	if loglevel == -1:
	if hasattr(logging,arg.upper()):
	loglevel = getattr(logging,arg.upper())
	else:
	loglevel = logging.DEBUG

	if '-u' == arg or '--user' == arg:
	process_username = -1
	continue

	if arg.startswith('--user='):
	process_username = arg.split('=')[1]

	if process_username == -1:
	process_username = arg

	if '-g' == arg or '--group' == arg:
	process_groupname = -1
	continue

	if arg.startswith('--group='):
	process_groupname = arg.split('=')[1]

	if process_groupname == -1:
	process_groupname = arg

	def __init__(self, args, *kw):
	if kw.has_key('name'):
	name = kw['name']
	elif len(args) == 1:
	name = args[0]
	else:
	name = 'pykolab'

	logging.Logger.__init__(self, name)

	plaintextformatter = logging.Formatter("%(asctime)s %(name)s %(levelname)s %(message)s")

	if not self.fork:
	self.console_stdout = logging.StreamHandler(sys.stdout)
	self.console_stdout.setFormatter(plaintextformatter)

	self.addHandler(self.console_stdout)

	if kw.has_key('logfile'):
	self.logfile = kw['logfile']
	else:
	self.logfile = '/var/log/kolab/pykolab.log'

	group_gid = 0
	user_uid = 0

	# Make sure (read: attempt to change) the permissions
	try:
	try:
	(ruid, euid, suid) = os.getresuid()
	(rgid, egid, sgid) = os.getresgid()
	except AttributeError, errmsg:
	ruid = os.getuid()
	rgid = os.getgid()

	if ruid == 0:
	# Means we can setreuid() / setregid() / setgroups()
	if rgid == 0:
	# Get group entry details
	try:
	(
	group_name,
	group_password,
	group_gid,
	group_members
	) = grp.getgrnam(self.process_groupname)

	except KeyError, errmsg:
	group_name = False

	if ruid == 0:
	# Means we haven't switched yet.
	try:
	(
	user_name,
	user_password,
	user_uid,
	user_gid,
	user_gecos,
	user_homedir,
	user_shell
	) = pwd.getpwnam(self.process_username)

	except KeyError, errmsg:
	user_name = False

	if os.path.isfile(self.logfile):
	try:
	if not user_uid == 0 or group_gid == 0:
	os.chown(
	self.logfile,
	user_uid,
	group_gid
	)
	os.chmod(self.logfile, 0660)

	except Exception, errmsg:
	self.error(_("Could not change permissions on %s: %r") % (self.logfile, errmsg))
	if self.debuglevel > 8:
	import traceback
	traceback.print_exc()

	except Exception, errmsg:
	if os.path.isfile(self.logfile):
	self.error(_("Could not change permissions on %s: %r") % (self.logfile, errmsg))
	if self.debuglevel > 8:
	import traceback
	traceback.print_exc()

	# Make sure the log file exists
	try:
	fhandle = file(self.logfile, 'a')
	try:
	os.utime(self.logfile, None)
	finally:
	fhandle.close()

	try:
	filelog_handler = logging.FileHandler(filename=self.logfile)
	filelog_handler.setFormatter(plaintextformatter)
	except IOError, e:
	print >> sys.stderr, _("Cannot log to file %s: %s") % (self.logfile, e)

	if not len(self.handlers) > 1:
	try:
	self.addHandler(filelog_handler)
	except:
	pass

	except IOError, errmsg:
	pass

	def remove_stdout_handler(self):
	if not self.fork:
	self.console_stdout.close()
	self.removeHandler(self.console_stdout)

	def debug(self, msg, level=1, args, *kw):
	self.setLevel(self.loglevel)
	# Work around other applications not using various levels of debugging
	if not self.name.startswith('pykolab') and not self.debuglevel == 9:
	return

	if level <= self.debuglevel:
	# TODO: Not the way it's supposed to work!
	self.log(logging.DEBUG, '[%d]: %s' % (os.getpid(),msg))


	logging.setLoggerClass(Logger)

File Metadata

Mime Type: text/x-script.python
Expires: Sun, Apr 5, 10:37 PM (2 w, 4 d ago)
Storage Engine: local-disk
Storage Format: Raw Data
Storage Handle: e2/b7/275f3898c3427b2b0f168a7447b8
Default Alt Text: logger.py (6 KB)

logger.pyNo OneTemporaryActions

logger.pyView Options

File Metadata

Event Timeline

logger.py
No OneTemporary
Actions

logger.py
View Options