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