|
|
# pip uninstall -y Pillow==9.5.0
|
|
|
# pip install Pillow==9.5.0
|
|
|
import os
|
|
|
from urllib.parse import unquote
|
|
|
import uuid
|
|
|
import shutil
|
|
|
import subprocess
|
|
|
import pkg_resources
|
|
|
import sys
|
|
|
|
|
|
|
|
|
def install(package):
|
|
|
subprocess.check_call(
|
|
|
[sys.executable, "-m", "pip", "install", package, "-i", "https://pypi.tuna.tsinghua.edu.cn/simple"])
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
# 安装ffmpeg
|
|
|
try:
|
|
|
# 尝试导入包
|
|
|
pkg_resources.get_distribution('ffmpeg')
|
|
|
except pkg_resources.DistributionNotFound:
|
|
|
# 如果未安装，使用pip安装ffmpeg
|
|
|
print("ffmpeg库未安装，正在安装...")
|
|
|
install('ffmpeg-python ')
|
|
|
finally:
|
|
|
# 包安装后或者已存在时，导入包
|
|
|
import ffmpeg
|
|
|
|
|
|
print("ffmpeg库已加载。")
|
|
|
|
|
|
# 安装Pillow
|
|
|
try:
|
|
|
# 尝试导入包
|
|
|
pkg_resources.get_distribution('Pillow')
|
|
|
except pkg_resources.DistributionNotFound:
|
|
|
# 如果未安装，使用pip安装ffmpeg
|
|
|
print("Pillow库未安装，正在安装...")
|
|
|
install('Pillow==9.5.0')
|
|
|
finally:
|
|
|
# 包安装后或者已存在时，导入包
|
|
|
from PIL import Image
|
|
|
|
|
|
print("Pillow库已加载。")
|
|
|
|
|
|
# 创建目录
|
|
|
directory = "assets/images/"
|
|
|
if os.path.exists("target.txt"):
|
|
|
os.remove("target.txt")
|
|
|
|
|
|
# 检查目录是否存在
|
|
|
if os.path.exists(directory):
|
|
|
# 删除目录及其所有内容
|
|
|
shutil.rmtree(directory)
|
|
|
os.makedirs(directory)
|
|
|
|
|
|
# 打开文件
|
|
|
with open('source.txt', 'r', encoding='utf-8') as file:
|
|
|
# 逐行读取文件内容
|
|
|
for line in file:
|
|
|
# 每一行
|
|
|
video_path = line.strip()
|
|
|
|
|
|
# 获取信息
|
|
|
probe = ffmpeg.probe(video_path)
|
|
|
# print('source_video_path: {}'.format(video_path))
|
|
|
format = probe['format']
|
|
|
bit_rate = int(format['bit_rate']) / 1000
|
|
|
size = int(format['size']) / 1024 / 1024
|
|
|
video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)
|
|
|
if video_stream is None:
|
|
|
print('No video stream found!')
|
|
|
continue
|
|
|
width = int(video_stream['width'])
|
|
|
height = int(video_stream['height'])
|
|
|
num_frames = int(video_stream['nb_frames'])
|
|
|
fps = int(video_stream['r_frame_rate'].split('/')[0]) / int(video_stream['r_frame_rate'].split('/')[1])
|
|
|
duration = float(video_stream['duration'])
|
|
|
|
|
|
# Convert size to MB
|
|
|
video_size = "{:.1f}MB".format(size)
|
|
|
video_size = video_size.replace("M", "")
|
|
|
# print(video_size)
|
|
|
# Convert duration to minutes and seconds
|
|
|
minutes = int(duration // 60)
|
|
|
seconds = int(duration % 60)
|
|
|
# Format minutes and seconds with leading zeros if necessary
|
|
|
minutes_str = str(minutes).zfill(2)
|
|
|
seconds_str = str(seconds).zfill(2)
|
|
|
video_length = "{}:{}".format(minutes_str, seconds_str)
|
|
|
# print(video_length)
|
|
|
|
|
|
# 获取中文名称
|
|
|
# 使用字符串分割方法，获取最后一个'/'后面的部分
|
|
|
last_part = video_path.split('/')[-1]
|
|
|
# 对URL编码的字符串进行解码
|
|
|
video_name = unquote(last_part)
|
|
|
|
|
|
# 设置输出图片路径
|
|
|
# 生成一个随机的 UUID
|
|
|
new_uuid = uuid.uuid4()
|
|
|
|
|
|
# 将 UUID 对象转换为字符串类型
|
|
|
uuid_string = str(new_uuid)
|
|
|
video_thumb = directory + uuid_string + '.jpg'
|
|
|
if os.path.exists(video_thumb):
|
|
|
os.remove(video_thumb)
|
|
|
|
|
|
# 使用 ffmpeg-python 进行视频截取
|
|
|
ffmpeg.input(video_path).filter('select', 'gte(n,0)').output(video_thumb, vframes=1).run()
|
|
|
|
|
|
# 打开原始图像
|
|
|
image = Image.open(video_thumb)
|
|
|
# 获取原始图像的宽度和高度
|
|
|
width, height = image.size
|
|
|
# 计算等比例缩放后的高度
|
|
|
new_height = int((320 / width) * height)
|
|
|
# 缩放图像
|
|
|
resized_image = image.resize((320, new_height), Image.ANTIALIAS)
|
|
|
# 保存缩放后的图像
|
|
|
resized_image.save(video_thumb)
|
|
|
# 关闭原始图像
|
|
|
image.close()
|
|
|
|
|
|
# 读入模板文件
|
|
|
with open('Template.html', 'r', encoding='utf-8') as tFile:
|
|
|
# 读取文件的全部内容
|
|
|
content = tFile.read()
|
|
|
content = content.replace("{{video_url}}", video_path)
|
|
|
content = content.replace("{{video_size}}", video_size)
|
|
|
content = content.replace("{{video_length}}", video_length)
|
|
|
content = content.replace("{{video_thumb}}", video_thumb)
|
|
|
content = content.replace("{{video_name}}", video_name.replace(".mp4",""))
|
|
|
# 打开文件并设置为追加模式
|
|
|
with open('target.txt', 'a', encoding='utf-8') as fw:
|
|
|
# 向文件追加内容
|
|
|
fw.write(content + "\n")
|
|
|
|
|
|
print("恭喜，所有操作成功完成！")
|
