def _download_parts(self, manifests, directory): bucket = self.args.get('bucket') for manifest in manifests: parts = get_manifest_parts(os.path.join(directory, manifest)) download_files(bucket, parts, directory, service=self.service, config=self.config, show_progress=self.args.get('show_progress', True))
def _delete_all_bundles(self): bucket = self.args.get('bucket') directory = tempfile.mkdtemp() try: manifest_keys = get_manifest_keys(bucket, service=self.service, config=self.config) download_files(bucket, manifest_keys, directory, service=self.service, config=self.config) self._delete_manifest_parts(manifest_keys, directory) self._delete_manifest_keys(manifest_keys) finally: shutil.rmtree(directory)
def _delete_by_prefix(self): bucket = self.args.get('bucket') directory = tempfile.mkdtemp() try: manifest_keys = ["{0}.manifest.xml".format( self.args.get('prefix'))] try: download_files(bucket, manifest_keys, directory, service=self.service, config=self.config) except AWSError as err: if err.code == 'NoSuchEntity': error = ("manifest file '{0}' does not exist in bucket " "'{1}'.") raise ArgumentError(error.format(manifest_keys[0], bucket)) else: raise self._delete_manifest_parts(manifest_keys, directory) self._delete_manifest_keys(manifest_keys) finally: shutil.rmtree(directory)
def _delete_by_prefix(self): bucket = self.args.get('bucket') directory = tempfile.mkdtemp() try: manifest_keys = [ "{0}.manifest.xml".format(self.args.get('prefix')) ] try: download_files(bucket, manifest_keys, directory, service=self.service, config=self.config) except AWSError as err: if err.code == 'NoSuchEntity': error = ("manifest file '{0}' does not exist in bucket " "'{1}'.") raise ArgumentError(error.format(manifest_keys[0], bucket)) else: raise self._delete_manifest_parts(manifest_keys, directory) self._delete_manifest_keys(manifest_keys) finally: shutil.rmtree(directory)
def _download_by_prefix(self, directory): bucket = self.args.get('bucket') prefix = self.args.get('prefix') manifest_keys = get_manifest_keys(bucket, prefix, service=self.service, config=self.config) if not manifest_keys: if prefix: raise ArgumentError( "no manifests found with prefix '{0}' in bucket '{1}'." .format(prefix, bucket)) else: raise ArgumentError("no manifests found in bucket '{0}'." .format(bucket)) try: download_files(bucket, manifest_keys, directory, service=self.service, config=self.config, show_progress=self.args.get('show_progress', True)) except AWSError as err: if err.code != 'NoSuchEntity': raise raise ArgumentError( "cannot find manifest file(s) {0} in bucket '{1}'." .format(",".join(manifest_keys), bucket)) self._download_parts(manifest_keys, directory)