def do_download(self):
self.log.debug("downloading %s" % self.args.container)
cli, meta = self._setup_client()
if cli is None:
return 1
elif 'client' in meta:
self.log.error("%s is locked, downloading a container in use is unreliable" % self.args.container)
return 1
object_size = int(meta['object-size'])
objects = int(meta['objects'])
store = SwiftStorage(self.authurl,
self.username,
self.password,
self.args.container,
object_size,
objects,
)
try:
store.lock("ctl-download")
except StorageError as ex:
self.log.error(ex)
return 1
size = 0
fdo = None
try:
fdo = open(self.args.image, "w")
while True:
data = store.read(object_size)
if data == '':
break
fdo.write(data)
size += len(data)
if not self.args.quiet:
sys.stdout.write("\rDownloading %s [%.2d%%]" % (self.args.container, 100*size/(objects*object_size)))
sys.stdout.flush()
except IOError as ex:
self.log.error(ex)
return 1
except KeyboardInterrupt:
self.log.warning("user interrupt")
return 1
finally:
if fdo:
fdo.close()
try:
store.unlock()
except StorageError as ex:
self.log.warning("Failed to unlock %s: %s" % (self.args.container, ex))
if not self.args.quiet:
sys.stdout.write("\r")
sys.stdout.flush()
self.log.info("Done, %s bytes written" % size)
return 0
def do_download(self):
self.log.debug("downloading %s" % self.args.container)
cli, meta = self._setup_client()
if cli is None:
return 1
elif 'client' in meta:
self.log.error(
"%s is locked, downloading a container in use is unreliable" %
self.args.container)
return 1
object_size = int(meta['object-size'])
objects = int(meta['objects'])
store = SwiftStorage(
self.auth,
self.args.container,
object_size,
objects,
)
try:
store.lock("ctl-download")
except StorageError as ex:
self.log.error(ex)
return 1
size = 0
fdo = None
try:
fdo = open(self.args.image, "wb")
while True:
data = store.read(object_size)
if data == b'':
break
fdo.write(data)
size += len(data)
if not self.args.quiet:
sys.stdout.write("\rDownloading %s [%.2d%%]" %
(self.args.container, 100 * size /
(objects * object_size)))
sys.stdout.flush()
except IOError as ex:
self.log.error(ex)
return 1
except KeyboardInterrupt:
self.log.warning("user interrupt")
return 1
finally:
if fdo:
fdo.close()
try:
store.unlock()
except StorageError as ex:
self.log.warning("Failed to unlock %s: %s" %
(self.args.container, ex))
if not self.args.quiet:
sys.stdout.write("\r")
sys.stdout.flush()
self.log.info("Done, %s bytes written" % size)
return 0
