import imageio
imageio.plugins.ffmpeg.download()
import win_unicode_console
win_unicode_console.enable()
import sys,os
from PyQt5.QtCore import *
from PyQt5.QtWidgets import (QWidget, QPushButton, QLineEdit,QLabel,
                              QApplication,QFileDialog)
from moviepy.video.io.VideoFileClip  import VideoFileClip
class login(QWidget):
    def __init__(self):
        super(login,self).__init__()
        self.initUI()
    def initUI(self):
        #源文件選擇按鈕和選擇編輯框
        self.source_btn = QPushButton('源文件', self)
        self.source_btn.move(30, 30)
        self.source_btn.resize(60,30)
        self.source_btn.clicked.connect(self.select_source)
        self.source_le = QLineEdit(self)
        self.source_le.move(120, 30)
        self.source_le.resize(250,30)
        # 存儲文件選擇按鈕和選擇編輯框
        self.target_btn = QPushButton('目標路徑', self)
        self.target_btn.move(30, 90)
        self.target_btn.resize(60, 30)
        self.target_btn.clicked.connect(self.select_target)
        self.target_le = QLineEdit(self)
        self.target_le.move(120, 90)
        self.target_le.resize(250, 30)
        #截切開始時間輸入框和提示
        self.startLabel = QLabel(self)
        self.startLabel.move(30, 150)
        self.startLabel.resize(60,30)
        self.startLabel.setText("開始秒")
        self.start_le = QLineEdit(self)
        self.start_le.move(120,150)
        self.start_le.resize(50,30)
        # 截切結(jié)束時間輸入框和提示
        self.stopLabel = QLabel(self)
        self.stopLabel.move(230, 150)
        self.stopLabel.resize(60,30)
        self.stopLabel.setText("結(jié)束秒")
        self.stop_le = QLineEdit(self)
        self.stop_le.move(320,150)
        self.stop_le.resize(50,30)
        #保存按鈕，調(diào)取數(shù)據(jù)增加函數(shù)等
        self.save_btn = QPushButton('開始',self)
        self.save_btn.move(30, 210)
        self.save_btn.resize(140, 30)
        self.save_btn.clicked.connect(self.addNum)
 
        #執(zhí)行成功返回值顯示位置設(shè)置
        self.result_le = QLabel(self)
        self.result_le.move(30, 270)
        self.result_le.resize(340, 30)
        #整體界面設(shè)置
        self.setGeometry(400, 400, 400, 400)
        self.setWindowTitle('視頻剪切')#設(shè)置界面標題名
        self.show()
    # 打開的視頻文件名稱
    def select_source(self):
        target,fileType = QFileDialog.getOpenFileName(self, "選擇源文件", "C:/")
        self.source_le.setText(str(target))
    #保存的視頻文件名稱，要寫上后綴名
    def select_target(self):
        target,fileType = QFileDialog.getSaveFileName(self, "選擇保存路徑", "C:/")
        self.target_le.setText(str(target))
    def addNum(self):
        source = self.source_le.text().strip()#獲取需要剪切的文件
        target = self.target_le.text().strip()#獲取剪切后視頻保存的文件
        start_time = self.start_le.text().strip()#獲取開始剪切時間
        stop_time = self.stop_le.text().strip()#獲取剪切的結(jié)束時間
        video = VideoFileClip(source)#視頻文件加載
        video = video.subclip(int(start_time), int(stop_time))#執(zhí)行剪切操作
        video.to_videofile(target, fps=20, remove_temp=True)#輸出文件
        self.result_le.setText("ok!")#輸出文件后界面返回OK
        self.result_le.setStyleSheet("color:red;font-size:40px")#設(shè)置OK顏色為紅色，大小為四十像素
        self.result_le.setAlignment(Qt.AlignCenter)#OK在指定框內(nèi)居中
if __name__=="__main__":
    app = QApplication(sys.argv)
    ex = login()
    sys.exit(app.exec_())