~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
159.2.46 by Michael Hudson
add explicit import of logging.handlers
20
import logging.handlers
166.1.2 by Michael Hudson
more loggerhead.conf compatible vhosting
21
from optparse import OptionParser
89 by Robey Pointer
fix up dev.cfg so that nobody will ever have to edit it, by letting the
22
import os
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
23
import sys
166.1.2 by Michael Hudson
more loggerhead.conf compatible vhosting
24
import urlparse
89 by Robey Pointer
fix up dev.cfg so that nobody will ever have to edit it, by letting the
25
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
217.1.1 by Guillermo Gonzalez
* refactor of logging-related functions (setup_logging and make_handler) into (new) trace module
34
from loggerhead.trace import make_handler, 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
159.2.45 by Michael Hudson
testing is grate
121
    try:
122
        httpserver.serve(
123
            app, host=server_host, port=server_port,
124
            threadpool_workers=nworkers)
125
    finally:
126
        log.info('Shutdown.')
127
        try:
197.1.4 by Jelmer Vernooij
Add --pidfile argument to start-loggerhead.
128
            os.remove(options.pidfile)
159.2.45 by Michael Hudson
testing is grate
129
        except OSError:
130
            pass
128.2.3 by Robey Pointer
add a '-C' option for ensuring loggerhead is running (from a cron job, for
131
132
133
if __name__ == '__main__':
134
    main()