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
|
#!/usr/bin/python2.4
# Copyright 2006-2007 Canonical Ltd. All rights reserved.
# pylint: disable-msg=C0103,W0403
import _pythonpath
from optparse import OptionParser
from twisted.internet import defer, reactor
from twisted.python import log as tplog
from twisted.web.xmlrpc import Proxy
from canonical.codehosting.puller import mirror, scheduler
from canonical.config import config
from canonical.launchpad.interfaces import BranchType
from canonical.launchpad.scripts import logger_options
from canonical.twistedsupport.loggingsupport import set_up_logging_for_script
def clean_shutdown(ignored):
reactor.stop()
def shutdown_with_errors(failure):
tplog.err(failure)
failure.printTraceback()
reactor.stop()
def run_mirror(log, manager):
# It's conceivable that mirror() might raise an exception before it
# returns a Deferred -- maybeDeferred means we don't have to worry.
deferred = defer.maybeDeferred(mirror, log, manager)
deferred.addCallback(clean_shutdown)
deferred.addErrback(shutdown_with_errors)
if __name__ == '__main__':
parser = OptionParser()
logger_options(parser)
(options, arguments) = parser.parse_args()
which = arguments.pop(0)
if arguments:
parser.error("Unhandled arguments %s" % repr(arguments))
branch_type_map = {
'upload': BranchType.HOSTED,
'mirror': BranchType.MIRRORED,
'import': BranchType.IMPORTED
}
try:
branch_type = branch_type_map[which]
except KeyError:
parser.error(
'Expected one of %s, but got: %r'
% (branch_type_map.keys(), which))
log = set_up_logging_for_script(options, 'supermirror_%s_puller' % which)
manager = scheduler.JobScheduler(
Proxy(config.codehosting.branch_puller_endpoint), log, branch_type)
reactor.callWhenRunning(run_mirror, log, manager)
reactor.run()
|