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('--memory-profile', action='store_true',
54
dest='memory_profile',
55
help='Profile the memory usage using heapy.')
56
parser.add_option("--prefix", dest="user_prefix",
57
help="Specify host prefix.")
58
parser.add_option("--profile", action="store_true", dest="profile",
59
help="Generate callgrind profile data to "
60
"%d-stats.callgrind on each request.")
61
parser.add_option("--reload", action="store_true", dest="reload",
62
help="Restarts the application when changing python"
63
" files. Only used for development purposes.")
64
parser.add_option('--log-folder', dest="log_folder",
65
type=str, help="The directory to place log files in.")
66
parser.add_option("--version", action="store_true", dest="show_version",
67
help="Print the software version and exit")
72
parser = command_line_parser()
73
(options, args) = parser.parse_args(sys.argv[1:])
75
if options.show_version:
76
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):
87
if not os.path.isdir(path):
88
print "%s is not a directory" % path
91
if options.trunk_dir and not options.user_dirs:
92
print "--trunk-dir is only valid with --user-dirs"
96
if Reloader.is_installed():
99
return Reloader.restart_with_reloader()
101
if options.user_dirs:
102
if not options.trunk_dir:
103
print "You didn't specify a directory for the trunk directories."
105
app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
107
app = BranchesFromFileSystemRoot(path)
60
110
logging.basicConfig()
61
111
logging.getLogger('').setLevel(logging.DEBUG)
62
logger = logging.getLogger('loggerhead')
63
if config.get_option('log_folder'):
64
logfile_path = os.path.join(
65
config.get_option('log_folder'), 'serve-branches.log')
112
logger = getattr(app, 'log', logging.getLogger('loggerhead'))
113
if options.log_folder:
114
logfile_path = os.path.join(options.log_folder, 'serve-branches.log')
67
116
logfile_path = 'serve-branches.log'
68
117
logfile = logging.FileHandler(logfile_path, 'a')
71
120
logfile.setFormatter(formatter)
72
121
logfile.setLevel(logging.DEBUG)
73
122
logger.addHandler(logfile)
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"
82
if config.get_option('reload'):
83
if Reloader.is_installed():
86
return Reloader.restart_with_reloader()
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."
92
app = UserBranchesFromTransportRoot(base, config)
94
app = BranchesFromTransportRoot(base, config)
98
if config.get_option('profile'):
124
memprofile = logging.getLogger('loggerhead-memprofile')
125
memprofile.setLevel(logging.DEBUG)
126
memprofile.addHandler(logging.FileHandler('loggerhead-memprofile'))
128
# setup_logging() #end
130
app = TransLogger(app, logger=logger)
99
132
from loggerhead.middleware.profile import LSProfMiddleware
100
133
app = LSProfMiddleware(app)
101
if config.get_option('memory_profile'):
102
from dozer import Dozer
134
if options.memory_profile:
135
from loggerhead.middleware.profile import MemoryProfileMiddleware
136
app = MemoryProfileMiddleware(app)
105
if not config.get_option('user_prefix'):
138
if not options.user_prefix:
108
prefix = config.get_option('user_prefix')
141
prefix = options.user_prefix
109
142
if not prefix.startswith('/'):
110
143
prefix = '/' + prefix
130
163
app = HTTPExceptionHandler(app)
131
164
app = ErrorHandlerApp(app)
132
app = TransLogger(app, logger=logging.getLogger('loggerhead'))
140
config, path = get_config_and_path(args)
142
app = make_app_for_config_and_path(config, path)
144
if not config.get_option('user_port'):
166
if not options.user_port:
147
port = config.get_option('user_port')
169
port = options.user_port
149
if not config.get_option('user_host'):
171
if not options.user_host:
152
host = config.get_option('user_host')
154
if not config.get_option('protocol'):
157
protocol = config.get_option('protocol')
159
if protocol == 'http':
160
httpserver.serve(app, host=host, port=port)
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
169
print 'Unknown protocol: %s.' % (protocol)
171
WSGIServer(app, bindAddress=(host, int(port))).run()
174
host = options.user_host
178
httpserver.serve(app, host=host, port=port)
181
if __name__ == "__main__":
added
removed
serve-branches
