def _get_sample_userdata(scripts=[IGNORED, BASH_SCRIPT], compress=True, use_cloudinit=True): files = utils.strings_to_files(scripts, fname_prefix='sc') return userdata.bundle_userdata_files(files, compress=compress, use_cloudinit=use_cloudinit)
if f.name in filenames: continue contents = StringIO.StringIO(orig_tf.extractfile(f).read()) new_tf.addfile(f, contents) new_tf.close() tarstr.seek(0) new_tf = tarfile.TarFile(fileobj=tarstr, mode='r') return userdata_script_from_files([], tar_file=new_tf) else: mpmime = get_mp_from_userdata(userdata_string, decompress=decompress) msgs = [] for msg in mpmime.get_payload(): if msg.get_filename() in filenames: continue msgs.append(msg) mpmime.set_payload(msgs) return mp_userdata_from_files([], multipart_mime=mpmime, compress=decompress) if __name__ == '__main__': files = utils.strings_to_files(['#!/bin/bash\nhostname', '#!/bin/bash\ndate'], fname_prefix='sc_userdata_file') files += utils.string_to_file('#ignored\nblahblahblah', 'sc_metadata') script = bundle_userdata_files(files, use_cloudinit=False) f = open('/tmp/tester.sh', 'w') f.write(script) f.close() os.chmod('/tmp/tester.sh', 0750)
continue contents = StringIO.StringIO(orig_tf.extractfile(f).read()) new_tf.addfile(f, contents) new_tf.close() tarstr.seek(0) new_tf = tarfile.TarFile(fileobj=tarstr, mode='r') return userdata_script_from_files([], tar_file=new_tf) else: mpmime = get_mp_from_userdata(userdata_string, decompress=decompress) msgs = [] for msg in mpmime.get_payload(): if msg.get_filename() in filenames: continue msgs.append(msg) mpmime.set_payload(msgs) return mp_userdata_from_files([], multipart_mime=mpmime, compress=decompress) if __name__ == '__main__': files = utils.strings_to_files( ['#!/bin/bash\nhostname', '#!/bin/bash\ndate'], fname_prefix='sc_userdata_file') files += utils.string_to_file('#ignored\nblahblahblah', 'sc_metadata') script = bundle_userdata_files(files, use_cloudinit=False) f = open('/tmp/tester.sh', 'w') f.write(script) f.close() os.chmod('/tmp/tester.sh', 0750)