py 多线程 m3u8 转mp4 过滤广告,结合ffmpeg使用

本文主要是介绍py 多线程 m3u8 转mp4 过滤广告,结合ffmpeg使用，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

python代码：

import csv
import os
import subprocess
from concurrent.futures import ThreadPoolExecutor, as_completed
from urllib.parse import urljoin
import sys
import requestsdef resource_path(relative_path):"""获取资源文件的绝对路径，兼容PyInstaller打包后的环境"""try:base_path = sys._MEIPASSexcept AttributeError:base_path = os.path.abspath(".")return os.path.join(base_path, relative_path)def process_csv(csv_file, output_dir, begin_num, max_workers=8):csv_file = resource_path(csv_file)with open(csv_file, 'r', encoding='utf-8') as file:reader = csv.reader(file)next(reader)  # 跳过表头with ThreadPoolExecutor(max_workers=max_workers) as executor:futures = []try:for index, row in enumerate(reader, start=1):if index < begin_num:continueprint(f"Processing row {index}...")title, _, m3u8_url = rowbase_name = f"{index}.m3u8"cleaned_m3u8_file_name = f"{index}_cleaned.m3u8"# 提交任务到线程池future = executor.submit(download_and_process_m3u8, m3u8_url, output_dir, base_name,cleaned_m3u8_file_name, title, index)futures.append(future)# 等待所有线程完成for future in as_completed(futures):try:future.result()except Exception as e:print(f"Task generated an exception: {e}")except KeyboardInterrupt:print("用户中断，正在取消所有任务...")for future in futures:future.cancel()executor.shutdown(wait=False)sys.exit(1)def download_and_process_m3u8(m3u8_url, output_dir, m3u8_file_name, cleaned_m3u8_file_name, title, index):m3u8_content = requests.get(m3u8_url).textprint(f"Downloaded m3u8 file from {m3u8_url}, content: {m3u8_content}")m3u8_file_path = os.path.join(output_dir, m3u8_file_name)with open(m3u8_file_path, 'w') as m3u8_file:m3u8_file.write(m3u8_content)print(f"m3u8 file downloaded and saved as {m3u8_file_path}")tsM3u8 = os.path.join(output_dir, f"{index}_ts.m3u8")download_ts_files(m3u8_file_path, m3u8_url, tsM3u8)cleaned_m3u8_file_path = os.path.join(output_dir, cleaned_m3u8_file_name)cleaned_m3u8_path = filter_advertisement(m3u8_url, tsM3u8, cleaned_m3u8_file_path)convert_to_mp4(cleaned_m3u8_path, "video_output", title, index)def download_ts_files(m3u8_file_path, m3u8_url, ts_filename):with open(m3u8_file_path, 'r') as m3u8_file:lines = m3u8_file.readlines()for line in lines:if line.startswith('#'):continuets_url = urljoin(m3u8_url, line.strip())with requests.get(ts_url, stream=True) as r:if r.status_code == 200:with open(ts_filename, 'wb') as ts_file:for chunk in r.iter_content(chunk_size=8192):ts_file.write(chunk)print(f"Downloaded {ts_filename}")else:print(f"Failed to download {ts_url}, status code: {r.status_code}")print("All .ts files have been downloaded.")def filter_advertisement(base_url, m3u8_file_path, cleaned_m3u8_file_path):with open(m3u8_file_path, 'r') as m3u8_file:lines = m3u8_file.readlines()cleaned_lines = []skip = Falsead_removed = Falsefor i, line in enumerate(lines):if line.startswith("#EXT-X-KEY") and not ad_removed:skip = Truead_removed = Trueprint(f"Removing key and associated segments starting with: {line.strip()}")continueif skip and line.startswith("#EXTINF"):continueif skip and (line.startswith("http") or line.startswith("/")):continueif line.strip().startswith("https"):last_element = cleaned_lines[-1]if last_element.startswith("#EXTINF"):cleaned_lines.pop()continueif skip and line.startswith("#EXT-X-DISCONTINUITY"):skip = Falseif not skip:cleaned_lines.append(line)new_lines = []for line in cleaned_lines:if line.startswith('#EXT-X-KEY'):uri_part = line.split('URI="')[1].split('"')[0]if not uri_part.startswith('http'):full_uri = urljoin(base_url, uri_part)line = line.replace(uri_part, full_uri)new_lines.append(line)elif line.startswith('/') and not line.startswith('http'):line = urljoin(base_url, line.strip())new_lines.append(line + "\n")else:new_lines.append(line)with open(cleaned_m3u8_file_path, 'w') as cleaned_m3u8_file:cleaned_m3u8_file.writelines(new_lines)print(f"Filtered m3u8 file saved as {cleaned_m3u8_file_path}")return cleaned_m3u8_file_pathdef convert_to_mp4(m3u8_file_path, output_dir, title, index):output_mp4 = os.path.join(output_dir, f"{index}_{title}.mp4")ffmpeg_command = ["ffmpeg","-protocol_whitelist", "file,http,https,tcp,tls,crypto","-i", m3u8_file_path,"-c", "copy","-bsf:a", "aac_adtstoasc",f"{output_mp4}"]print(f"命令行：{' '.join(ffmpeg_command)}")try:subprocess.run(ffmpeg_command, check=True)print(f"Successfully created {output_mp4}")except subprocess.CalledProcessError as e:print(f"Failed to create MP4: {e}")if __name__ == "__main__":csv_file = "ai_video.csv"output_dir = "ts_files"video_outputh = "video_output"max_workers = 4if not os.path.exists(video_outputh):os.makedirs(video_outputh)if not os.path.exists(output_dir):os.makedirs(output_dir)files = [f for f in os.listdir(video_outputh) ifos.path.isfile(os.path.join(video_outputh, f)) and not f.startswith('.')]files_sorted = sorted(files, key=lambda x: int(x.split('_')[0]))print("files:", files_sorted)begin_num = 1if files_sorted:last_file = files_sorted[-1]num = int(last_file.split('_')[0])if num > 0 and num - max_workers > 0:begin_num = num - max_workers + 1print("最后一个文件名是:", last_file, begin_num)else:print("目录中没有符合条件的文件。")try:process_csv(csv_file, output_dir, begin_num, max_workers=max_workers)except KeyboardInterrupt:print("程序被用户中断。")

写入 pip freeze > requirements.txt

打包使用github action
mkdir -p .github/workflows

name: Build Windows Executableon: [push]jobs:build:runs-on: windows-lateststeps:- name: Checkout codeuses: actions/checkout@v2- name: Set up Pythonuses: actions/setup-python@v2with:python-version: '3.12.4'  # 使用适合你的 Python 版本- name: Install dependenciesrun: |python -m pip install --upgrade pippip install pyinstaller requests  # 安装 pyinstaller 和 requestspip install -r requirements.txt  # 如果你有 requirements.txt 文件- name: Build executablerun: pyinstaller --onefile --add-data "ai_video.csv;." test_ts.py- name: Upload artifactuses: actions/upload-artifact@v2with:name: Windows Executablepath: dist/test_ts.exe

ffmpeg 常用命令


ffmpeg -i demo.mp4 -ss 1 -f image2 -vframes 1 out.jpg  // 原视频截图
ffmpeg -hide_banner -i demo.mp4 -i logo.png -filter_complex "overlay=x=xxx:y=xxx" with_watermark.mp4 -y // 原视频添加水印
ffmpeg -i with_watermark.mp4 -ss 1 -f image2 -vframes 1 with_watermark.jpg  // 对添加水印视频截图
ffmpeg -i with_watermark.mp4 -vf "delogo=x=xxx:y=xxx:w=xxx:h=xxx:show=0" -c:a copy no_watermark.mp4 // 给添加水印的视频，去除水印
ffmpeg -protocol_whitelist "file,http,https,tcp,tls,crypto" -i ./ts_files/cleaned_index.m3u8 -c copy -bsf:a aac_adtstoasc output.mp4 转成视频
ffmpeg -i "https://xxx.com/20240814/Zf8gOK3i/index.m3u8" -c copy output.ts 转视频ps网页榜：https://www.nuanque.com/ps/ffmpeg -i demo.mp4 -ss 16 -f image2 -vframes 1 out.jpg  // 原视频截图
ffmpeg -i demo.mp4 -vf "delogo=x=892:y=589:w=385:h=113:show=0" -c:a copy no_watermark.mp4 // 给添加水印的视频，去除水印
ffmpeg -i demo.mp4 -vf "delogo=x=892:y=589:w=385:h=113:show=0, delogo=x=100:y=100:w=150:h=50:show=0" -c:a copy no_watermark.mp4