From: Daira Hopwood Date: Sat, 18 May 2013 01:31:19 +0000 (+0100) Subject: Move failure handling for create_container to make it more testable. refs #1971 X-Git-Url: https://git.rkrishnan.org/%5B/%5D%20/uri//%22%22.?a=commitdiff_plain;h=c1fc44c6c812a74c5ad3616053ec061f14ef9d71;p=tahoe-lafs%2Ftahoe-lafs.git Move failure handling for create_container to make it more testable. refs #1971 Signed-off-by: Daira Hopwood --- diff --git a/src/allmydata/scripts/admin.py b/src/allmydata/scripts/admin.py index 1f772937..e89455e2 100644 --- a/src/allmydata/scripts/admin.py +++ b/src/allmydata/scripts/admin.py @@ -64,35 +64,40 @@ for more details. def create_container(options): from twisted.internet import reactor, defer - err = options.stderr - d = defer.maybeDeferred(do_create_container, options) - def _failed(f): - print >>err, "Container creation failed." - print >>err, "%s: %s" % (f.value.__class__.__name__, f.value) - print >>err - return f - d.addErrback(_failed) d.addCallbacks(lambda ign: os._exit(0), lambda ign: os._exit(1)) reactor.run() def do_create_container(options): + from twisted.internet import defer from allmydata.node import ConfigOnly from allmydata.client import Client - out = options.stderr - config = ConfigOnly(options['basedir']) - (backend, _) = Client.configure_backend(config) - - d = backend.create_container() - def _done(res): - if res is False: - print >>out, ("It is not necessary to create a container for this backend type (%s)." - % (backend.__class__.__name__,)) - else: - print >>out, "The container was successfully created." - print >>out - d.addCallback(_done) + out = options.stdout + err = options.stderr + + d = defer.succeed(None) + def _do_create(ign): + config = ConfigOnly(options['basedir']) + (backend, _) = Client.configure_backend(config) + + d2 = backend.create_container() + def _done(res): + if res is False: + print >>out, ("It is not necessary to create a container for this backend type (%s)." + % (backend.__class__.__name__,)) + else: + print >>out, "The container was successfully created." + print >>out + d2.addCallback(_done) + return d2 + d.addCallback(_do_create) + def _failed(f): + print >>err, "Container creation failed." + print >>err, "%s: %s" % (f.value.__class__.__name__, f.value) + print >>err + return f + d.addErrback(_failed) return d