← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to start-loggerhead

  • Committer: Andrew Bennetts
  • Date: 2011-06-28 13:13:05 UTC
  • Revision ID: andrew.bennetts@canonical.com-20110628131305-skv4h2z02z3iw8wc
Extract some refactoring of TemplatedBranchView and BranchWSGIApp from the jsonify branch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
start-loggerhead
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
 
 
18
 
 
19
 
import logging
20
 
import logging.handlers
21
 
from optparse import OptionParser
22
 
import os
23
 
import sys
24
 
import urlparse
25
 
 
26
 
from configobj import ConfigObj
27
 
 
28
 
from paste import httpserver
29
 
from paste.httpexceptions import make_middleware
30
 
from paste.translogger import make_filter
31
 
 
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
36
 
 
37
 
def main():
38
 
    home = os.path.realpath(os.path.dirname(__file__))
39
 
    default_pidfile = os.path.join(home, 'loggerhead.pid')
40
 
    default_configfile = os.path.join(home, 'loggerhead.conf')
41
 
    default_log_folder = os.path.join(home, 'logs')
42
 
    parser = OptionParser(usage='usage: %prog [options]', version='%prog')
43
 
    parser.add_option('-f', '--foreground', action='store_true', dest='foreground', default=False,
44
 
                      help="run in the foreground; don't daemonize")
45
 
    parser.add_option('-C', '--check', action='store_true', dest='check', default=False,
46
 
                      help="only start if not already running (useful for cron jobs)")
47
 
    parser.add_option('-p', '--pidfile', dest="pidfile", default=default_pidfile,
48
 
                      type=str, help="override pidfile location")
49
 
    parser.add_option('-c', '--config-file', dest="configfile", default=default_configfile,
50
 
                                          type=str, help="override configuration file location")
51
 
    parser.add_option('-L', '--log-folder', dest="log_folder", default=default_log_folder,
52
 
                      type=str, help="override log file directory")
53
 
    options, args = parser.parse_args()
54
 
    if len(args) > 0:
55
 
        parser.error('No filename arguments are used, only options.')
56
 
 
57
 
    if options.check:
58
 
        if daemon.is_running(options.pidfile):
59
 
            sys.exit(0)
60
 
        sys.stderr.write('Did not find loggerhead running in %r; restarting...\n' % (options.pidfile,))
61
 
 
62
 
    # read loggerhead config
63
 
 
64
 
    config = ConfigObj(options.configfile, encoding='utf-8')
65
 
    extra_path = config.get('bzrpath', None)
66
 
    if extra_path:
67
 
        sys.path.insert(0, extra_path)
68
 
 
69
 
    potential_overrides = [ ('server.socket_port', int),
70
 
                            ('server.webpath', str),
71
 
                            ('server.thread_pool', int),
72
 
                            ('server.socket_host' ,str) ]
73
 
    server_port = int(config.get('server.socket_port', 8080))
74
 
    nworkers = int(config.get('server.thread_pool', 10))
75
 
    server_host = config.get('server.socket_host', '0.0.0.0')
76
 
    webpath = config.get('server.webpath', None)
77
 
 
78
 
    for key, keytype in potential_overrides:
79
 
        value = config.get(key, None)
80
 
        if value is not None:
81
 
            value = keytype(value)
82
 
 
83
 
    if not options.foreground:
84
 
        sys.stderr.write('\n')
85
 
        sys.stderr.write('Launching loggerhead into the background.\n')
86
 
        sys.stderr.write('PID file: %s\n' % (options.pidfile,))
87
 
        sys.stderr.write('\n')
88
 
 
89
 
        daemon.daemonize(options.pidfile, home)
90
 
 
91
 
    setup_logging(options.log_folder, config, foreground=options.foreground)
92
 
 
93
 
    log = logging.getLogger('loggerhead')
94
 
    log.info('Starting up...')
95
 
 
96
 
    app = Root(config)
97
 
 
98
 
    app = app
99
 
    app = make_middleware(app)
100
 
    app = make_filter(app, None, logger_name=log.name+'.access')
101
 
    app = ErrorHandlerApp(app)
102
 
 
103
 
    if webpath:
104
 
        scheme, netloc, path, blah, blah, blah = urlparse.urlparse(webpath)
105
 
        def app(environ, start_response, orig=app):
106
 
            environ['SCRIPT_NAME'] = path
107
 
            environ['HTTP_HOST'] = netloc
108
 
            return orig(environ, start_response)
109
 
 
110
 
    try:
111
 
        httpserver.serve(
112
 
            app, host=server_host, port=server_port,
113
 
            threadpool_workers=nworkers)
114
 
    finally:
115
 
        log.info('Shutdown.')
116
 
        try:
117
 
            os.remove(options.pidfile)
118
 
        except OSError:
119
 
            pass
120
 
 
121
 
 
122
 
if __name__ == '__main__':
123
 
    main()