~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to loggerhead/trace.py

  • Committer: Matt Nordhoff
  • Date: 2009-05-27 07:23:03 UTC
  • mto: This revision was merged to the branch mainline in revision 356.
  • Revision ID: mnordhoff@mattnordhoff.com-20090527072303-0499znz4mxgj1jzd
BetterĀ line-wrapping

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#
 
2
# Copyright (C) 2008  Canonical Ltd.
 
3
#                     (Authored by Martin Albisetti <argentina@gmail.com)
 
4
# Copyright (C) 2008  Guillermo Gonzalez <guillo.gonzo@gmail.com>
 
5
# Copyright (C) 2006  Robey Pointer <robey@lag.net>
 
6
# Copyright (C) 2006  Goffredo Baroncelli <kreijack@inwind.it>
 
7
#
 
8
# This program is free software; you can redistribute it and/or modify
 
9
# it under the terms of the GNU General Public License as published by
 
10
# the Free Software Foundation; either version 2 of the License, or
 
11
# (at your option) any later version.
 
12
#
 
13
# This program is distributed in the hope that it will be useful,
 
14
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
15
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
16
# GNU General Public License for more details.
 
17
#
 
18
# You should have received a copy of the GNU General Public License
 
19
# along with this program; if not, write to the Free Software
 
20
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
21
#
 
22
 
 
23
import os
 
24
import logging
 
25
import sys
 
26
 
 
27
 
 
28
def make_handler(config, filename):
 
29
    roll = config.get('log.roll', 'never')
 
30
    if roll == 'daily':
 
31
        h = logging.handlers.TimedRotatingFileHandler(filename,
 
32
                                                      'midnight',
 
33
                                                      1,
 
34
                                                      100)
 
35
    elif roll == 'weekly':
 
36
        h = logging.handlers.TimedRotatingFileHandler(filename, 'W0', 1, 100)
 
37
    else:
 
38
        h = logging.FileHandler(filename)
 
39
    return h
 
40
 
 
41
 
 
42
def setup_logging(log_folder, config, foreground):
 
43
    # i hate that stupid logging config format, so just set up logging here.
 
44
 
 
45
    if not os.path.exists(log_folder):
 
46
        os.mkdir(log_folder)
 
47
 
 
48
    f = logging.Formatter('%(levelname)-.3s [%(asctime)s.%(msecs)03d]'
 
49
                          ' %(name)s: %(message)s',
 
50
                          '%Y%m%d-%H:%M:%S')
 
51
    debug_log = make_handler(config, os.path.join(log_folder, 'debug.log'))
 
52
    debug_log.setLevel(logging.DEBUG)
 
53
    debug_log.setFormatter(f)
 
54
    if foreground:
 
55
        stdout_log = logging.StreamHandler(sys.stdout)
 
56
        stdout_log.setLevel(logging.DEBUG)
 
57
        stdout_log.setFormatter(f)
 
58
    f = logging.Formatter('[%(asctime)s.%(msecs)03d] %(message)s',
 
59
                          '%Y%m%d-%H:%M:%S')
 
60
    access_log = make_handler(config, os.path.join(log_folder, 'access.log'))
 
61
    access_log.setLevel(logging.INFO)
 
62
    access_log.setFormatter(f)
 
63
 
 
64
    error_log = make_handler(config, os.path.join(log_folder, 'error.log'))
 
65
    error_log.setLevel(logging.ERROR)
 
66
    error_log.setFormatter(f)
 
67
 
 
68
    logging.getLogger('').setLevel(logging.DEBUG)
 
69
    logging.getLogger('').addHandler(debug_log)
 
70
    logging.getLogger('loggerhead.access').addHandler(access_log)
 
71
    logging.getLogger('loggerhead').setLevel(logging.ERROR)
 
72
    logging.getLogger('loggerhead').addHandler(error_log)
 
73
 
 
74
    if foreground:
 
75
        logging.getLogger('').addHandler(stdout_log)