用python导出mht文件中的JPG图片

import os
import sys
import re
import time
import base64

def extract_data(filename, dir):
	file = open(filename, "r")
	content = file.read()
	index = 0
	for match in re.finditer(r'(?=/9j)(.|\n)*?(?=(\n\n))', content):
		index = index + 1
		print("%s/%s.jpg" % (dir, index))
		newfile = open("%s/%s.jpg" % (dir, index), "wb")
		real = re.sub("\n", "", match.group())
		length = len(real)
		times = (4 - length % 4) % 4
		i = 0
		
		while i < times:
			real = "%sA===" % real
			i = i + 1
			
		#newfile.write(real)	
		newfile.write(base64.b64decode(real))
		newfile.close()
	file.close()
	
def export_file(file_path):
	r  = re.search(r"(?<=\\|/).*", file_path)
	file_name_ext = file_path
	while r != None:	
		r = r.group()
		file_name_ext = r
		r = re.search(r"(?<=\\|/).*", r)		
	
	pattern = "%s"  % file_name_ext
	pattern = re.sub("\[", "\\\[", pattern)
	pattern = re.sub("\]", "\\\]", pattern)
	pattern = re.sub("\(", "\\\(", pattern)
	pattern = re.sub("\)", "\\\)", pattern)
	pattern = ".*?(?=%s)" % pattern
	
	path = re.search(pattern, file_path)
	if path != None:
		path = path.group()
	else:
		path = ""
	dot = re.search(r"\.", file_name_ext)
	file_name = file_name_ext
	if dot != None:
		file_name = re.search(r".*?(?=\.)", file_name_ext)
		file_name = file_name.group()
	else:
		file_name = file_name + "_dir"
		
	full_path = path + file_name
	bExist = os.path.exists(full_path)
	if bExist != True:
		os.mkdir(full_path)		
	else:
		print("The Directory \"%s\" Exists!\n" % full_path)
	extract_data(file_path, full_path)
	
def main():
	print(os.name)
	
	index = 0
	for arg in sys.argv:
		index += 1
		if index == 1:
			continue
		if index == 2:
			export_file(arg)
	
	if index == 1:
		file_path = "E:\\Scripts\\test.mht"
		export_file(file_path)
		
if __name__ == "__main__":
	main()
	print("just wait for 2 seconds!\n")
	time.sleep(2)