GetPeFileSize.py

import pefile


def BuildHeaderBuffer(address):
	buffer = ''

	if Word(address) != 0x5a4d:
		raise ValueError("invalid PE header")

	for i in range(1024):
		buffer += chr(Byte(address+i))

	return buffer


def GetPeFileSize(address):
	size = 0

	headers = BuildHeaderBuffer(address)

	pe = pefile.PE(data=headers)

	size += pe.OPTIONAL_HEADER.SizeOfHeaders

	for section in pe.sections:
		size += section.SizeOfRawData

	return hex(size)