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進行視頻剪切
  :param source_file: 原視頻的路徑，mp4格式
  :param target_file: 生成的目標視頻路徑，mp4格式
  :param start_time: 剪切的起始時間點（第start_time秒）
  :param stop_time: 剪切的結(jié)束時間點（第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進行音頻剪切。pydub支持源文件為 mp4格式，因此這里的輸入可以與視頻剪切源文件一致
  :param source_file: 原視頻的路徑，mp4格式
  :param target_file: 生成的目標視頻路徑，mp4格式
  :param start_time: 剪切的起始時間點（第start_time秒）
  :param stop_time: 剪切的結(jié)束時間點（第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將視頻和音頻進行合成
  :param video_file:
  :param audio_file:
  :param target_file:
  :param delete_tmp: 是否刪除剪切過程生成的原視頻/音頻文件
  :return:
  """
  validate_file(video_file)
  validate_file(audio_file)
  # 注：需要先指定音頻再指定視頻，否則可能出現(xiàn)無聲音的情況
  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):
  """
  將一個視頻文件按指定時間區(qū)間進行剪切
  :param source_file: 原視頻文件
  :param target_file: 目標視頻文件
  :param start_time: 剪切的起始時間點（第start_time秒）
  :param stop_time: 剪切的結(jié)束時間點（第stop_time秒）
  :param tmp_path: 剪切過程的文件存放位置
  :param delete_tmp: 是否刪除剪切生成的文件
  :return:
  """
  # 設(shè)置臨時文件名
  if tmp_path is None or not os.path.exists(tmp_path):
    # 如果沒有指定臨時文件路徑，則默認與目標文件的位置相同
    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("沒有找到該文件：" + source_file)


def test_example():
  """
  測試例子
  :return:
  """
  root_path = 'XXX/videos/'
  video_name = "test.mp4"
  source_file = root_path + video_name
  start_time = 5
  stop_time = 6

  # 設(shè)置目標文件名
  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()