def test_write_string(self): w0 = wad.WadFile(self.buff, 'w') w0.writestr('test.cfg', b'bind ALT +strafe') w0.writestr(wad.WadInfo('readme.txt'), 'test') info = wad.WadInfo('bytes') info.file_size = len(b'bytes') info.type = wad.TYPE_LUMP w0.writestr(info, io.BytesIO(b'bytes')) w0.close() self.buff.seek(0) w1 = wad.WadFile(self.buff, 'r') self.assertTrue('test.cfg' in w1.namelist(), 'Cfg file should be in Wad file') self.assertTrue('readme.txt' in w1.namelist(), 'Txt file should be in Wad file') self.assertTrue('bytes' in w1.namelist(), 'Bytes should be in Wad file') self.assertEqual(w1.read('test.cfg'), b'bind ALT +strafe', 'Cfg file content should not change') self.assertEqual(w1.read('readme.txt').decode('ascii'), 'test', 'Txt file content should not change') self.assertEqual(w1.read('bytes'), b'bytes', 'Bytes content should not change') w1.close() self.buff.close()
def test_empty_pak_file(self): with wad.WadFile(self.buff, 'w'): pass self.buff.seek(0) with wad.WadFile(self.buff, 'r') as wad_file: self.assertEqual(len(wad_file.namelist()), 0, 'Wad file should have not entries') self.assertEqual(wad_file.start_of_directory, 12, 'Directory should start immediately after header')
def test_zero_byte_file(self): with wad.WadFile(self.buff, 'w') as wad_file: wad_file.writestr('zero.txt', b'') self.buff.seek(0) with wad.WadFile(self.buff) as wad_file: info = wad_file.getinfo('zero.txt') self.assertEqual(info.file_offset, 12, 'File Info offset of test file should be 12') self.assertEqual(info.file_size, 0, 'File Info size of test file should be 0') data = wad_file.read('zero.txt') self.assertEqual(len(data), 0, 'Length of bytes read should be zero.')
def test_context_manager(self): with wad.WadFile('./test_data/test.wad', 'r') as wad_file: self.assertFalse(wad_file.fp.closed, 'File should be open') self.assertEqual(wad_file.mode, 'r', 'File mode should be \'r\'') fp = wad_file.fp wad_file._did_modify = False self.assertTrue(fp.closed, 'File should be closed') self.assertIsNone(wad_file.fp, 'File pointer should be cleaned up')
def test_append(self): f = open('./test_data/test.wad', 'rb') buff = io.BytesIO(f.read(-1)) f.close() wad_file = wad.WadFile(buff, 'a') wad_file.write('./test_data/test.bsp') wad_file.close() buff.seek(0) wad_file = wad.WadFile(buff, 'r') self.assertTrue('test.bsp' in wad_file.namelist(), 'Appended file should be in Wad file') fp = wad_file.fp wad_file.close() self.assertFalse(buff.closed, 'Wad file should not close passed file-like object') buff.close()
def test_write(self): wad_file = wad.WadFile(self.buff, 'w') self.assertFalse(wad_file.fp.closed, 'File should be open') wad_file.write('./test_data/test.mdl') wad_file.write('./test_data/test.bsp', 'e1m1.bsp') self.assertTrue('test.mdl' in wad_file.namelist(), 'Mdl file should be in Wad file') self.assertTrue('e1m1.bsp' in wad_file.namelist(), 'Bsp file should be in Wad file') fp = wad_file.fp wad_file.close() self.assertFalse(fp.closed, 'File should be open') self.assertIsNone(wad_file.fp, 'File pointer should be cleaned up') self.buff.close()
def test_read(self): wad_file = wad.WadFile('./test_data/test.wad', 'r') self.assertFalse(wad_file.fp.closed, 'File should be open') info = wad_file.getinfo('test') self.assertIsNotNone(info, 'FileInfo should not be None') self.assertEqual(info.filename, 'test') self.assertEqual(info.file_size, 5480, 'FileInfo size of test file should be 5480') self.assertEqual(info.file_offset, 12, 'FileInfo offset of test file should be 12') self.assertEqual(info.type, wad.TYPE_MIPTEX, 'FileInfo type of test file should be MIPTEX') file = wad_file.open('test') self.assertIsNotNone(file, 'File should not be None') file.close() fp = wad_file.fp wad_file.close() self.assertTrue(fp.closed, 'File should be closed') self.assertIsNone(wad_file.fp, 'File pointer should be cleaned up')
'--version', dest='version', action='version', help=argparse.SUPPRESS, version='{} version {}'.format(parser.prog, __version__)) args = parser.parse_args() if not wad.is_wadfile(args.file): print('{0}: cannot find or open {1}'.format(parser.prog, args.file), file=sys.stderr) sys.exit(1) if args.list: with wad.WadFile(args.file) as wad_file: info_list = sorted(wad_file.infolist(), key=lambda i: i.filename) lump_types = { 0: 'NONE', 1: 'LABEL', 64: 'LUMP', 65: 'QTEX', 66: 'QPIC', 67: 'SOUND', 68: 'MIPTEX' } def lump_type(num): if num in lump_types: return lump_types[num]
if args.dest == os.getcwd(): wad_path = os.path.dirname(file) if len(args.list) == 1: wad_name = os.path.basename(file).split('.')[0] + '.wad' else: wad_name = 'out.wad' args.dest = os.path.join(wad_path, wad_name) dir = os.path.dirname(args.dest) or '.' if not os.path.exists(dir): os.makedirs(dir) with wad.WadFile(args.dest, mode='w') as wad_file: if not args.quiet: print('Archive: %s' % os.path.basename(args.dest)) for miptex in miptextures: if not miptex: continue buff = io.BytesIO() bsp_file.Miptexture.write(buff, miptex) buff.seek(0) info = wad.WadInfo(miptex.name) info.file_size = 40 + len(miptex.pixels) info.disk_size = info.file_size info.compression = wad.CMP_NONE
version='{} version {}'.format(parser.prog, __version__)) args = parser.parse_args() if not args.list: parser.error('the following arguments are required: list') dir = os.path.dirname(args.file) or '.' if not os.path.exists(dir): os.makedirs(dir) filemode = 'a' if not os.path.isfile(args.file): filemode = 'w' with wad.WadFile(args.file, filemode) as wad_file: if not args.quiet: print('Archive: %s' % os.path.basename(args.file)) # Flatten out palette palette = [] for p in bsp.default_palette: palette += p # Create palette image for Image.quantize() palette_image = Image.frombytes('P', (16, 16), bytes(palette)) palette_image.putpalette(palette) # Process input files for file in args.list: if args.type == 'LUMP':