~loggerhead-team/loggerhead/trunk-rich

161 by Michael Hudson
don't insist on python 2.4
1
#!/usr/bin/env python
183.2.1 by John Arbash Meinel
Add Copyright information to most files.
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
89 by Robey Pointer
fix up dev.cfg so that nobody will ever have to edit it, by letting the
18
93 by Robey Pointer
slight cleanup, and add '-f' option for running in the foreground.
19
import logging
166.1.2 by Michael Hudson
more loggerhead.conf compatible vhosting
20
from optparse import OptionParser
89 by Robey Pointer
fix up dev.cfg so that nobody will ever have to edit it, by letting the
21
import os
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
22
import sys
166.1.2 by Michael Hudson
more loggerhead.conf compatible vhosting
23
import urlparse
89 by Robey Pointer
fix up dev.cfg so that nobody will ever have to edit it, by letting the
24
270.1.1 by Michael Hudson
defer loading of plugins until after we've set up the logging
25
from bzrlib.plugin import load_plugins
255.1.1 by Jelmer Vernooij
Fix import of ConfigObj.
26
from bzrlib.util.configobj.configobj import ConfigObj
159.2.50 by Michael Hudson
reorder imports in start-loggerhead.py
27
159.2.28 by Michael Hudson
woohoo it works
28
from paste import httpserver
29
from paste.httpexceptions import make_middleware
30
from paste.translogger import make_filter
1 by Robey Pointer
initial checkin
31
204 by Martin Albisetti
Clean up unused/inexistant import
32
from loggerhead import daemon
159.2.50 by Michael Hudson
reorder imports in start-loggerhead.py
33
from loggerhead.apps.config import Root
329.1.3 by Matt Nordhoff
Clean up some unused imports and syntax things.
34
from loggerhead.trace import setup_logging
217.1.4 by Guillermo Gonzalez
* new apps module: "error" and ErrorHandlerApp middleware
35
from loggerhead.apps.error import ErrorHandlerApp
93 by Robey Pointer
slight cleanup, and add '-f' option for running in the foreground.
36
230.1.1 by Steve 'Ashcrow' Milner
Updated to follow pep8.
37
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
38
def main():
197.1.4 by Jelmer Vernooij
Add --pidfile argument to start-loggerhead.
39
    home = os.path.realpath(os.path.dirname(__file__))
40
    default_pidfile = os.path.join(home, 'loggerhead.pid')
197.1.5 by Jelmer Vernooij
Add --configfile argument.
41
    default_configfile = os.path.join(home, 'loggerhead.conf')
197.1.6 by Jelmer Vernooij
add --log-folder option.
42
    default_log_folder = os.path.join(home, 'logs')
205 by Martin Albisetti
Further cleaning
43
    parser = OptionParser(usage='usage: %prog [options]', version='%prog')
230.1.1 by Steve 'Ashcrow' Milner
Updated to follow pep8.
44
    parser.add_option('-f', '--foreground',
45
                      action='store_true', dest='foreground', default=False,
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
46
                      help="run in the foreground; don't daemonize")
230.1.1 by Steve 'Ashcrow' Milner
Updated to follow pep8.
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,
197.2.1 by Jelmer Vernooij
Fix syntax.
53
                      type=str, help="override pidfile location")
230.1.1 by Steve 'Ashcrow' Milner
Updated to follow pep8.
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,
197.2.1 by Jelmer Vernooij
Fix syntax.
59
                      type=str, help="override log file directory")
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
60
    options, args = parser.parse_args()
61
    if len(args) > 0:
62
        parser.error('No filename arguments are used, only options.')
159.2.27 by Michael Hudson
begin a compatibility app
63
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
64
    if options.check:
197.1.4 by Jelmer Vernooij
Add --pidfile argument to start-loggerhead.
65
        if daemon.is_running(options.pidfile):
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
66
            sys.exit(0)
230.1.1 by Steve 'Ashcrow' Milner
Updated to follow pep8.
67
        sys.stderr.write('Did not find loggerhead running in %r;' % (
68
                         options.pidfile))
69
        sys.stderr.write(' restarting...\n')
159.2.27 by Michael Hudson
begin a compatibility app
70
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
71
    # read loggerhead config
159.2.27 by Michael Hudson
begin a compatibility app
72
197.1.5 by Jelmer Vernooij
Add --configfile argument.
73
    config = ConfigObj(options.configfile, encoding='utf-8')
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
74
    extra_path = config.get('bzrpath', None)
75
    if extra_path:
76
        sys.path.insert(0, extra_path)
159.2.27 by Michael Hudson
begin a compatibility app
77
230.1.1 by Steve 'Ashcrow' Milner
Updated to follow pep8.
78
    potential_overrides = [('server.socket_port', int),
79
                           ('server.webpath', str),
80
                           ('server.thread_pool', int),
81
                           ('server.socket_host', str)]
159.2.36 by Michael Hudson
more compatibility
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)
86
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
87
    for key, keytype in potential_overrides:
88
        value = config.get(key, None)
89
        if value is not None:
90
            value = keytype(value)
159.2.28 by Michael Hudson
woohoo it works
91
159.2.29 by Michael Hudson
re-enable daemonization, remove require("TurboGears")
92
    if not options.foreground:
93
        sys.stderr.write('\n')
94
        sys.stderr.write('Launching loggerhead into the background.\n')
230.1.1 by Steve 'Ashcrow' Milner
Updated to follow pep8.
95
        sys.stderr.write('PID file: %s\n' % options.pidfile)
159.2.29 by Michael Hudson
re-enable daemonization, remove require("TurboGears")
96
        sys.stderr.write('\n')
159.2.28 by Michael Hudson
woohoo it works
97
197.1.4 by Jelmer Vernooij
Add --pidfile argument to start-loggerhead.
98
        daemon.daemonize(options.pidfile, home)
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
99
197.1.6 by Jelmer Vernooij
add --log-folder option.
100
    setup_logging(options.log_folder, config, foreground=options.foreground)
159.2.27 by Michael Hudson
begin a compatibility app
101
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
102
    log = logging.getLogger('loggerhead')
103
    log.info('Starting up...')
159.2.27 by Michael Hudson
begin a compatibility app
104
159.2.28 by Michael Hudson
woohoo it works
105
    app = Root(config)
106
107
    app = app
108
    app = make_middleware(app)
217.1.4 by Guillermo Gonzalez
* new apps module: "error" and ErrorHandlerApp middleware
109
    app = make_filter(app, None, logger_name=log.name+'.access')
110
    app = ErrorHandlerApp(app)
159.2.28 by Michael Hudson
woohoo it works
111
159.2.36 by Michael Hudson
more compatibility
112
    if webpath:
166.1.2 by Michael Hudson
more loggerhead.conf compatible vhosting
113
        scheme, netloc, path, blah, blah, blah = urlparse.urlparse(webpath)
230.1.1 by Steve 'Ashcrow' Milner
Updated to follow pep8.
114
166.1.2 by Michael Hudson
more loggerhead.conf compatible vhosting
115
        def app(environ, start_response, orig=app):
116
            environ['SCRIPT_NAME'] = path
117
            environ['HTTP_HOST'] = netloc
266 by Michael Hudson
apply neror's patch to fix bug #260547 ("start-loggerhead script doesn't properly set the wsgi.url_scheme from the server.webpath option")
118
            environ['wsgi.url_scheme'] = scheme
166.1.2 by Michael Hudson
more loggerhead.conf compatible vhosting
119
            return orig(environ, start_response)
159.2.36 by Michael Hudson
more compatibility
120
270.1.1 by Michael Hudson
defer loading of plugins until after we've set up the logging
121
    load_plugins()
122
159.2.45 by Michael Hudson
testing is grate
123
    try:
124
        httpserver.serve(
125
            app, host=server_host, port=server_port,
126
            threadpool_workers=nworkers)
127
    finally:
128
        log.info('Shutdown.')
129
        try:
197.1.4 by Jelmer Vernooij
Add --pidfile argument to start-loggerhead.
130
            os.remove(options.pidfile)
159.2.45 by Michael Hudson
testing is grate
131
        except OSError:
132
            pass
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
133
134
135
if __name__ == '__main__':
136
    main()