66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
import math
|
|
import os
|
|
import re
|
|
from pathlib import Path
|
|
from entity.moment_msg import MomentMsg, Media
|
|
|
|
|
|
class VideoExporter:
|
|
def __init__(self, dir_name):
|
|
self.dir_name = dir_name
|
|
if not os.path.exists(f'output/{self.dir_name}/videos/'):
|
|
os.mkdir(f'output/{self.dir_name}/videos/')
|
|
|
|
|
|
def find_video_by_md5(self, md5):
|
|
"""
|
|
使用MD5匹配视频
|
|
"""
|
|
folder_path = Path(f'output/{self.dir_name}/videos/')
|
|
pattern = re.compile(r'^(.*?)(?=_)')
|
|
|
|
for file_path in folder_path.iterdir():
|
|
match = pattern.search(file_path.name)
|
|
if match:
|
|
filename_md5 = match.group()
|
|
if filename_md5 == md5:
|
|
return file_path.name
|
|
|
|
def find_video_by_duration(self, duration):
|
|
"""
|
|
使用视频时长匹配视频
|
|
"""
|
|
folder_path = Path(f'output/{self.dir_name}/videos/')
|
|
pattern = re.compile(r'_([0-9.]+)\.mp4')
|
|
|
|
for file_path in folder_path.iterdir():
|
|
match = pattern.search(file_path.name)
|
|
if match:
|
|
filename_duration = float(match.group(1))
|
|
if math.isclose(filename_duration, duration, abs_tol=0.005):
|
|
return file_path.name
|
|
|
|
def get_videos(self, msg: MomentMsg) -> list[str]:
|
|
""" 获取一条朋友圈的全部视频， 返回值是一个文件路径列表
|
|
"""
|
|
results = []
|
|
if not msg.timelineObject.ContentObject.mediaList:
|
|
return results
|
|
|
|
media = msg.timelineObject.ContentObject.mediaList.media
|
|
for media_item in media:
|
|
if media_item.type == "6":
|
|
duration = media_item.videoDuration
|
|
rounded_duration = round(float(duration), 2)
|
|
# 先用MD5匹配缓存中的视频
|
|
# 如果找不到使用视频时长再次匹配
|
|
video = self.find_video_by_md5(media_item.url.md5)
|
|
if video:
|
|
results.append(f'videos/{video}')
|
|
else:
|
|
video = self.find_video_by_duration(rounded_duration)
|
|
if video:
|
|
results.append(f'videos/{video}')
|
|
|
|
return results
|