def main(self): manifest = BundleManifest.read_from_fileobj( self.args['manifest'], privkey_filename=self.args['privatekey']) for part in manifest.image_parts: part_path = os.path.join(self.args['source'], part.filename) while part_path.startswith('./'): part_path = part_path[2:] if os.path.exists(part_path): part.filename = part_path else: raise RuntimeError( "bundle part '{0}' does not exist; you may need to use " "-s to specify where to find the bundle's parts" .format(part_path)) part_reader_out_r, part_reader_out_w = open_pipe_fileobjs() part_reader = multiprocessing.Process( target=self.__read_bundle_parts, args=(manifest, part_reader_out_w)) part_reader.start() part_reader_out_w.close() waitpid_in_thread(part_reader.pid) image_filename = os.path.join(self.args['destination'], manifest.image_name) with open(image_filename, 'w') as image: unbundlestream = UnbundleStream( config=self.config, source=part_reader_out_r, dest=image, enc_key=manifest.enc_key, enc_iv=manifest.enc_iv, image_size=manifest.image_size, sha1_digest=manifest.image_digest, show_progress=self.args.get('show_progress', False)) unbundlestream.main() return image_filename
def main(self): manifest = self.fetch_manifest( self.service, privkey_filename=self.args['privatekey']) download_out_r, download_out_w = open_pipe_fileobjs() try: self.__create_download_pipeline(download_out_w) finally: download_out_w.close() image_filename = self.__open_dest(manifest) unbundlestream = UnbundleStream( config=self.config, source=download_out_r, dest=self.args['dest'], enc_key=manifest.enc_key, enc_iv=manifest.enc_iv, image_size=manifest.image_size, sha1_digest=manifest.image_digest, show_progress=self.args.get('show_progress', False)) unbundlestream.main() return image_filename
def main(self): manifest = BundleManifest.read_from_fileobj( self.args['manifest'], privkey_filename=self.args['privatekey']) for part in manifest.image_parts: part_path = os.path.join(self.args['source'], part.filename) while part_path.startswith('./'): part_path = part_path[2:] if os.path.exists(part_path): part.filename = part_path else: raise RuntimeError( "bundle part '{0}' does not exist; you may need to use " "-s to specify where to find the bundle's parts".format( part_path)) part_reader_out_r, part_reader_out_w = open_pipe_fileobjs() part_reader = multiprocessing.Process(target=self.__read_bundle_parts, args=(manifest, part_reader_out_w)) part_reader.start() part_reader_out_w.close() waitpid_in_thread(part_reader.pid) image_filename = os.path.join(self.args['destination'], manifest.image_name) with open(image_filename, 'w') as image: unbundlestream = UnbundleStream(config=self.config, source=part_reader_out_r, dest=image, enc_key=manifest.enc_key, enc_iv=manifest.enc_iv, image_size=manifest.image_size, sha1_digest=manifest.image_digest, show_progress=self.args.get( 'show_progress', False)) unbundlestream.main() return image_filename