← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to start-loggerhead

  • Committer: Jelmer Vernooij
  • Date: 2008-12-15 21:23:24 UTC
  • mto: This revision was merged to the branch mainline in revision 256.
  • Revision ID: jelmer@samba.org-20081215212324-9dnuhxpcamgho4ci
Fix import of ConfigObj.

Show diffs side-by-side

added added

removed removed

Lines of Context:
start-loggerhead
1
 
#!/usr/bin/env python2.4
2
 
 
3
 
import pkg_resources
4
 
pkg_resources.require("TurboGears")
 
1
#!/usr/bin/env python
 
2
# This program is free software; you can redistribute it and/or modify
 
3
# it under the terms of the GNU General Public License as published by
 
4
# the Free Software Foundation; either version 2 of the License, or
 
5
# (at your option) any later version.
 
6
#
 
7
# This program is distributed in the hope that it will be useful,
 
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
9
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
10
# GNU General Public License for more details.
 
11
#
 
12
# You should have received a copy of the GNU General Public License
 
13
# along with this program; if not, write to the Free Software
 
14
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
 
 
16
"""A script for starting the loggerhead process."""
 
17
 
5
18
 
6
19
import logging
 
20
import logging.handlers
 
21
from optparse import OptionParser
7
22
import os
8
23
import sys
9
 
from optparse import OptionParser
10
 
 
11
 
from loggerhead.apps.config import Root
 
24
import urlparse
 
25
 
 
26
from bzrlib.util.configobj.configobj import ConfigObj
 
27
 
12
28
from paste import httpserver
13
29
from paste.httpexceptions import make_middleware
14
30
from paste.translogger import make_filter
15
31
 
16
 
from loggerhead import daemon, release
17
 
 
18
 
 
19
 
def make_handler(config, filename):
20
 
    roll = config.get('log.roll', 'never')
21
 
    if roll == 'daily':
22
 
        h = logging.handlers.TimedRotatingFileHandler(filename, 'midnight', 0, 100)
23
 
    elif roll == 'weekly':
24
 
        h = logging.handlers.TimedRotatingFileHandler(filename, 'W0', 0, 100)
25
 
    else:
26
 
        h = logging.FileHandler(filename)
27
 
    return h
28
 
 
29
 
 
30
 
def setup_logging(home, config, foreground):
31
 
    # i hate that stupid logging config format, so just set up logging here.
32
 
 
33
 
    log_folder = os.path.join(home, 'logs')
34
 
    if not os.path.exists(log_folder):
35
 
        os.mkdir(log_folder)
36
 
 
37
 
    f = logging.Formatter('%(levelname)-.3s [%(asctime)s.%(msecs)03d] %(name)s: %(message)s',
38
 
                          '%Y%m%d-%H:%M:%S')
39
 
    debug_log = make_handler(config, os.path.join(log_folder, 'debug.log'))
40
 
    debug_log.setLevel(logging.DEBUG)
41
 
    debug_log.setFormatter(f)
42
 
    if foreground:
43
 
        stdout_log = logging.StreamHandler(sys.stdout)
44
 
        stdout_log.setLevel(logging.DEBUG)
45
 
        stdout_log.setFormatter(f)
46
 
    f = logging.Formatter('[%(asctime)s.%(msecs)03d] %(message)s',
47
 
                          '%Y%m%d-%H:%M:%S')
48
 
    access_log = make_handler(config, os.path.join(log_folder, 'access.log'))
49
 
    access_log.setLevel(logging.INFO)
50
 
    access_log.setFormatter(f)
51
 
 
52
 
    logging.getLogger('').addHandler(debug_log)
53
 
    logging.getLogger('turbogears.access').addHandler(access_log)
54
 
    logging.getLogger('turbogears.controllers').setLevel(logging.INFO)
55
 
 
56
 
    if foreground:
57
 
        logging.getLogger('').addHandler(stdout_log)
58
 
 
 
32
from loggerhead import daemon
 
33
from loggerhead.apps.config import Root
 
34
from loggerhead.trace import make_handler, setup_logging
 
35
from loggerhead.apps.error import ErrorHandlerApp
59
36
 
60
37
 
61
38
def main():
62
 
    parser = OptionParser(usage='usage: %prog [options]', version='%prog ' + release.version)
63
 
    parser.add_option('-f', '--foreground', action='store_true', dest='foreground', default=False,
 
39
    home = os.path.realpath(os.path.dirname(__file__))
 
40
    default_pidfile = os.path.join(home, 'loggerhead.pid')
 
41
    default_configfile = os.path.join(home, 'loggerhead.conf')
 
42
    default_log_folder = os.path.join(home, 'logs')
 
43
    parser = OptionParser(usage='usage: %prog [options]', version='%prog')
 
44
    parser.add_option('-f', '--foreground',
 
45
                      action='store_true', dest='foreground', default=False,
64
46
                      help="run in the foreground; don't daemonize")
65
 
    parser.add_option('-C', '--check', action='store_true', dest='check', default=False,
66
 
                      help="only start if not already running (useful for cron jobs)")
 
47
    parser.add_option('-C', '--check', action='store_true',
 
48
                      dest='check', default=False,
 
49
                      help=("only start if not already running (useful for "
 
50
                            "cron jobs)"))
 
51
    parser.add_option('-p', '--pidfile', dest="pidfile",
 
52
                      default=default_pidfile,
 
53
                      type=str, help="override pidfile location")
 
54
    parser.add_option('-c', '--config-file', dest="configfile",
 
55
                      default=default_configfile,
 
56
                      type=str, help="override configuration file location")
 
57
    parser.add_option('-L', '--log-folder', dest="log_folder",
 
58
                      default=default_log_folder,
 
59
                      type=str, help="override log file directory")
67
60
    options, args = parser.parse_args()
68
61
    if len(args) > 0:
69
62
        parser.error('No filename arguments are used, only options.')
70
63
 
71
 
 
72
 
    home = os.path.realpath(os.path.dirname(__file__))
73
 
    pidfile = os.path.join(home, 'loggerhead.pid')
74
 
 
75
64
    if options.check:
76
 
        if daemon.is_running(pidfile):
 
65
        if daemon.is_running(options.pidfile):
77
66
            sys.exit(0)
78
 
        sys.stderr.write('Did not find loggerhead running in %r; restarting...\n' % (pidfile,))
 
67
        sys.stderr.write('Did not find loggerhead running in %r;' % (
 
68
                         options.pidfile))
 
69
        sys.stderr.write(' restarting...\n')
79
70
 
80
71
    # read loggerhead config
81
72
 
82
 
    from configobj import ConfigObj
83
 
    config = ConfigObj(os.path.join(home, 'loggerhead.conf'), encoding='utf-8')
 
73
    config = ConfigObj(options.configfile, encoding='utf-8')
84
74
    extra_path = config.get('bzrpath', None)
85
75
    if extra_path:
86
76
        sys.path.insert(0, extra_path)
87
77
 
88
 
    #turbogears.update_config(configfile="dev.cfg", modulename="loggerhead.config")
 
78
    potential_overrides = [('server.socket_port', int),
 
79
                           ('server.webpath', str),
 
80
                           ('server.thread_pool', int),
 
81
                           ('server.socket_host', str)]
 
82
    server_port = int(config.get('server.socket_port', 8080))
 
83
    nworkers = int(config.get('server.thread_pool', 10))
 
84
    server_host = config.get('server.socket_host', '0.0.0.0')
 
85
    webpath = config.get('server.webpath', None)
89
86
 
90
 
    potential_overrides = [ ('server.socket_port', int),
91
 
                            ('server.webpath', str),
92
 
                            ('server.thread_pool', int),
93
 
                            ('server.socket_host' ,str) ]
94
87
    for key, keytype in potential_overrides:
95
88
        value = config.get(key, None)
96
89
        if value is not None:
97
90
            value = keytype(value)
98
 
            #turbogears.config.update({ key: value })
99
 
 
100
 
    ## if not options.foreground:
101
 
    ##     sys.stderr.write('\n')
102
 
    ##     sys.stderr.write('Launching loggerhead into the background.\n')
103
 
    ##     sys.stderr.write('PID file: %s\n' % (pidfile,))
104
 
    ##     sys.stderr.write('\n')
105
 
 
106
 
    ##     daemon.daemonize(pidfile, home)
107
 
 
108
 
    setup_logging(home, config, foreground=options.foreground)
 
91
 
 
92
    if not options.foreground:
 
93
        sys.stderr.write('\n')
 
94
        sys.stderr.write('Launching loggerhead into the background.\n')
 
95
        sys.stderr.write('PID file: %s\n' % options.pidfile)
 
96
        sys.stderr.write('\n')
 
97
 
 
98
        daemon.daemonize(options.pidfile, home)
 
99
 
 
100
    setup_logging(options.log_folder, config, foreground=options.foreground)
109
101
 
110
102
    log = logging.getLogger('loggerhead')
111
103
    log.info('Starting up...')
114
106
 
115
107
    app = app
116
108
    app = make_middleware(app)
117
 
    app = make_filter(app, None)
118
 
 
119
 
    httpserver.serve(app, host='127.0.0.1', port='9876')
 
109
    app = make_filter(app, None, logger_name=log.name+'.access')
 
110
    app = ErrorHandlerApp(app)
 
111
 
 
112
    if webpath:
 
113
        scheme, netloc, path, blah, blah, blah = urlparse.urlparse(webpath)
 
114
 
 
115
        def app(environ, start_response, orig=app):
 
116
            environ['SCRIPT_NAME'] = path
 
117
            environ['HTTP_HOST'] = netloc
 
118
            return orig(environ, start_response)
 
119
 
 
120
    try:
 
121
        httpserver.serve(
 
122
            app, host=server_host, port=server_port,
 
123
            threadpool_workers=nworkers)
 
124
    finally:
 
125
        log.info('Shutdown.')
 
126
        try:
 
127
            os.remove(options.pidfile)
 
128
        except OSError:
 
129
            pass
120
130
 
121
131
 
122
132
if __name__ == '__main__':