from PyQt5 import QtCore, QtGui, QtWidgets
import time
from SenGen import generate_vid, generate_channel, generate_playlist

class Worker(QtCore.QThread):
    updateProgress = QtCore.pyqtSignal(int)

    def __init__(self,link,limit,trans,lang,title,delEng,useRange,min,max,deck,note,field,video=False,channel=False,playlist=False, parent=None,field2=None,file=None):
        super(QtCore.QThread, self).__init__()
        self.video = video
        self.channel = channel
        self.playlist = playlist
        self.link = link
        self.limit = limit
        self.trans = trans
        self.lang = lang
        self.title = title
        self.delEng = delEng
        self.useRange = useRange
        self.min = min
        self.max = max
        self.deck = deck
        self.note = note
        self.field = field
        self.field2 = field2
        self.file = file


        def run():
            if video:
                generate_vid(link,limit,trans,lang,title,delEng,useRange,min,max,deck,note,field,field2,file,Worker.updateProgress)
            elif channel:
                generate_channel(link,limit,trans,lang,title,delEng,useRange,min,max,deck,note,field,field2,file,Worker.updateProgress)
            else:
                generate_playlist(link,limit,trans,lang,title,delEng,useRange,min,max,deck,note,field,field2,file,Worker.updateProgress)


class Ui_Form(object):
    def setupUi(self, Form):
        Form.setObjectName("Form")
        Form.resize(341, 62)
        self.verticalLayout = QtWidgets.QVBoxLayout(Form)
        self.verticalLayout.setObjectName("verticalLayout")
        self.label = QtWidgets.QLabel(Form)
        font = QtGui.QFont()
        font.setPointSize(8)
        self.label.setFont(font)
        self.label.setObjectName("label")
        self.verticalLayout.addWidget(self.label, 0, QtCore.Qt.AlignHCenter)
        self.progressBar = QtWidgets.QProgressBar(Form)
        self.progressBar.setProperty("value", 0)
        self.progressBar.setObjectName("progressBar")
        self.verticalLayout.addWidget(self.progressBar)

        self.retranslateUi(Form)
        QtCore.QMetaObject.connectSlotsByName(Form)
        self.worker = Worker("https://www.youtube.com/watch?v=OkHkuT59Rw0",30,False,"english",False,False,False,0,0,"comments","Basic","Front",True)
        self.worker.start()
        self.worker.updateProgress.connect(self.setProgress)

    def retranslateUi(self, Form):
        _translate = QtCore.QCoreApplication.translate
        Form.setWindowTitle(_translate("Form", "Form"))
        self.label.setText(_translate("Form", "Generating"))

    def setProgress(self, progress):
        self.progressBar.setValue(progress)




if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    Form = QtWidgets.QWidget()
    ui = Ui_Form()
    ui.setupUi(Form)
    Form.show()
    sys.exit(app.exec_())