Win3x.Org

#!/usr/bin/python3

import pathlib

class Mz2File:
	def __init__(self, image):
		self.path = image.read(0x51).decode().rstrip('\0')
		self.name = image.read(0x0C).decode().rstrip('\0')
		image.seek(0x01, 1)
		self.param1 = int.from_bytes(image.read(4), byteorder='little')
		self.offset = int.from_bytes(image.read(4), byteorder='little')
		self.param2 = int.from_bytes(image.read(4), byteorder='little')
		self.content = None
		self.size = 0
	
	def getPath(self):
		return self.path
	
	def getName(self):
		return self.name
		
	def getFullPath(self):
		return (self.path + self.name)
	
	def getOffset(self):
		return self.offset
	
	def setFileContent(self, content):
		self.content = content
	
	def setSize(self, size:int):
		self.size = size
	
	def getSize(self):
		return self.size
		
	def getContent(self):
		return self.content
	
	def getParam1(self):
		return str(self.param1)
	
	def getParam2(self):
		return str(self.param2)
		
class Mz2Image:
	def __init__(self, filePath):
		image = open(str(filePath), "rb")
		image.seek(0x02, 0)
		self.fileCount = int.from_bytes(image.read(2), byteorder='little')
		image.seek(0x6F, 0)
		self.label = image.read(11).decode()
		image.seek(0xB9, 0)
		self.files = []
		for i in range(self.fileCount):
			self.addMz2File(Mz2File(image))
			image.seek(0x4A, 1)
		
		self.files.sort(key=lambda x: x.offset, reverse=False)
		i = 0
		for i in range(self.fileCount):
			if(i > 0):
				self.files[i - 1].setSize(self.files[i].getOffset() - self.files[i - 1].getOffset())
				self.files[i - 1].setFileContent(image.read(self.files[i - 1].getSize()))
			if(i == self.fileCount -1):
				self.files[i].setFileContent(image.read())
		
		image.close()
		
	def addMz2File(self, file:Mz2File):
		self.files.append(file)
	
	def getFiles(self):
		return self.files
		
	def getLabel(self):
		return self.label
	
	def dirImage(self):
		str = ""
		for file in self.files:
			str += file.getFullPath() + " " + file.getParam1() + " " + file.getParam2() + '\n'
		return str
	
	def getFileCount(self):
		return self.fileCount

for path in pathlib.Path(".").iterdir():

	if path.is_file() and str(path)[-3:] == "MZ2":
		print(path)
		image = Mz2Image(path)
		print(image.label)
		print(image.dirImage())