~launchpad-pqm/launchpad/devel

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
#!/usr/bin/python2.4 -u
# pylint: disable-msg=W0403

import _pythonpath

import sys
import xmlrpclib

from canonical.codehosting.branchfsclient import BlockingProxy
from canonical.codehosting.rewrite import BranchRewriter
from canonical.config import config
from canonical.launchpad.scripts.base import LaunchpadScript

# XXX MichaelHudson, bug=309240: This script currently logs to stderr, which
# will end up in Apache's error.log.  It should instead log to a file
# specified in the config, and unhandled exceptions should log an oops.

class BranchRewriteScript(LaunchpadScript):

    def __init__(self, name):
        LaunchpadScript.__init__(self, name)
        proxy = xmlrpclib.ServerProxy(config.codehosting.branchfs_endpoint)
        self.rewriter = BranchRewriter(self.logger, BlockingProxy(proxy))

    def run(self, use_web_security=False, implicit_begin=True,
            isolation=None):
        """See `LaunchpadScript.run`.

        As this script does not need the component architecture or a
        connection to the database, we override this method to avoid setting
        them up.
        """
        self.main()

    def main(self):
        self.logger.debug("Starting up...")
        while True:
            try:
                line = sys.stdin.readline().strip()
                print self.rewriter.rewriteLine(line)
            except KeyboardInterrupt:
                sys.exit()
            except:
                self.logger.exception('Oops.')
                print "NULL"


if __name__ == '__main__':
    BranchRewriteScript("branch-rewrite").run()