← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to loggerhead/main.py

  • Committer: Jelmer Vernooij
  • Date: 2010-01-11 02:54:48 UTC
  • mfrom: (398 trunk)
  • mto: This revision was merged to the branch mainline in revision 399.
  • Revision ID: jelmer@samba.org-20100111025448-lokdg33w2awf9f8n
Merge trunk.

Show diffs side-by-side

added added

removed removed

Lines of Context:
loggerhead/main.py
1
 
#!/usr/bin/env python
 
1
#
 
2
# Copyright (C) 2008, 2009 Canonical Ltd
 
3
#
2
4
# This program is free software; you can redistribute it and/or modify
3
5
# it under the terms of the GNU General Public License as published by
4
6
# the Free Software Foundation; either version 2 of the License, or
20
22
import sys
21
23
 
22
24
from bzrlib.plugin import load_plugins
23
 
from bzrlib.transport import get_transport
24
25
 
25
26
from paste import httpserver
26
27
from paste.httpexceptions import HTTPExceptionHandler, HTTPInternalServerError
34
35
from loggerhead.apps.error import ErrorHandlerApp
35
36
 
36
37
 
37
 
def main(args):
38
 
    config = LoggerheadConfig()
 
38
def get_config_and_path(args):
 
39
    config = LoggerheadConfig(args)
39
40
 
40
41
    if config.get_option('show_version'):
41
 
        print "loggerhead %s" % __version__
 
42
        print "loggerhead %s" % (__version__,)
42
43
        sys.exit(0)
43
44
 
44
45
    if config.arg_count > 1:
45
46
        config.print_help()
46
47
        sys.exit(1)
47
48
    elif config.arg_count == 1:
48
 
        path = config.get_arg(0)
49
 
    else:
50
 
        path = '.'
51
 
 
52
 
    load_plugins()
53
 
 
54
 
    transport = get_transport(path)
55
 
 
56
 
    if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
57
 
        print "--trunk-dir is only valid with --user-dirs"
58
 
        sys.exit(1)
59
 
 
60
 
    if config.get_option('reload'):
61
 
        if Reloader.is_installed():
62
 
            Reloader.install()
63
 
        else:
64
 
            return Reloader.restart_with_reloader()
65
 
 
66
 
    if config.get_option('user_dirs'):
67
 
        if not config.get_option('trunk_dir'):
68
 
            print "You didn't specify a directory for the trunk directories."
69
 
            sys.exit(1)
70
 
        app = UserBranchesFromTransportRoot(transport, config)
71
 
    else:
72
 
        app = BranchesFromTransportRoot(transport, config)
73
 
 
74
 
    # setup_logging()
 
49
        base = config.get_arg(0)
 
50
    else:
 
51
        base = '.'
 
52
 
 
53
    if not config.get_option('allow_writes'):
 
54
        base = 'readonly+' + base
 
55
 
 
56
    return config, base
 
57
 
 
58
 
 
59
def setup_logging(config):
75
60
    logging.basicConfig()
76
61
    logging.getLogger('').setLevel(logging.DEBUG)
77
 
    logger = getattr(app, 'log', logging.getLogger('loggerhead'))
 
62
    logger = logging.getLogger('loggerhead')
78
63
    if config.get_option('log_folder'):
79
64
        logfile_path = os.path.join(
80
65
            config.get_option('log_folder'), 'serve-branches.log')
86
71
    logfile.setFormatter(formatter)
87
72
    logfile.setLevel(logging.DEBUG)
88
73
    logger.addHandler(logfile)
89
 
 
90
 
    # setup_logging() #end
91
 
 
92
 
    app = TransLogger(app, logger=logger)
 
74
    return logger
 
75
 
 
76
 
 
77
def make_app_for_config_and_path(config, base):
 
78
    if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
 
79
        print "--trunk-dir is only valid with --user-dirs"
 
80
        sys.exit(1)
 
81
 
 
82
    if config.get_option('reload'):
 
83
        if Reloader.is_installed():
 
84
            Reloader.install()
 
85
        else:
 
86
            return Reloader.restart_with_reloader()
 
87
 
 
88
    if config.get_option('user_dirs'):
 
89
        if not config.get_option('trunk_dir'):
 
90
            print "You didn't specify a directory for the trunk directories."
 
91
            sys.exit(1)
 
92
        app = UserBranchesFromTransportRoot(base, config)
 
93
    else:
 
94
        app = BranchesFromTransportRoot(base, config)
 
95
 
 
96
    setup_logging(config)
 
97
 
93
98
    if config.get_option('profile'):
94
99
        from loggerhead.middleware.profile import LSProfMiddleware
95
100
        app = LSProfMiddleware(app)
124
129
 
125
130
    app = HTTPExceptionHandler(app)
126
131
    app = ErrorHandlerApp(app)
 
132
    app = TransLogger(app, logger=logging.getLogger('loggerhead'))
 
133
 
 
134
    return app
 
135
 
 
136
 
 
137
def main(args):
 
138
    load_plugins()
 
139
 
 
140
    config, path = get_config_and_path(args)
 
141
 
 
142
    app = make_app_for_config_and_path(config, path)
127
143
 
128
144
    if not config.get_option('user_port'):
129
145
        port = '8080'
135
151
    else:
136
152
        host = config.get_option('user_host')
137
153
 
138
 
    httpserver.serve(app, host=host, port=port)
139
 
 
140
 
 
141
 
if __name__ == "__main__":
142
 
    main(sys.argv)
 
154
    if not config.get_option('protocol'):
 
155
        protocol = 'http'
 
156
    else:
 
157
        protocol = config.get_option('protocol')
 
158
 
 
159
    if protocol == 'http':
 
160
        httpserver.serve(app, host=host, port=port)
 
161
    else:
 
162
        if protocol == 'fcgi':
 
163
            from flup.server.fcgi import WSGIServer
 
164
        elif protocol == 'scgi':
 
165
            from flup.server.scgi import WSGIServer
 
166
        elif protocol == 'ajp':
 
167
            from flup.server.ajp import WSGIServer
 
168
        else:
 
169
            print 'Unknown protocol: %s.' % (protocol)
 
170
            sys.exit(1)
 
171
        WSGIServer(app, bindAddress=(host, int(port))).run()