def __init__(self, parent=None):
QtGui.QMainWindow.__init__(self, parent)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
path = os.path.join(app_root, 'UI', 'images', 'icon.png')
self.setWindowIcon(QtGui.QIcon(path))
self.batch_dialog = BatchAddDialogue(self)
self.ui.saveToLineEdit.setText(desktop_path)
self.ui.BrowseConvertToLineEdit.setText(os.getcwd())
self.ui.BrowseConvertLineEdit.files = []
self.ui.statusbar.showMessage('Ready.')
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.thread_pool = {}
self.ui.tableWidget.horizontalHeader().setResizeMode(0, QtGui.QHeaderView.Stretch)
self.rowcount = 0
#My code start
self.opt_list = []
with f as open('links_to_resume.json', 'r+'):
self.opt_list = json.load(f)
print(opt_list)
self.connect_menu_action()
self.resume_links(opt_list)
#My code end
self.show()
def __init__(self, parent=None, *args, **kwargs):
super(MainWindow, self).__init__(parent, *args, **kwargs)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
path = os.path.join(app_root, 'UI', 'images', 'icon.png')
self.setWindowIcon(QtGui.QIcon(path))
self.batch_dialog = BatchAddDialogue(self)
self.ui.saveToLineEdit.setText(desktop_path)
self.ui.BrowseConvertToLineEdit.setText(os.getcwd())
self.ui.BrowseConvertLineEdit.files = []
self.ui.statusbar.showMessage('Ready.')
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.threadpool = QtCore.QThreadPool()
self.ui.tableWidget.horizontalHeader().setSectionResizeMode(
0, QtWidgets.QHeaderView.Stretch)
self.rowcount = 0
self.connect_menu_action()
self.about = AboutDialog(self)
self.license = LicenseDialogue(self)
self.show()
def __init__(self, parent=None, *args, **kwargs):
super(MainWindow, self).__init__(parent, *args, **kwargs)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
# For getting the icon to work
try:
from PyQt5.QtWinExtras import QtWin
myappid = "my_company.my_product.sub_product.version"
QtWin.setCurrentProcessExplicitAppUserModelID(
myappid) # type: ignore
except ImportError:
pass
global app_root
if app_root:
ico_path = str(Path(app_root) / Path("UI/images/icon.ico"))
else:
app_root = ":/"
ico_path = app_root + "icon.ico"
self.setWindowIcon(QtGui.QIcon(ico_path))
self.batch_dialog = BatchAddDialogue(self)
self.ui.saveToLineEdit.setText(desktop_path)
self.ui.BrowseConvertToLineEdit.setText(str(Path().cwd()))
self.ui.BrowseConvertLineEdit.files = []
self.ui.statusbar.showMessage("Ready.")
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.threadpool = QtCore.QThreadPool()
self.ui.tableWidget.horizontalHeader().setSectionResizeMode(
0, QtWidgets.QHeaderView.Stretch)
self.rowcount = 0
self.connect_menu_action()
self.about = AboutDialog(self)
self.license = LicenseDialogue(self)
self.closing = CloseSignals()
self.show()
def __init__(self):
super(MainWindow, self).__init__()
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
path = os.path.join(app_root, 'UI', 'images', 'icon.png')
self.setWindowIcon(QIcon(path))
self.batch_dialog = BatchAddDialogue(self)
self.ui.saveToLineEdit.setText(downloads_dir)
self.ui.statusbar.showMessage('Ready.')
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.thread_pool = {}
self.ui.tableWidget.horizontalHeader().setSectionResizeMode(
0, QHeaderView.Stretch)
self.rowcount = 0
self.connect_menu_action()
self.show()
def __init__(self, parent=None):
QtWidgets.QMainWindow.__init__(self,
parent,
flags=QtCore.Qt.FramelessWindowHint)
self.m_titleBar = TitleBar(self, "PokerStats")
self.m_content = QtWidgets.QWidget(self)
self.m_content = QtWidgets.QWidget(self)
self.size = []
self.size.append(380)
self.size.append(HEIGHT - 38)
self.move((WIDTH - self.size[0]), 0)
self.ui = Ui_MainWindow()
self.ui.setupUi(self, self.size)
self.ui.verticalLayoutTitle.addWidget(self.m_titleBar)
path = os.path.join(app_root, 'UI', 'images', 'icon.png')
self.setWindowIcon(QtGui.QIcon(path))
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.thread_pool = {}
self.rowcount = 0
self.show()
self.cal = NeoOdds()
self.loopthread = LoopThread(self.cal, self)
self.buttonStart = False
defaults = {
'path':
'C:\\Users\\theis_p\\AppData\\Local\\PokerStars.FR\\PokerStars.log.0',
'result': '2000'
}
self.config = configparser.SafeConfigParser(defaults=defaults)
self.loadIni()
for i in range(200):
self.ui.listWidgetHistory.addItem(str(i))
class MainWindow(QtWidgets.QMainWindow):
def __init__(self, parent=None, *args, **kwargs):
super(MainWindow, self).__init__(parent, *args, **kwargs)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
path = os.path.join(app_root, 'UI', 'images', 'icon.png')
self.setWindowIcon(QtGui.QIcon(path))
self.batch_dialog = BatchAddDialogue(self)
self.ui.saveToLineEdit.setText(desktop_path)
self.ui.BrowseConvertToLineEdit.setText(os.getcwd())
self.ui.BrowseConvertLineEdit.files = []
self.ui.statusbar.showMessage('Ready.')
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.threadpool = QtCore.QThreadPool()
self.ui.tableWidget.horizontalHeader().setSectionResizeMode(
0, QtWidgets.QHeaderView.Stretch)
self.rowcount = 0
self.connect_menu_action()
self.about = AboutDialog(self)
self.license = LicenseDialogue(self)
self.show()
def set_connections(self):
self.ui.download_btn.clicked.connect(self.handleButton)
self.ui.browse_btn.clicked.connect(self.set_destination)
self.ui.BatchAdd.clicked.connect(self.batch_file)
self.ui.BrowseConvertButton.clicked.connect(self.convert_file_browse)
self.ui.ConvertMultipleButton.clicked.connect(self.convert_button)
self.ui.BrowseConvertToButton.clicked.connect(
self.browse_convert_destination)
def batch_file(self):
self.batch_dialog.exec_()
if self.batch_dialog.download is True:
urls = list(self.batch_dialog.ui.UrlList.toPlainText().split('\n'))
for url in urls:
if url.strip():
self.download_url(str(url))
else:
return
def set_destination(self):
file_name = str(
QtWidgets.QFileDialog.getExistingDirectory(self,
"Select Directory"))
if file_name:
self.ui.saveToLineEdit.setText(file_name)
def browse_convert_destination(self):
file_name = str(
QtWidgets.QFileDialog.getExistingDirectory(self,
"Select Directory"))
if file_name:
self.ui.BrowseConvertToLineEdit.setText(file_name)
def convert_button(self):
preferred_format = str(self.ui.ConvertMultipleComboBox.currentText())
out_path = str(self.ui.BrowseConvertToLineEdit.text())
delete_temp = self.ui.DeleteFileCheckBox.isChecked()
if len(self.ui.BrowseConvertLineEdit.files) < 1:
QtWidgets.QMessageBox.information(self, "Error!",
"No files given!")
return
for file_path in self.ui.BrowseConvertLineEdit.files:
self.convert_file(file_path, out_path, preferred_format,
delete_temp)
def convert_file_browse(self):
file_names, _ = QtWidgets.QFileDialog.getOpenFileNames(
self,
"Select files",
filter='Videos (*.mp4 *.ogg *.webm *.flv *.mkv)')
if len(file_names) > 1:
self.ui.BrowseConvertLineEdit.setText('{} Files selected'.format(
len(file_names)))
self.ui.BrowseConvertLineEdit.files = file_names
elif len(file_names) == 1:
self.ui.BrowseConvertLineEdit.setText(file_names[0])
self.ui.BrowseConvertLineEdit.files = file_names
else:
self.ui.BrowseConvertLineEdit.files = file_names
# TODO: Fix it
def convert_file(self,
file_path,
out_path,
preferred_format,
delete_tmp=False):
if file_path.split('.')[-1] == preferred_format:
self.ui.statusbar.showMessage(
'The source and destination formats are same')
return
# TODO: Check file_path (file_name to final preferred_format)
if file_path not in self.convert_list:
options = {
'file_path': file_path,
'preferred_format': preferred_format,
'row_count': self.rowcount,
'delete_tmp': delete_tmp,
'parent': self,
'out_path': out_path,
}
# Using ThreadPool
pprocessorThread = PostProcessor(options)
pprocessorThread.signals.statusSignal.connect(
self.ui.statusbar.showMessage)
pprocessorThread.signals.list_Signal.connect(self.add_to_table)
pprocessorThread.signals.row_Signal.connect(self.decrease_rowcount)
self.threadpool.start(pprocessorThread)
self.ui.tabWidget.setCurrentIndex(2)
self.convert_list.append(file_path)
self.add_to_table([
self.rowcount,
os.path.split(file_path)[-1].split('.')[0], '', '00:00',
'-- KiB/s', 'Converting'
])
self.rowcount += 1
else:
self.ui.statusbar.showMessage('Already Converted')
def download_url(self, url, row=None):
if row is None:
row = self.rowcount
else:
row = row
directory = str(self.ui.saveToLineEdit.text())
quality = False
if self.ui.ConvertCheckBox.isChecked():
quality = str(self.ui.ConvertComboBox.currentText())
options = {
'url': url,
'directory': directory,
'rowcount': row,
'proxy': '',
'convert_format': quality,
'parent': self,
}
if not self.ui.DeleteFileCheckBox.isChecked():
options['keep_file'] = True
# Using ThreadPool
downloadThread = Download(options)
downloadThread.signals.status_bar_signal.connect(
self.ui.statusbar.showMessage)
downloadThread.signals.remove_url_signal.connect(self.remove_url)
downloadThread.signals.add_update_list_signal.connect(
self.add_to_table)
downloadThread.signals.remove_row_signal.connect(
self.decrease_rowcount)
self.threadpool.start(downloadThread)
self.ui.tabWidget.setCurrentIndex(2)
self.ui.statusbar.showMessage('Extracting information...')
self.url_list.append(url)
self.complete_url_list[row] = url
self.rowcount += 1
if len(self.url_list):
if len(self.url_list) < 2:
self.ui.statusbar.showMessage('Downloading {0} file'.format(
len(self.url_list)))
else:
self.ui.statusbar.showMessage('Downloading {0} files'.format(
len(self.url_list)))
else:
self.ui.statusbar.showMessage("Done!")
def handleButton(self):
url = str(self.ui.videoUrlLineEdit.text())
if url == '':
QtWidgets.QMessageBox.information(self, "Error!", "No url given!")
return
can_download, rowcount = self.can_download(url)
if can_download:
self.download_url(url, rowcount)
else:
QtWidgets.QMessageBox.information(
self, "Error!", "This url is already being downloaded")
if len(self.url_list) != 0:
if len(self.url_list) < 2:
self.ui.statusbar.showMessage(
'Downloading {0} file'.format(len(self.url_list)))
else:
self.ui.statusbar.showMessage(
'Downloading {0} files'.format(len(self.url_list)))
else:
self.ui.statusbar.showMessage("Done!")
def can_download(self, url):
if url not in self.url_list:
for row, _url in self.complete_url_list.items():
if url == _url:
return True, row
return True, self.rowcount
else:
return False, self.rowcount
def remove_url(self, url):
try:
self.url_list.remove(url)
except:
pass
def add_to_table(self, values):
self.ui.tableWidget.setRowCount(self.rowcount)
row = values[0]
m = 0
for key in values[1:]:
new_item = QtWidgets.QTableWidgetItem(key)
self.ui.tableWidget.setItem(row, m, new_item)
m += 1
def decrease_rowcount(self):
self.rowcount -= 1
def connect_menu_action(self):
self.ui.actionExit.triggered.connect(self.close)
self.ui.actionLicense.triggered.connect(self.showLicense)
self.ui.actionAbout.triggered.connect(self.showAbout)
def showLicense(self):
self.license.show()
def showAbout(self):
self.about.show()
def closeEvent(self, event):
if len(self.url_list):
msgBox = QtWidgets.QMessageBox(self)
msgBox.setWindowTitle("Exit")
msgBox.setText("Some files are currently being downloaded.")
msgBox.setInformativeText("Do you really want to close?")
msgBox.setStandardButtons(QtWidgets.QMessageBox.Close
| QtWidgets.QMessageBox.Cancel)
msgBox.setDefaultButton(QtWidgets.QMessageBox.Cancel)
ret = msgBox.exec_()
if ret == QtWidgets.QMessageBox.Cancel:
event.ignore()
else:
self.kill_all_threads()
else:
self.kill_all_threads()
def kill_all_threads(self):
self.threadpool.waitForDone(1)
self.close()
class MainWindow(QtWidgets.QMainWindow):
def __init__(self, parent=None):
QtWidgets.QMainWindow.__init__(self,
parent,
flags=QtCore.Qt.FramelessWindowHint)
self.m_titleBar = TitleBar(self, "PokerStats")
self.m_content = QtWidgets.QWidget(self)
self.m_content = QtWidgets.QWidget(self)
self.size = []
self.size.append(380)
self.size.append(HEIGHT - 38)
self.move((WIDTH - self.size[0]), 0)
self.ui = Ui_MainWindow()
self.ui.setupUi(self, self.size)
self.ui.verticalLayoutTitle.addWidget(self.m_titleBar)
path = os.path.join(app_root, 'UI', 'images', 'icon.png')
self.setWindowIcon(QtGui.QIcon(path))
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.thread_pool = {}
self.rowcount = 0
self.show()
self.cal = NeoOdds()
self.loopthread = LoopThread(self.cal, self)
self.buttonStart = False
defaults = {
'path':
'C:\\Users\\theis_p\\AppData\\Local\\PokerStars.FR\\PokerStars.log.0',
'result': '2000'
}
self.config = configparser.SafeConfigParser(defaults=defaults)
self.loadIni()
for i in range(200):
self.ui.listWidgetHistory.addItem(str(i))
def set_connections(self):
self.ui.btnStart.clicked.connect(self.handleButton)
self.ui.btnApply.clicked.connect(self.saveIni)
self.ui.btnSrc.clicked.connect(self.set_destination)
# self.ui.BatchAdd.clicked.connect(self.batch_file)
# self.ui.BrowseConvertButton.clicked.connect(self.convert_file_browse)
# self.ui.ConvertMultipleButton.clicked.connect(self.convert_button)
# self.ui.BrowseConvertToButton.clicked.connect(self.browse_convert_destination)
def set_destination(self):
file_name = str(
QtWidgets.QFileDialog.getExistingDirectory(self,
"Select Directory"))
if file_name is not '':
self.ui.lineEditSrc.setText(file_name)
def handleButton(self):
if not self.buttonStart:
self.src = str(self.ui.lineEditSrc.text())
self.ui.labelInfo.setText(self.src)
self.cal.config(self.src)
self.onThread()
self.buttonStart = True
self.ui.btnStart.setText("Stop")
else:
self.ui.labelInfo.setText("Stop...")
self.loopthread.stop()
self.ui.btnStart.setText("Start ")
self.reset()
self.buttonStart = False
def onThread(self):
self.loopthread = LoopThread(self.cal, self)
self.loopthread.notifyProgress.connect(self.onProgress)
self.loopthread.notifyOddsFlop.connect(self.onOddsFlop)
self.loopthread.notifyOddsTurn.connect(self.onOddsTurn)
self.loopthread.notifyOddsRiver.connect(self.onOddsRiver)
self.loopthread.notifyEquityFlop.connect(self.onEquityFlop)
self.loopthread.notifyEquityTurn.connect(self.onEquityTurn)
self.loopthread.notifyEquityRiver.connect(self.onEquityRiver)
self.loopthread.notifyOddsFlopP.connect(self.onOddsFlopP)
self.loopthread.notifyOddsTurnP.connect(self.onOddsTurnP)
self.loopthread.notifyOddsRiverP.connect(self.onOddsRiverP)
self.loopthread.start()
def onProgress(self, i):
self.ui.progressBarLoop.setValue(i)
def onOddsFlop(self, a):
if a == 0:
self.ui.progressBar_3.setEnabled(False)
else:
self.ui.progressBar_3.setEnabled(True)
self.ui.progressBar_3.setValue(a)
def onOddsTurn(self, a):
if a == 0:
self.ui.progressBar_4.setEnabled(False)
else:
self.ui.progressBar_4.setEnabled(True)
self.ui.progressBar_4.setValue(a)
def onOddsRiver(self, a):
if a == 0:
self.ui.progressBar_5.setEnabled(False)
else:
self.ui.progressBar_5.setEnabled(True)
self.ui.progressBar_5.setValue(a)
def onEquityFlop(self, a):
if a == 0:
self.ui.progressBar.setEnabled(False)
else:
self.ui.progressBar.setEnabled(True)
self.ui.progressBar.setValue(a)
def onEquityTurn(self, a):
if a == 0:
self.ui.progressBar_2.setEnabled(False)
else:
self.ui.progressBar_2.setEnabled(True)
self.ui.progressBar_2.setValue(a)
def onEquityRiver(self, a):
if a == 0:
self.ui.progressBar_6.setEnabled(False)
else:
self.ui.progressBar_3.setEnabled(True)
self.ui.progressBar_6.setValue(a)
def onOddsFlopP(self, a):
self.ui.labelOddsFlopP.setText(str(round(a, 3)) + '%')
def onOddsTurnP(self, a):
self.ui.labelOddsTurnP.setText(str(round(a, 3)) + '%')
def onOddsRiverP(self, a):
self.ui.labelOddsRiverP.setText(str(round(a, 3)) + '%')
def reset(self):
self.ui.progressBar.setValue(0)
self.ui.progressBar_2.setValue(0)
self.ui.progressBar_6.setValue(0)
self.ui.progressBar_3.setValue(0)
self.ui.progressBar_4.setValue(0)
self.ui.progressBar_5.setValue(0)
self.ui.labelBoardCard.setText('')
self.ui.labelHandCard.setText('')
self.ui.lcdNumberPlayer.display(0)
self.ui.labelOddsFlopP.setText('0%')
self.ui.labelOddsTurnP.setText('0%')
self.ui.labelOddsRiverP.setText('0%')
self.cards = []
def loadIni(self):
self.config.read('pokerstats.ini')
if not self.config.has_section('config'):
self.config.add_section('config')
self.ui.lineEditSrc.setText(self.config.get('config', 'path'))
self.ui.spinBox.setValue(int(self.config.get('config', 'result')))
def saveIni(self):
self.config.set('config', 'path', str(self.ui.lineEditSrc.text()))
self.config.set('config', 'result', str(self.ui.spinBox.value()))
with open('pokerstats.ini', 'w') as configfile:
self.config.write(configfile)
class MainWindow(QtGui.QMainWindow):
def __init__(self, parent=None):
QtGui.QMainWindow.__init__(self, parent)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
path = os.path.join(app_root, 'UI', 'images', 'icon.png')
self.setWindowIcon(QtGui.QIcon(path))
self.batch_dialog = BatchAddDialogue(self)
self.ui.saveToLineEdit.setText(desktop_path)
self.ui.BrowseConvertToLineEdit.setText(os.getcwd())
self.ui.BrowseConvertLineEdit.files = []
self.ui.statusbar.showMessage('Ready.')
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.thread_pool = {}
self.ui.tableWidget.horizontalHeader().setResizeMode(0, QtGui.QHeaderView.Stretch)
self.rowcount = 0
self.connect_menu_action()
self.show()
def set_connections(self):
self.ui.download_btn.clicked.connect(self.handleButton)
self.ui.browse_btn.clicked.connect(self.set_destination)
self.ui.BatchAdd.clicked.connect(self.batch_file)
self.ui.BrowseConvertButton.clicked.connect(self.convert_file_browse)
self.ui.ConvertMultipleButton.clicked.connect(self.convert_button)
self.ui.BrowseConvertToButton.clicked.connect(self.browse_convert_destination)
def batch_file(self):
self.batch_dialog.exec_()
if self.batch_dialog.download is True:
urls = list(self.batch_dialog.ui.UrlList.toPlainText().split('\n'))
for url in urls:
self.download_url(str(url))
else:
return
def set_destination(self):
file_name = str(QtGui.QFileDialog.getExistingDirectory(self, "Select Directory"))
if file_name is not '':
self.ui.saveToLineEdit.setText(file_name)
def browse_convert_destination(self):
file_name = str(QtGui.QFileDialog.getExistingDirectory(self, "Select Directory"))
if file_name is not '':
self.ui.BrowseConvertToLineEdit.setText(file_name)
def convert_button(self):
preferred_format = str(self.ui.ConvertMultipleComboBox.currentText())
out_path = str(self.ui.BrowseConvertToLineEdit.text())
delete_temp = self.ui.DeleteFileCheckBox.isChecked()
if len(self.ui.BrowseConvertLineEdit.files) < 1:
QtGui.QMessageBox.information(self, "Error!","No files given!")
return
for file_path in self.ui.BrowseConvertLineEdit.files:
self.convert_file(file_path, out_path, preferred_format, delete_temp)
def convert_file_browse(self):
file_names = [str(file_n) for file_n in list(
QtGui.QFileDialog.getOpenFileNames(self, "Select files",
filter=QtCore.QString('Videos (*.mp4 *.ogg *.webm *.flv *.mkv)')
))
]
if len(file_names) > 1:
self.ui.BrowseConvertLineEdit.setText('{} Files selected'.format(len(file_names)))
self.ui.BrowseConvertLineEdit.files=file_names
elif len(file_names) == 1:
self.ui.BrowseConvertLineEdit.setText(file_names[0])
self.ui.BrowseConvertLineEdit.files=file_names
else:
self.ui.BrowseConvertLineEdit.files=file_names
def convert_file(self,file_path, out_path, preferred_format, delete_tmp=False):
if file_path.split('.')[-1] == preferred_format:
self.ui.statusbar.showMessage('The source and destination formats are same')
return
if file_path not in self.convert_list:
options= {
'file_path': file_path,
'preferred_format': preferred_format,
'row_count': self.rowcount,
'delete_tmp': delete_tmp,
'parent': self,
'out_path': out_path,
}
self.thread_pool['thread{}'.format(self.rowcount)] = PostProcessor(options)
self.thread_pool['thread{}'.format(self.rowcount)].statusSignal.connect(self.ui.statusbar.showMessage)
self.thread_pool['thread{}'.format(self.rowcount)].list_Signal.connect(self.add_to_table)
self.thread_pool['thread{}'.format(self.rowcount)].row_Signal.connect(self.decrease_rowcount)
self.thread_pool['thread{}'.format(self.rowcount)].start()
self.ui.tabWidget.setCurrentIndex(2)
self.convert_list.append(file_path)
self.add_to_table([
self.rowcount,
os.path.split(file_path)[-1].split('.')[0],
'',
'00:00',
'-- KiB/s',
'Converting'
])
self.rowcount += 1
else:
self.ui.statusbar.showMessage('Already Converted')
def download_url(self, url, row = None):
if row >= 0:
row = row
elif row is None:
row = self.rowcount
directory = str(self.ui.saveToLineEdit.text())
quality = False
if self.ui.ConvertCheckBox.isChecked():
quality = str(self.ui.ConvertComboBox.currentText())
print row, self.rowcount
options = {
'url': url,
'directory': directory,
'rowcount': row,
'proxy': '',
'convert_format': quality,
'parent':self,
}
if not self.ui.DeleteFileCheckBox.isChecked():
options['keep_file'] = True
self.thread_pool['thread{}'.format(row)] = Download(options)
self.thread_pool['thread{}'.format(row)].status_bar_signal.connect(self.ui.statusbar.showMessage)
self.thread_pool['thread{}'.format(row)].remove_url_signal.connect(self.remove_url)
self.thread_pool['thread{}'.format(row)].add_update_list_signal.connect(self.add_to_table)
self.thread_pool['thread{}'.format(row)].remove_row_signal.connect(self.decrease_rowcount)
self.thread_pool['thread{}'.format(row)].start()
self.ui.tabWidget.setCurrentIndex(2)
self.ui.statusbar.showMessage('Extracting information..')
self.url_list.append(url)
self.complete_url_list[row] = url
self.rowcount += 1
if len(self.url_list) is not 0:
if len(self.url_list) < 2:
self.ui.statusbar.showMessage('Downloading {0} file'.format(len(self.url_list)))
else:
self.ui.statusbar.showMessage('Downloading {0} files'.format(len(self.url_list)))
else:
self.ui.statusbar.showMessage("done")
def handleButton(self):
url = str(self.ui.videoUrlLineEdit.text())
if url is '':
QtGui.QMessageBox.information(self, "Error!","No url given!")
return
can_download, rowcount = self.can_download(url)
if can_download:
self.download_url(url, rowcount)
else:
QtGui.QMessageBox.information(self, "Error!","This url is already being downloaded")
if len(self.url_list) is not 0:
if len(self.url_list) < 2:
self.ui.statusbar.showMessage('Downloading {0} file'.format(len(self.url_list)))
else:
self.ui.statusbar.showMessage('Downloading {0} files'.format(len(self.url_list)))
else:
self.ui.statusbar.showMessage("done")
def can_download(self,url):
if url not in self.url_list:
for row, _url in self.complete_url_list.iteritems():
if url == _url:
print "url already there:"
print row, self.rowcount
return True, row
print "url not already there:"
print self.rowcount, self.rowcount
return True, self.rowcount
else:
return False, self.rowcount
def remove_url(self,url):
try:
self.url_list.remove(url)
except:
print url
print self.url_list
return
def add_to_table(self, values):
self.ui.tableWidget.setRowCount(self.rowcount)
row = values[0]
m = 0
for key in values[1:]:
new_item = QtGui.QTableWidgetItem(key)
self.ui.tableWidget.setItem(row, m, new_item)
m += 1
def decrease_rowcount(self):
self.rowcount -= 1
def connect_menu_action(self):
self.ui.actionExit.triggered.connect(self.close)
self.ui.actionLicense.triggered.connect(self.showLicense)
self.ui.actionAbout.triggered.connect(self.showAbout)
def showLicense(self):
license = LicenseDialogue(self)
license.show()
def showAbout(self):
about = AboutDialog(self)
about.show()
def closeEvent(self, event):
if len(self.url_list) is not 0:
msgBox = QtGui.QMessageBox(self)
msgBox.setWindowTitle("Exit")
msgBox.setText("Some files are currently being downloaded.")
msgBox.setInformativeText("Do you really want to close?")
msgBox.setStandardButtons(QtGui.QMessageBox.Close | QtGui.QMessageBox.Cancel)
msgBox.setDefaultButton(QtGui.QMessageBox.Cancel)
ret = msgBox.exec_()
if ret == QtGui.QMessageBox.Cancel:
event.ignore()
else:
self.kill_all_threads()
else:
self.kill_all_threads()
def kill_all_threads(self):
for key, value in self.thread_pool.iteritems():
if value.done is False:
del value
else:
continue
self.close()
class MainWindow(QtWidgets.QMainWindow):
def __init__(self, parent=None, *args, **kwargs):
super(MainWindow, self).__init__(parent, *args, **kwargs)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
# For getting the icon to work
try:
from PyQt5.QtWinExtras import QtWin
myappid = "my_company.my_product.sub_product.version"
QtWin.setCurrentProcessExplicitAppUserModelID(
myappid) # type: ignore
except ImportError:
pass
global app_root
if app_root:
ico_path = str(Path(app_root) / Path("UI/images/icon.ico"))
else:
app_root = ":/"
ico_path = app_root + "icon.ico"
self.setWindowIcon(QtGui.QIcon(ico_path))
self.batch_dialog = BatchAddDialogue(self)
self.ui.saveToLineEdit.setText(desktop_path)
self.ui.BrowseConvertToLineEdit.setText(str(Path().cwd()))
self.ui.BrowseConvertLineEdit.files = []
self.ui.statusbar.showMessage("Ready.")
self.set_connections()
self.url_list = []
self.complete_url_list = {}
self.convert_list = []
self.threadpool = QtCore.QThreadPool()
self.ui.tableWidget.horizontalHeader().setSectionResizeMode(
0, QtWidgets.QHeaderView.Stretch)
self.rowcount = 0
self.connect_menu_action()
self.about = AboutDialog(self)
self.license = LicenseDialogue(self)
self.closing = CloseSignals()
self.show()
def set_connections(self):
self.ui.download_btn.clicked.connect(self.handleButton)
self.ui.browse_btn.clicked.connect(self.set_destination)
self.ui.BatchAdd.clicked.connect(self.batch_file)
self.ui.BrowseConvertButton.clicked.connect(self.convert_file_browse)
self.ui.ConvertMultipleButton.clicked.connect(self.convert_button)
self.ui.BrowseConvertToButton.clicked.connect(
self.browse_convert_destination)
def batch_file(self):
self.batch_dialog.exec_()
if self.batch_dialog.download is True:
urls = list(self.batch_dialog.ui.UrlList.toPlainText().split("\n"))
for url in urls:
if url.strip():
self.download_url(str(url))
def set_destination(self):
file_name: str = QtWidgets.QFileDialog.getExistingDirectory(
self, "Select Directory")
if file_name:
self.ui.saveToLineEdit.setText(str(Path(file_name).resolve()))
def browse_convert_destination(self):
file_name = QtWidgets.QFileDialog.getExistingDirectory(
self, "Select Directory")
if file_name:
self.ui.BrowseConvertToLineEdit.setText(
str(Path(file_name).resolve()))
def convert_button(self):
preferred_format = self.ui.ConvertMultipleComboBox.currentText()
out_path = self.ui.BrowseConvertToLineEdit.text()
delete_temp = self.ui.DeleteFileCheckBox.isChecked()
if len(self.ui.BrowseConvertLineEdit.files) < 1:
QtWidgets.QMessageBox.information(self, "Error!",
"No files given!")
else:
for file_path in self.ui.BrowseConvertLineEdit.files:
self.convert_file(file_path, out_path, preferred_format,
delete_temp)
def convert_file_browse(self):
file_names, _ = QtWidgets.QFileDialog.getOpenFileNames(
self,
"Select files",
filter="Videos (*.mp4 *.ogg *.webm *.flv *.mkv)")
if len(file_names) > 1:
self.ui.BrowseConvertLineEdit.setText(
f"{len(file_names)} Files selected")
elif len(file_names) == 1:
file_name = str(Path(file_names[0]).resolve())
self.ui.BrowseConvertLineEdit.setText(file_name)
self.ui.BrowseConvertLineEdit.files = file_names
def convert_file(self,
file_path,
out_path,
preferred_format,
delete_tmp=False):
if Path(file_path).suffix == preferred_format:
self.ui.statusbar.showMessage(
"The source and destination formats are same")
return
if file_path not in self.convert_list:
options = {
"file_path": file_path,
"preferred_format": preferred_format,
"row_count": self.rowcount,
"delete_tmp": delete_tmp,
"parent": self,
"out_path": out_path,
}
# Using ThreadPool
pprocessorThread = PostProcessor(options)
pprocessorThread.signals.statusBar_Signal.connect(
self.ui.statusbar.showMessage)
pprocessorThread.signals.list_Signal.connect(self.add_to_table)
pprocessorThread.signals.row_Signal.connect(self.decrease_rowcount)
self.threadpool.start(pprocessorThread)
self.ui.tabWidget.setCurrentIndex(2)
self.convert_list.append(file_path)
self.add_to_table([
self.rowcount,
Path(file_path).stem,
"",
"00:00",
"-- KiB/s",
"Converting",
])
self.rowcount += 1
else:
self.ui.statusbar.showMessage("Already Converted")
def download_url(self, url, row=None):
if row is None:
row = self.rowcount
directory = str(Path(self.ui.saveToLineEdit.text()).resolve())
quality = False
if self.ui.ConvertCheckBox.isChecked():
quality = self.ui.ConvertComboBox.currentText()
options = {
"url": url,
"directory": directory,
"rowcount": row,
"proxy": "",
"convert_format": quality,
"parent": self,
}
if not self.ui.DeleteFileCheckBox.isChecked():
options["keep_file"] = True
# Using ThreadPool
downloadThread = Download(options)
downloadThread.signals.status_bar_signal.connect(
self.ui.statusbar.showMessage)
downloadThread.signals.remove_url_signal.connect(self.remove_url)
downloadThread.signals.add_update_list_signal.connect(
self.add_to_table)
downloadThread.signals.remove_row_signal.connect(
self.decrease_rowcount)
self.closing.closed.connect(downloadThread.stop)
self.threadpool.start(downloadThread)
self.ui.tabWidget.setCurrentIndex(2)
self.ui.statusbar.showMessage("Extracting information...")
self.url_list.append(url)
self.complete_url_list[row] = url
self.rowcount += 1
if len(self.url_list):
if len(self.url_list) == 1:
self.ui.statusbar.showMessage("Downloading 1 file")
else:
self.ui.statusbar.showMessage(
f"Downloading {len(self.url_list)} files")
else:
self.ui.statusbar.showMessage("Done!")
def handleButton(self):
url = self.ui.videoUrlLineEdit.text()
if not url:
QtWidgets.QMessageBox.information(self, "Error!", "No url given!")
return
can_download, rowcount = self.can_download(url)
if can_download:
self.download_url(url, rowcount)
else:
QtWidgets.QMessageBox.information(
self, "Error!", "This url is already being downloaded")
if len(self.url_list):
if len(self.url_list) == 1:
self.ui.statusbar.showMessage("Downloading 1 file")
else:
self.ui.statusbar.showMessage(
f"Downloading {len(self.url_list)} files")
else:
self.ui.statusbar.showMessage("Done!")
def can_download(self, url):
if url not in self.url_list:
for row, _url in self.complete_url_list.items():
if url == _url:
return True, row
return True, self.rowcount
else:
return False, self.rowcount
def remove_url(self, url):
# TODO: Need synchronized List/Queue
try:
self.url_list.remove(url)
except:
pass
def add_to_table(self, values):
self.ui.tableWidget.setRowCount(self.rowcount)
row = values[0]
column = 0
for key in values[1:]:
new_item = QtWidgets.QTableWidgetItem(key)
self.ui.tableWidget.setItem(row, column, new_item)
column += 1
def decrease_rowcount(self):
# TODO: Check decrease >= 0
self.rowcount -= 1
def connect_menu_action(self):
self.ui.actionExit.triggered.connect(self.close)
self.ui.actionLicense.triggered.connect(self.showLicense)
self.ui.actionAbout.triggered.connect(self.showAbout)
def showLicense(self):
self.license.show()
def showAbout(self):
self.about.show()
def closeEvent(self, event):
if len(self.url_list):
msgBox = QtWidgets.QMessageBox(self)
msgBox.setWindowTitle("Exit")
msgBox.setText("Some files are currently being downloaded.")
msgBox.setInformativeText("Do you really want to close?")
msgBox.setStandardButtons(QtWidgets.QMessageBox.Close
| QtWidgets.QMessageBox.Cancel)
msgBox.setDefaultButton(QtWidgets.QMessageBox.Cancel)
ret = msgBox.exec_()
if ret == QtWidgets.QMessageBox.Cancel:
event.ignore()
else:
self.kill_all_threads()
else:
self.kill_all_threads()
def kill_all_threads(self):
self.closing.closed.emit()
self.threadpool.waitForDone()
self.close()