28
28
from paste.translogger import TransLogger
30
30
from loggerhead import __version__
31
from loggerhead.apps.transport import (
32
BranchesFromTransportRoot, UserBranchesFromTransportRoot)
33
from loggerhead.config import LoggerheadConfig
31
from loggerhead.apps.filesystem import (
32
BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
34
33
from loggerhead.util import Reloader
35
34
from loggerhead.apps.error import ErrorHandlerApp
38
def get_config_and_path(args):
39
config = LoggerheadConfig(args)
41
if config.get_option('show_version'):
42
print "loggerhead %s" % (__version__,)
37
def command_line_parser():
38
parser = OptionParser("%prog [options] <path>")
44
parser.add_option("--user-dirs", action="store_true", dest="user_dirs",
45
help="Serve user directories as ~user.")
46
parser.add_option("--trunk-dir", metavar="DIR",
47
help="The directory that contains the trunk branches.")
48
parser.add_option("--port", dest="user_port",
49
help=("Port Loggerhead should listen on "
50
"(defaults to 8080)."))
51
parser.add_option("--host", dest="user_host",
52
help="Host Loggerhead should listen on.")
53
parser.add_option("--prefix", dest="user_prefix",
54
help="Specify host prefix.")
55
parser.add_option("--profile", action="store_true", dest="profile",
56
help="Generate callgrind profile data to "
57
"%d-stats.callgrind on each request.")
58
parser.add_option("--reload", action="store_true", dest="reload",
59
help="Restarts the application when changing python"
60
" files. Only used for development purposes.")
61
parser.add_option('--log-folder', dest="log_folder",
62
type=str, help="The directory to place log files in.")
63
parser.add_option("--version", action="store_true", dest="show_version",
64
help="Print the software version and exit")
69
parser = command_line_parser()
70
(options, args) = parser.parse_args(sys.argv[1:])
72
if options.show_version:
73
print "loggerhead %s" % __version__
45
if config.arg_count > 1:
48
elif config.arg_count == 1:
49
base = config.get_arg(0)
53
if not config.get_option('allow_writes'):
54
base = 'readonly+' + base
59
def setup_logging(config, init_logging=True, log_file=None):
60
log_level = config.get_log_level()
63
if log_level is not None:
64
logging.getLogger('').setLevel(log_level)
65
logger = logging.getLogger('loggerhead')
66
if log_level is not None:
67
logger.setLevel(log_level)
68
if log_file is not None:
69
handler = logging.StreamHandler(log_file)
71
if config.get_option('log_folder'):
72
logfile_path = os.path.join(
73
config.get_option('log_folder'), 'serve-branches.log')
75
logfile_path = 'serve-branches.log'
76
handler = logging.FileHandler(logfile_path, 'a')
77
formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s:'
79
handler.setFormatter(formatter)
80
# We set the handler to accept all messages, the *logger* won't emit them
81
# if it is configured to suppress it
82
handler.setLevel(logging.DEBUG)
83
logger.addHandler(handler)
84
def _restrict_logging(logger_name):
85
logger = logging.getLogger(logger_name)
86
if logger.getEffectiveLevel() < logging.INFO:
87
logger.setLevel(logging.INFO)
88
# simpleTAL is *very* verbose in DEBUG mode, which is otherwise the
89
# default. So quiet it up a bit.
90
_restrict_logging('simpleTAL')
91
_restrict_logging('simpleTALES')
95
def make_app_for_config_and_path(config, base):
96
if config.get_option('trunk_dir') and not config.get_option('user_dirs'):
84
if not os.path.isdir(path):
85
print "%s is not a directory" % path
88
if options.trunk_dir and not options.user_dirs:
97
89
print "--trunk-dir is only valid with --user-dirs"
100
if config.get_option('reload'):
101
93
if Reloader.is_installed():
102
94
Reloader.install()
104
96
return Reloader.restart_with_reloader()
106
if config.get_option('user_dirs'):
107
if not config.get_option('trunk_dir'):
99
if not options.trunk_dir:
108
100
print "You didn't specify a directory for the trunk directories."
110
app = UserBranchesFromTransportRoot(base, config)
112
app = BranchesFromTransportRoot(base, config)
114
setup_logging(config)
116
if config.get_option('profile'):
102
app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
104
app = BranchesFromFileSystemRoot(path)
107
logging.basicConfig()
108
logging.getLogger('').setLevel(logging.DEBUG)
109
logger = getattr(app, 'log', logging.getLogger('loggerhead'))
110
if options.log_folder:
111
logfile_path = os.path.join(options.log_folder, 'serve-branches.log')
113
logfile_path = 'serve-branches.log'
114
logfile = logging.FileHandler(logfile_path, 'a')
115
formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s:'
117
logfile.setFormatter(formatter)
118
logfile.setLevel(logging.DEBUG)
119
logger.addHandler(logfile)
120
# setup_logging() #end
121
app = TransLogger(app, logger=logger)
117
123
from loggerhead.middleware.profile import LSProfMiddleware
118
124
app = LSProfMiddleware(app)
119
if config.get_option('memory_profile'):
120
from dozer import Dozer
123
if not config.get_option('user_prefix'):
126
if not options.user_prefix:
126
prefix = config.get_option('user_prefix')
129
prefix = options.user_prefix
127
130
if not prefix.startswith('/'):
128
131
prefix = '/' + prefix
148
151
app = HTTPExceptionHandler(app)
149
152
app = ErrorHandlerApp(app)
150
app = TransLogger(app, logger=logging.getLogger('loggerhead'))
158
config, path = get_config_and_path(args)
160
app = make_app_for_config_and_path(config, path)
162
if not config.get_option('user_port'):
154
if not options.user_port:
165
port = config.get_option('user_port')
157
port = options.user_port
167
if not config.get_option('user_host'):
159
if not options.user_host:
170
host = config.get_option('user_host')
172
if not config.get_option('protocol'):
175
protocol = config.get_option('protocol')
177
if protocol == 'http':
178
httpserver.serve(app, host=host, port=port)
180
if protocol == 'fcgi':
181
from flup.server.fcgi import WSGIServer
182
elif protocol == 'scgi':
183
from flup.server.scgi import WSGIServer
184
elif protocol == 'ajp':
185
from flup.server.ajp import WSGIServer
187
print 'Unknown protocol: %s.' % (protocol)
189
WSGIServer(app, bindAddress=(host, int(port))).run()
162
host = options.user_host
166
httpserver.serve(app, host=host, port=port)
169
if __name__ == "__main__":