import os
from moviepy.video.io.VideoFileClip import VideoFileClip
from pydub import AudioSegment


def clip_video(source_file, target_file, start_time, stop_time):
  """
  利用moviepy進(jìn)行視頻剪切
  :param source_file: 原視頻的路徑，mp4格式
  :param target_file: 生成的目標(biāo)視頻路徑，mp4格式
  :param start_time: 剪切的起始時(shí)間點(diǎn)（第start_time秒）
  :param stop_time: 剪切的結(jié)束時(shí)間點(diǎn)（第stop_time秒）
  :return:
  """
  validate_file(source_file)
  source_video = VideoFileClip(source_file)
  video = source_video.subclip(int(start_time), int(stop_time)) # 執(zhí)行剪切操作
  video.write_videofile(target_file) # 輸出文件


def clip_audio(source_file, target_file, start_time, stop_time):
  """
  利用pydub進(jìn)行音頻剪切。pydub支持源文件為 mp4格式，因此這里的輸入可以與視頻剪切源文件一致
  :param source_file: 原視頻的路徑，mp4格式
  :param target_file: 生成的目標(biāo)視頻路徑，mp4格式
  :param start_time: 剪切的起始時(shí)間點(diǎn)（第start_time秒）
  :param stop_time: 剪切的結(jié)束時(shí)間點(diǎn)（第stop_time秒）
  :return:
  """
  validate_file(source_file)
  audio = AudioSegment.from_file(source_file, "mp4")
  audio = audio[start_time * 1000: stop_time * 1000]
  audio_format = target_file[target_file.rindex(".") + 1:]
  audio.export(target_file, format=audio_format)


def combine_video_audio(video_file, audio_file, target_file, delete_tmp=False):
  """
  利用 ffmpeg將視頻和音頻進(jìn)行合成
  :param video_file:
  :param audio_file:
  :param target_file:
  :param delete_tmp: 是否刪除剪切過(guò)程生成的原視頻/音頻文件
  :return:
  """
  validate_file(video_file)
  validate_file(audio_file)
  # 注：需要先指定音頻再指定視頻，否則可能出現(xiàn)無(wú)聲音的情況
  command = "ffmpeg -y -i {0} -i {1} -vcodec copy -acodec copy {2}".format(audio_file, video_file, target_file)
  os.system(command)
  if delete_tmp:
    os.remove(video_file)
    os.remove(audio_file)


def clip_handle(source_file, target_file, start_time, stop_time, tmp_path=None, delete_tmp=False):
  """
  將一個(gè)視頻文件按指定時(shí)間區(qū)間進(jìn)行剪切
  :param source_file: 原視頻文件
  :param target_file: 目標(biāo)視頻文件
  :param start_time: 剪切的起始時(shí)間點(diǎn)（第start_time秒）
  :param stop_time: 剪切的結(jié)束時(shí)間點(diǎn)（第stop_time秒）
  :param tmp_path: 剪切過(guò)程的文件存放位置
  :param delete_tmp: 是否刪除剪切生成的文件
  :return:
  """
  # 設(shè)置臨時(shí)文件名
  if tmp_path is None or not os.path.exists(tmp_path):
    # 如果沒(méi)有指定臨時(shí)文件路徑，則默認(rèn)與目標(biāo)文件的位置相同
    tmp_path = target_file[: target_file.rindex("/") + 1]
  target_file_name = target_file[target_file.rindex("/") + 1: target_file.rindex(".")]
  tmp_video = tmp_path + "v_" + target_file_name + ".mp4"
  tmp_audio = tmp_path + "a_" + target_file_name + ".mp4"

  # 執(zhí)行文件剪切及合成
  clip_video(source_file, tmp_video, start_time, stop_time)
  clip_audio(source_file, tmp_audio, start_time, stop_time)
  combine_video_audio(tmp_video, tmp_audio, target_file, delete_tmp)


def validate_file(source_file):
  if not os.path.exists(source_file):
    raise FileNotFoundError("沒(méi)有找到該文件：" + source_file)


def test_example():
  """
  測(cè)試?yán)?
  :return:
  """
  root_path = 'XXX/videos/'
  video_name = "test.mp4"
  source_file = root_path + video_name
  start_time = 5
  stop_time = 6

  # 設(shè)置目標(biāo)文件名
  target_name = str(start_time) + "_" + str(stop_time)
  target_file = root_path + "c_" + target_name + ".mp4"
  # 處理主函數(shù)
  clip_handle(source_file, target_file, start_time, stop_time)


if __name__ == "__main__":
  test_example()