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


def download_m3u8(download_path='./mp4/'):
    try:
        log = Log().getlog()
        # 初始化数据库连接
        movie_config = MySqlUtil("movie")

        # 获取未处理的电影记录
        movie_message = MySqlUtil.get_one(movie_config, 'SELECT * FROM `movie` WHERE is_ok=0 LIMIT 1')
        if not movie_message or len(movie_message) < 3:  # 校验结果是否有效
            log.info("没有找到电影记录或无效数据。")
            return

        id, name, url = movie_message[0], movie_message[1], movie_message[2]

        # 构造目标文件路径
        file_path = Path(download_path).joinpath(f"{name}.mp4")

        # 更新数据库状态，使用参数化查询防止 SQL 注入
        sql = f'UPDATE `movie`.`movie` SET `is_ok` = 1 WHERE `id` = {id}'
        MySqlUtil.update(movie_config, sql=sql)

        log.info(f"任务下载中，正在下载 {name}...")

        # 下载 m3u8 文件并转换为 MP4
        m3u8_to_mp4.multithread_download(url, file_path=str(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()