1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
#!/usr/bin/python -uS
#
# Copyright 2009-2011 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
# pylint: disable-msg=W0403
"""Script intended to run as a :prg: RewriteMap.
See http://httpd.apache.org/docs/2.2/mod/mod_rewrite.html#rewritemap for the
documentation of the very simple 'protocol' Apache uses to talk to us, and
lp.codehosting.rewrite.BranchRewriter for the logic of the rewritemap.
"""
import os
import sys
import _pythonpath
from lp.services.config import config
from lp.services.database.lpstorm import ISlaveStore
from lp.code.model.branch import Branch
from lp.codehosting.rewrite import BranchRewriter
from lp.services.log.loglevels import (
INFO,
WARNING,
)
from lp.services.scripts.base import LaunchpadScript
class BranchRewriteScript(LaunchpadScript):
# By default, only emit WARNING and above messages to stderr, which
# will end up in the Apache error log.
loglevel = WARNING
def add_my_options(self):
"""Make the logging go to a file by default.
Because this script is run by Apache, logging to stderr results in our
log output ending up in Apache's error.log, which is not so useful.
We hack the OptionParser to set the default (which will be applied;
Apache doesn't pass any arguments to the script it starts up) to a
value from the config.
"""
log_file_location = config.codehosting.rewrite_script_log_file
log_file_directory = os.path.dirname(log_file_location)
if not os.path.isdir(log_file_directory):
os.makedirs(log_file_directory)
self.parser.defaults['log_file'] = log_file_location
self.parser.defaults['log_file_level'] = INFO
def main(self):
rewriter = BranchRewriter(self.logger)
self.logger.debug("Starting up...")
while True:
try:
line = sys.stdin.readline()
# Mod-rewrite always gives us a newline terminated string.
if line:
print rewriter.rewriteLine(line.strip())
else:
# Standard input has been closed, so die.
return
except KeyboardInterrupt:
sys.exit()
except Exception:
self.logger.exception('Exception occurred:')
print "NULL"
# The exception might have been a DisconnectionError or
# similar. Cleanup such as database reconnection will
# not happen until the transaction is rolled back.
# XXX StuartBishop 2011-08-31 bug=819282: We are
# explicitly rolling back the store here as a workaround
# instead of using transaction.abort()
try:
ISlaveStore(Branch).rollback()
except Exception:
self.logger.exception('Exception occurred in rollback:')
if __name__ == '__main__':
BranchRewriteScript("branch-rewrite", dbuser='branch-rewrite').run(
isolation='autocommit')
|