← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to serve-branches

  • Committer: Matt Nordhoff
  • Date: 2009-04-17 01:50:05 UTC
  • Revision ID: mnordhoff@mattnordhoff.com-20090417015005-56qyf7o28q8k480u
Don't use the "sets" module.

1.) The code wasn't using the built-in "set" type when it was available.

2.) bzrlib isn't compatible with Python <2.4 anyway, so there's no need to fall back to the "sets" module.

Show diffs side-by-side

added added

removed removed

Lines of Context:
serve-branches
16
16
"""Search for branches underneath a directory and serve them all."""
17
17
 
18
18
import logging
 
19
import os
19
20
import sys
20
21
 
 
22
from bzrlib.plugin import load_plugins
 
23
 
21
24
from paste import httpserver
22
 
from paste.httpexceptions import HTTPExceptionHandler
 
25
from paste.httpexceptions import HTTPExceptionHandler, HTTPInternalServerError
23
26
from paste.translogger import TransLogger
24
27
 
25
 
from loggerhead.apps.filesystem import BranchesFromFileSystemRoot
26
 
 
27
 
 
28
 
logging.basicConfig()
29
 
logging.getLogger().setLevel(logging.DEBUG)
30
 
 
31
 
if len(sys.argv) > 1:
32
 
    path = sys.argv[1]
33
 
else:
34
 
    path = '.'
35
 
 
36
 
app = BranchesFromFileSystemRoot(path)
37
 
 
38
 
app = HTTPExceptionHandler(app)
39
 
app = TransLogger(app)
40
 
 
41
 
try:
42
 
    from paste.deploy.config import PrefixMiddleware
43
 
except ImportError:
44
 
    pass
45
 
else:
46
 
    app = PrefixMiddleware(app)
47
 
 
48
 
#from paste.evalexception import EvalException
49
 
#app = EvalException(app)
50
 
 
51
 
httpserver.serve(app, host='0.0.0.0', port='8080')
 
28
from loggerhead import __version__
 
29
from loggerhead.apps.filesystem import (
 
30
    BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
 
31
from loggerhead.config import LoggerheadConfig
 
32
from loggerhead.util import Reloader
 
33
from loggerhead.apps.error import ErrorHandlerApp
 
34
 
 
35
 
 
36
def main(args):
 
37
    config = LoggerheadConfig()
 
38
 
 
39
    if config.get_option('show_version'):
 
40
        print "loggerhead %s" % __version__
 
41
        sys.exit(0)
 
42
 
 
43
    if config.arg_count > 1:
 
44
        config.print_help()
 
45
        sys.exit(1)
 
46
    elif config.arg_count == 1:
 
47
        path = config.get_arg(0)
 
48
    else:
 
49
        path = '.'
 
50
 
 
51
    if not os.path.isdir(path):
 
52
        print "%s is not a directory" % path
 
53
        sys.exit(1)
 
54
 
 
55
    if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
 
56
        print "--trunk-dir is only valid with --user-dirs"
 
57
        sys.exit(1)
 
58
 
 
59
    if config.get_option('reload'):
 
60
        if Reloader.is_installed():
 
61
            Reloader.install()
 
62
        else:
 
63
            return Reloader.restart_with_reloader()
 
64
 
 
65
    if config.get_option('user_dirs'):
 
66
        if not config.get_option('trunk_dir'):
 
67
            print "You didn't specify a directory for the trunk directories."
 
68
            sys.exit(1)
 
69
        app = UserBranchesFromFileSystemRoot(
 
70
            path, config.get_option('trunk_dir'))
 
71
    else:
 
72
        app = BranchesFromFileSystemRoot(path)
 
73
 
 
74
    # setup_logging()
 
75
    logging.basicConfig()
 
76
    logging.getLogger('').setLevel(logging.DEBUG)
 
77
    logger = getattr(app, 'log', logging.getLogger('loggerhead'))
 
78
    if config.get_option('log_folder'):
 
79
        logfile_path = os.path.join(
 
80
            config.get_option('log_folder'), 'serve-branches.log')
 
81
    else:
 
82
        logfile_path = 'serve-branches.log'
 
83
    logfile = logging.FileHandler(logfile_path, 'a')
 
84
    formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s:'
 
85
                                  ' %(message)s')
 
86
    logfile.setFormatter(formatter)
 
87
    logfile.setLevel(logging.DEBUG)
 
88
    logger.addHandler(logfile)
 
89
 
 
90
    if config.get_option('memory_profile'):
 
91
        memprofile = logging.getLogger('loggerhead-memprofile')
 
92
        memprofile.setLevel(logging.DEBUG)
 
93
        memprofile.addHandler(logging.FileHandler('loggerhead-memprofile'))
 
94
 
 
95
    # setup_logging() #end
 
96
 
 
97
    app = TransLogger(app, logger=logger)
 
98
    if config.get_option('profile'):
 
99
        from loggerhead.middleware.profile import LSProfMiddleware
 
100
        app = LSProfMiddleware(app)
 
101
    if config.get_option('memory_profile'):
 
102
        from loggerhead.middleware.profile import MemoryProfileMiddleware
 
103
        app = MemoryProfileMiddleware(app)
 
104
 
 
105
    if not config.get_option('user_prefix'):
 
106
        prefix = '/'
 
107
    else:
 
108
        prefix = config.get_option('user_prefix')
 
109
        if not prefix.startswith('/'):
 
110
            prefix = '/' + prefix
 
111
 
 
112
    try:
 
113
        from paste.deploy.config import PrefixMiddleware
 
114
    except ImportError:
 
115
        cant_proxy_correctly_message = (
 
116
            'Unsupported configuration: PasteDeploy not available, but '
 
117
            'loggerhead appears to be behind a proxy.')
 
118
        def check_not_proxied(app):
 
119
            def wrapped(environ, start_response):
 
120
                if 'HTTP_X_FORWARDED_SERVER' in environ:
 
121
                    exc = HTTPInternalServerError()
 
122
                    exc.explanation = cant_proxy_correctly_message
 
123
                    raise exc
 
124
                return app(environ, start_response)
 
125
            return wrapped
 
126
        app = check_not_proxied(app)
 
127
    else:
 
128
        app = PrefixMiddleware(app, prefix=prefix)
 
129
 
 
130
    app = HTTPExceptionHandler(app)
 
131
    app = ErrorHandlerApp(app)
 
132
 
 
133
    if not config.get_option('user_port'):
 
134
        port = '8080'
 
135
    else:
 
136
        port = config.get_option('user_port')
 
137
 
 
138
    if not config.get_option('user_host'):
 
139
        host = '0.0.0.0'
 
140
    else:
 
141
        host = config.get_option('user_host')
 
142
 
 
143
    load_plugins()
 
144
 
 
145
    httpserver.serve(app, host=host, port=port)
 
146
 
 
147
 
 
148
if __name__ == "__main__":
 
149
    main(sys.argv)