用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)