← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to serve-branches

  • Committer: Robert Collins
  • Date: 2008-12-05 05:07:33 UTC
  • Revision ID: robertc@robertcollins.net-20081205050733-x0op0z69lg067wbw
Load all bzrlib plugins.

Show diffs side-by-side

added added

removed removed

Lines of Context:
serve-branches
28
28
from loggerhead import __version__
29
29
from loggerhead.apps.filesystem import (
30
30
    BranchesFromFileSystemRoot, UserBranchesFromFileSystemRoot)
 
31
from loggerhead.util import Reloader
 
32
from loggerhead.apps.error import ErrorHandlerApp
31
33
 
32
34
 
33
35
def command_line_parser():
35
37
    parser.set_defaults(
36
38
        user_dirs=False,
37
39
        show_version=False,
 
40
        log_folder=None,
38
41
        )
39
42
    parser.add_option("--user-dirs", action="store_true", dest="user_dirs",
40
43
                      help="Serve user directories as ~user.")
41
44
    parser.add_option("--trunk-dir", metavar="DIR",
42
45
                      help="The directory that contains the trunk branches.")
43
46
    parser.add_option("--port", dest="user_port",
44
 
                      help="Port Loggerhead should listen on. 8080 is the default one.")
 
47
                      help=("Port Loggerhead should listen on "
 
48
                            "(defaults to 8080)."))
45
49
    parser.add_option("--host", dest="user_host",
46
50
                      help="Host Loggerhead should listen on.")
47
51
    parser.add_option("--prefix", dest="user_prefix",
48
52
                      help="Specify host prefix.")
 
53
    parser.add_option("--profile", action="store_true", dest="profile",
 
54
                      help="Generate callgrind profile data to "
 
55
                        "%d-stats.callgrind on each request.")
 
56
    parser.add_option("--reload", action="store_true", dest="reload",
 
57
                      help="Restarts the application when changing python"
 
58
                           " files. Only used for development purposes.")
 
59
    parser.add_option('--log-folder', dest="log_folder",
 
60
                      type=str, help="The directory to place log files in.")
49
61
    parser.add_option("--version", action="store_true", dest="show_version",
50
62
                      help="Print the software version and exit")
51
63
    return parser
52
64
 
53
65
 
54
66
def main(args):
55
 
    logging.basicConfig()
56
 
    logging.getLogger().setLevel(logging.DEBUG)
57
 
 
58
67
    parser = command_line_parser()
59
68
    (options, args) = parser.parse_args(sys.argv[1:])
60
69
 
78
87
        print "--trunk-dir is only valid with --user-dirs"
79
88
        sys.exit(1)
80
89
 
 
90
    if options.reload:
 
91
        if Reloader.is_installed():
 
92
            Reloader.install()
 
93
        else:
 
94
            return Reloader.restart_with_reloader()
 
95
 
81
96
    if options.user_dirs:
82
97
        if not options.trunk_dir:
83
98
            print "You didn't specify a directory for the trunk directories."
85
100
        app = UserBranchesFromFileSystemRoot(path, options.trunk_dir)
86
101
    else:
87
102
        app = BranchesFromFileSystemRoot(path)
 
103
 
 
104
    # setup_logging()
 
105
    logging.basicConfig()
 
106
    logging.getLogger('').setLevel(logging.DEBUG)
 
107
    logger = getattr(app, 'log', logging.getLogger('loggerhead'))
 
108
    if options.log_folder:
 
109
        logfile_path = os.path.join(options.log_folder, 'serve-branches.log')
 
110
    else:
 
111
        logfile_path = 'serve-branches.log'
 
112
    logfile = logging.FileHandler(logfile_path, 'a')
 
113
    formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s:'
 
114
                                  ' %(message)s')
 
115
    logfile.setFormatter(formatter)
 
116
    logfile.setLevel(logging.DEBUG)
 
117
    logger.addHandler(logfile)
 
118
    # setup_logging() #end
 
119
    app = ErrorHandlerApp(app)
88
120
    app = HTTPExceptionHandler(app)
89
 
    app = TransLogger(app)
 
121
    app = TransLogger(app, logger=logger)
 
122
    if options.profile:
 
123
        from loggerhead.middleware.profile import LSProfMiddleware
 
124
        app = LSProfMiddleware(app)
90
125
 
91
126
    if not options.user_prefix:
92
127
        prefix = '/'
99
134
        pass
100
135
    else:
101
136
        app = PrefixMiddleware(app, prefix=prefix)
102
 
    
 
137
 
103
138
    if not options.user_port:
104
139
        port = '8080'
105
140
    else: