117 lines
5.1 KiB
Python
117 lines
5.1 KiB
Python
|
import hashlib
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
import subprocess
|
|||
|
|
import sys
|
|||
|
|
import traceback
|
|||
|
|
from datetime import date
|
|||
|
|
from pathlib import Path
|
|||
|
|
import filetype
|
|||
|
|
|
|||
|
|
import log
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ImageDecrypter:
|
|||
|
|
|
|||
|
|
def __init__(self, gui: 'Gui', file_path):
|
|||
|
|
self.file_path = file_path
|
|||
|
|
self.gui = gui
|
|||
|
|
self.sns_cache_path = file_path + "/FileStorage/Sns/Cache"
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def get_output_path(dir_name, md5, duration):
|
|||
|
|
if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
|
|||
|
|
# 这是到_internal文件夹
|
|||
|
|
resource_dir = getattr(sys, '_MEIPASS')
|
|||
|
|
# 获取_internal上一级文件夹再拼接
|
|||
|
|
return os.path.join(os.path.dirname(resource_dir), 'output', dir_name, 'videos', f'{md5}_{duration}.mp4')
|
|||
|
|
else:
|
|||
|
|
return os.path.join(os.getcwd(), 'output', dir_name, 'videos', f'{md5}_{duration}.mp4')
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def calculate_md5(file_path):
|
|||
|
|
with open(file_path, "rb") as f:
|
|||
|
|
file_content = f.read()
|
|||
|
|
return hashlib.md5(file_content).hexdigest()
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def get_all_month_between_dates(start_date, end_date) -> list[str]:
|
|||
|
|
result = []
|
|||
|
|
current_date = start_date
|
|||
|
|
while current_date <= end_date:
|
|||
|
|
# 打印当前日期的年份和月份
|
|||
|
|
result.append(current_date.strftime("%Y-%m"))
|
|||
|
|
year = current_date.year + (current_date.month // 12)
|
|||
|
|
month = current_date.month % 12 + 1
|
|||
|
|
# 更新current_date到下个月的第一天
|
|||
|
|
current_date = date(year, month, 1)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def decode(magic, buf):
|
|||
|
|
return bytearray([b ^ magic for b in list(buf)])
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def guess_image_encoding_magic(buf):
|
|||
|
|
header_code, check_code = 0xff, 0xd8
|
|||
|
|
# 微信图片加密方法对字节逐一“异或”，即 源文件^magic(未知数)=加密后文件
|
|||
|
|
# 已知jpg的头字节是0xff，将0xff与加密文件的头字节做异或运算求解magic码
|
|||
|
|
magic = header_code ^ list(buf)[0] if buf else 0x00
|
|||
|
|
# 尝试使用magic码解密，如果第二字节符合jpg特质，则图片解密成功
|
|||
|
|
_, code = ImageDecrypter.decode(magic, buf[:2])
|
|||
|
|
if check_code == code:
|
|||
|
|
return magic
|
|||
|
|
|
|||
|
|
def decrypt_images(self, exporter, start_date, end_date, dir_name) -> None:
|
|||
|
|
"""将图片文件从缓存中复制出来，重命名为{主图字节数}_{缩略图字节数}.jpg
|
|||
|
|
duration单位为秒
|
|||
|
|
"""
|
|||
|
|
months = self.get_all_month_between_dates(start_date, end_date)
|
|||
|
|
|
|||
|
|
total_files = 0
|
|||
|
|
processed_files = 0
|
|||
|
|
for month in months:
|
|||
|
|
source_dir = self.sns_cache_path + "/" + month
|
|||
|
|
total_files = total_files + len(list(Path(source_dir).rglob('*')))
|
|||
|
|
|
|||
|
|
for month in months:
|
|||
|
|
source_dir = self.sns_cache_path + "/" + month
|
|||
|
|
for file in Path(source_dir).rglob('*'):
|
|||
|
|
# 排除缩略图
|
|||
|
|
if not exporter.stop_flag and file.is_file() and not file.name.endswith('_t'):
|
|||
|
|
try:
|
|||
|
|
with open(file, 'rb') as f:
|
|||
|
|
buff = bytearray(f.read())
|
|||
|
|
magic = self.guess_image_encoding_magic(buff)
|
|||
|
|
if magic:
|
|||
|
|
os.makedirs(f"output/{dir_name}/images/{month}/", exist_ok=True)
|
|||
|
|
os.makedirs(f"output/{dir_name}/thumbs/{month}/", exist_ok=True)
|
|||
|
|
main_file_size = file.stat().st_size
|
|||
|
|
thumb_file_size = 0
|
|||
|
|
# 找到对应缩略图
|
|||
|
|
thumb_file = Path(f'{source_dir}/{file.name}_t')
|
|||
|
|
if thumb_file.exists():
|
|||
|
|
thumb_file_size = thumb_file.stat().st_size
|
|||
|
|
# 读缩略图加密
|
|||
|
|
with open(thumb_file, 'rb') as f:
|
|||
|
|
thumb_buff = bytearray(f.read())
|
|||
|
|
|
|||
|
|
# 写缩略图
|
|||
|
|
thumb_destination = (f"output/{dir_name}/thumbs/{month}/"
|
|||
|
|
f"{main_file_size}_{thumb_file_size}.jpg")
|
|||
|
|
with open(thumb_destination, 'wb') as f:
|
|||
|
|
new_thumb_buff = self.decode(magic, thumb_buff)
|
|||
|
|
f.write(new_thumb_buff)
|
|||
|
|
|
|||
|
|
destination = (f"output/{dir_name}/images/{month}/"
|
|||
|
|
f"{main_file_size}_{thumb_file_size}.jpg")
|
|||
|
|
with open(destination, 'wb') as f:
|
|||
|
|
new_buf = self.decode(magic, buff)
|
|||
|
|
f.write(new_buf)
|
|||
|
|
except Exception:
|
|||
|
|
traceback.print_exc()
|
|||
|
|
processed_files = processed_files + 1
|
|||
|
|
# 15%的进度作为处理图片使用
|
|||
|
|
progress = round(processed_files / total_files * 15)
|
|||
|
|
self.gui.update_export_progressbar(progress)
|