53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
import m3u8_to_mp4
|
|
from utils.MySqlUtil import MySqlUtil
|
|
from apscheduler.schedulers.blocking import BlockingScheduler
|
|
import time
|
|
from utils.Log import Log
|
|
from pathlib import Path
|
|
from utils.Db_Execute import get_movie_list, movie_options, movie_add
|
|
import os
|
|
|
|
log = Log()
|
|
|
|
|
|
def download_m3u8():
|
|
try:
|
|
current_directory = os.path.dirname(os.path.abspath(__file__))
|
|
root_path = os.path.abspath(os.path.dirname(current_directory) + os.path.sep + ".")
|
|
project_name = root_path.split(os.path.sep)[-1]
|
|
project_root_path = os.path.abspath(os.path.dirname(__file__)).split(project_name)[0] + project_name + '/mp4/'
|
|
|
|
# 调用 get_movie_list() 获取电影列表
|
|
movie_list = get_movie_list()
|
|
for movie in movie_list:
|
|
|
|
if not movie or len(movie) < 3: # 校验结果是否有效
|
|
log.info("没有找到电影记录或无效数据。")
|
|
return
|
|
|
|
id, name, url = movie[0], movie[1], movie[2]
|
|
|
|
# 构造目标文件路径
|
|
file_path = Path(project_root_path).joinpath(f"{name}.mp4")
|
|
|
|
# 更新数据库状态，使用参数化查询防止 SQL 注入
|
|
movie_options(id)
|
|
|
|
log.info(f"任务下载中，正在下载 {name}...")
|
|
log.info(file_path)
|
|
|
|
# 下载 m3u8 文件并转换为 MP4
|
|
m3u8_to_mp4.multithread_download(url, file_path=file_path)
|
|
log.info(f"成功下载并转换 {name} to {file_path}.")
|
|
|
|
except Exception as e:
|
|
log.error(f"下载过程中出现错误: {e}")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
download_m3u8()
|
|
# str_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())
|
|
# sch = BlockingScheduler(timezone='Asia/Shanghai')
|
|
# sch.add_job(download_m3u8, 'cron', minute='*/2')
|
|
# sch.start()
|