~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to serve-branches

  • Committer: Guillermo Gonzalez
  • Date: 2008-09-10 00:02:06 UTC
  • mto: This revision was merged to the branch mainline in revision 221.
  • Revision ID: guillo.gonzo@gmail.com-20080910000206-kt35kgrzu8y4472v
 * update NEWS and serve-branches man page 
 * refactored all reload logic into loggerhead.util.Reloader class

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
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
"""Search for branches underneath a directory and serve them all."""
 
17
 
 
18
import logging
 
19
import os
 
20
import sys
 
21
 
 
22
from optparse import OptionParser
 
23
 
 
24
from paste import httpserver
 
25
from paste.httpexceptions import HTTPExceptionHandler
 
26
from paste.translogger import TransLogger
 
27
 
 
28
from loggerhead import __version__
 
29
from loggerhead.apps.filesystem import (
 
30
    BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
 
31
from loggerhead.util import Reloader
 
32
 
 
33
 
 
34
def command_line_parser():
 
35
    parser = OptionParser("%prog [options] <path>")
 
36
    parser.set_defaults(
 
37
        user_dirs=False,
 
38
        show_version=False,
 
39
        )
 
40
    parser.add_option("--user-dirs", action="store_true", dest="user_dirs",
 
41
                      help="Serve user directories as ~user.")
 
42
    parser.add_option("--trunk-dir", metavar="DIR",
 
43
                      help="The directory that contains the trunk branches.")
 
44
    parser.add_option("--port", dest="user_port",
 
45
                      help="Port Loggerhead should listen on (defaults to 8080).")
 
46
    parser.add_option("--host", dest="user_host",
 
47
                      help="Host Loggerhead should listen on.")
 
48
    parser.add_option("--prefix", dest="user_prefix",
 
49
                      help="Specify host prefix.")
 
50
    parser.add_option("--reload", action="store_true", dest="reload",
 
51
                      help="Restarts the application when changing python"
 
52
                           " files. Only used for development purposes.")
 
53
    parser.add_option("--version", action="store_true", dest="show_version",
 
54
                      help="Print the software version and exit")
 
55
    return parser
 
56
 
 
57
 
 
58
def main(args):
 
59
    logging.basicConfig()
 
60
    logging.getLogger().setLevel(logging.DEBUG)
 
61
 
 
62
    parser = command_line_parser()
 
63
    (options, args) = parser.parse_args(sys.argv[1:])
 
64
 
 
65
    if options.show_version:
 
66
        print "loggerhead %s" % __version__
 
67
        sys.exit(0)
 
68
 
 
69
    if len(args) > 1:
 
70
        parser.print_help()
 
71
        sys.exit(1)
 
72
    elif len(args) == 1:
 
73
        [path] = args
 
74
    else:
 
75
        path = '.'
 
76
 
 
77
    if not os.path.isdir(path):
 
78
        print "%s is not a directory" % path
 
79
        sys.exit(1)
 
80
 
 
81
    if options.trunk_dir and not options.user_dirs:
 
82
        print "--trunk-dir is only valid with --user-dirs"
 
83
        sys.exit(1)
 
84
        
 
85
    if options.reload:
 
86
        if Reloader.is_installed():
 
87
            Reloader.install()
 
88
        else:
 
89
            return Reloader.restart_with_reloader()
 
90
 
 
91
    if options.user_dirs:
 
92
        if not options.trunk_dir:
 
93
            print "You didn't specify a directory for the trunk directories."
 
94
            sys.exit(1)
 
95
        app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
 
96
    else:
 
97
        app = BranchesFromFileSystemRoot(path)
 
98
    app = HTTPExceptionHandler(app)
 
99
    app = TransLogger(app)
 
100
 
 
101
    if not options.user_prefix:
 
102
        prefix = '/'
 
103
    else:
 
104
        prefix = options.user_prefix
 
105
 
 
106
    try:
 
107
        from paste.deploy.config import PrefixMiddleware
 
108
    except ImportError:
 
109
        pass
 
110
    else:
 
111
        app = PrefixMiddleware(app, prefix=prefix)
 
112
    
 
113
    if not options.user_port:
 
114
        port = '8080'
 
115
    else:
 
116
        port = options.user_port
 
117
 
 
118
    if not options.user_host:
 
119
        host = '0.0.0.0'
 
120
    else:
 
121
        host = options.user_host
 
122
 
 
123
    httpserver.serve(app, host=host, port=port)
 
124
 
 
125
 
 
126
if __name__ == "__main__":
 
127
    main(sys.argv)