手机微信收到的图片data格式转码为jpg等图片文件 - 果核剥壳

手机微信中经常收到各种图片，想批量操作，但微信使用的是data格式，无法进行预览，如果图片较多，全部删除又怕删除有用的图片，使用以下代码可以直接读取手机中的微信图片data转成可以预览的图片，可以进行批量操作，或者用于清理手机中的垃圾图片，非常好用。不仅可以提取图片格式，还可以提取其它文件格式。
使用的是进程+线程的方式加快转换速度。

# 系统库
import sys
import os
import stat
# 数据类型库
from enum import Enum
import math
from time import perf_counter

#线程库
import threading
from multiprocessing import Pool,Process,cpu_count
import multiprocessing

sem = threading.Semaphore(1000)

class ImgType(Enum):
gif = 0x4749 # GIF (gif)，文件头：47494638
jpg = 0xFFD8 # JPEG (jpg)，文件头：FFD8FF
png = 0x8950 # PNG (png)，文件头：89504E47
tif = 0x4949 # TIFF (tif)，文件头：49492A00
bmp = 0x424D # Windows Bitmap (bmp)，文件头：424D
dwg = 0x4143 # CAD (dwg)，文件头：41433130
psd = 0x3842 # Adobe Photoshop (psd)，文件头：38425053
rtf = 0x7B5C # Rich Text Format (rtf)，文件头：7B5C727466
xml = 0x3C3F # XML (xml)，文件头：3C3F786D6C
html = 0x6874 # HTML (html)，文件头：68746D6C3E
eml = 0x4465 # Email [thorough only] (eml)，文件头：44656C69766572792D646174653A
dbx = 0xCFAD # Outlook Express (dbx)，文件头：CFAD12FEC5FD746F
pst = 0x2142 # Outlook (pst)，文件头：2142444E
xlsOrdoc = 0xD0CF # MS Word/Excel (xls.or.doc)，文件头：D0CF11E0
mdb = 0x5374 # MS Access (mdb)，文件头：5374616E64617264204A
wpd = 0xFF57 # WordPerfect (wpd)，文件头：FF575043
pdf = 0x2550 # Adobe Acrobat (pdf)，文件头：255044462D312E
qdf = 0xAC9E # Quicken (qdf)，文件头：AC9EBD8F
pwl = 0xE382 # Windows Password (pwl)，文件头：E3828596
zip = 0x504B # ZIP Archive (zip)，文件头：504B0304
rar = 0x5261 # RAR Archive (rar)，文件头：52617221
wav = 0x5741 # Wave (wav)，文件头：57415645
avi = 0x4156 # AVI (avi)，文件头：41564920
ram = 0x2E72 # Real Audio (ram)，文件头：2E7261FD
rm = 0x2E52 # Real Media (rm)，文件头：2E524D46
mpg = 0x0000 # MPEG (mpg)，文件头：000001BA
mpeg = 0x0000 # MPEG (mpg)，文件头：000001B3
mov = 0x6D6F # Quicktime (mov)，文件头：6D6F6F76
asf = 0x3026 # Windows Media (asf)，文件头：3026B2758E66CF11
mid = 0x4D54 # MIDI (mid)，文件头：4D546864

class WechatConvert(object):

@staticmethod
def find_img_type(file_path):
os.chmod(file_path, stat.S_IWRITE) # 取消文件只读属性
with open(file_path, 'rb+') as f:
byte1 = int.from_bytes(f.read(1), byteorder=sys.byteorder)
byte2 = int.from_bytes(f.read(1), byteorder=sys.byteorder)
for img_enum in ImgType:
png_tuple = WechatConvert.hex_to_tuple(img_enum.value)
if png_tuple[0] ^ byte1 == png_tuple[1] ^ byte2:
return img_enum.name, png_tuple[0] ^ byte1
return 'none', '0x' + hex(byte1) + hex(byte2) # 返回找不到的图片类型为:none
# raise Exception("不支持的图片类型")

@staticmethod
def hex_to_tuple(img_type):
return img_type >> 8, img_type & 0b11111111

def convert(self, file_path, output_path=".",number='1',total='1'):

with sem: # 限制最大线程数
file_name = file_path.split("\\")[-1]
# 获取图片类型
img_type,img_xor = WechatConvert.find_img_type(file_path)
convert_filetype = ['gif','jpg','png','tif','bmp']
if img_type!='none':# 只转换支持的图片类型
if img_type in convert_filetype:
print('正在转换图片' + str(number) + '/' + str(total) +'：' + file_name)
with open(file_path, 'rb+') as fd:
# 读取2 byte
with open(output_path + "\\" + file_name + "." + img_type, 'wb+') as w:
while True:
b = fd.read(1)
if not b:
break
real = int.from_bytes(b, byteorder=sys.byteorder) ^ img_xor
real_bytes = int.to_bytes(real, 1, sys.byteorder)
w.write(real_bytes)
print('图片转换完成：'+ str(number) + '/' + str(total) + '：' + file_name)
else:
os.rename(output_path + "\\" + file_name ,output_path + "\\" + file_name + "." + img_type) # 给无法转换的文件加后缀名

def thread_it(func, *args):
'''将函数打包进线程'''
file_path_list = args[0]
output_path = args[1]
number = args[2]
total = args[3]
for file_name in file_path_list:
t = threading.Thread(target=func, args=(file_name, output_path, number + file_path_list.index(file_name) + 1 , total)) # 创建
t.start() # 启动

if __name__ == '__main__':
multiprocessing.freeze_support()
start = perf_counter() # 记下开始时刻
file_name_list = os.listdir(os.getcwd())
cpu_count = cpu_count()
p = Pool(cpu_count)

convert_file_list=[]
for file_name in file_name_list:
splittext = os.path.splitext(file_name)[-1]
if splittext == '.dat':
convert_file_list.append(file_name)

convert_file_list_qty = math.ceil(len(convert_file_list) / cpu_count)
for n in range(cpu_count):#进程数
if convert_file_list_qty * n < len(convert_file_list):
p.apply_async(func=thread_it,args=(WechatConvert().convert,convert_file_list[convert_file_list_qty * n : convert_file_list_qty * (n+1)], ".",convert_file_list_qty * n , len(convert_file_list)))

p.close()
p.join()
end = perf_counter() # 记下结束时刻
process_time = end - start
if process_time > 60:
process_time = process_time /60
process_time_str = str(process_time) + '分'
elif process_time > 60 * 60:
process_time = process_time / 60 /60
process_time_str = str(process_time) + '小时'
else:
process_time_str = str(process_time) + '秒'
print('=================图片全部转换完成===========')
print('一共转换图片数：' + str(len(convert_file_list)))
print('一共用时：' + process_time_str)