← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to start-loggerhead

  • Committer: Michael Hudson
  • Date: 2008-10-01 20:02:14 UTC
  • mfrom: (227.1.1 loggerhead.python2.4)
  • Revision ID: michael.hudson@canonical.com-20081001200214-15go6vse1cklf345
restore python2.4 compatibility (Marius Kruger)

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()