def delete_confirmation(parent, _id: str, action: Callable):
msg = QMessageBox()
msg.setText("Are you sure, you want to remove ?")
msg.setStyleSheet('''
background-color: #1d251c;
color: #bdc0bd;
''')
msg.setParent(parent)
msg.setWindowModality(Qt.WindowModality.WindowModal)
style = '''
margin-top: 20px;
border: none;
'''
yes_btn = QPushButton(" Yes")
yes_btn.setIcon(QIcon(full_path("assets/images/icons/trash-can.ico")))
yes_btn.setStyleSheet(style)
no_btn = QPushButton(" Cancel")
no_btn.setStyleSheet(style)
no_btn.setIcon(QIcon(full_path("assets/images/icons/cancel.ico")))
yes_btn.setDefault(True)
msg.addButton(no_btn, msg.ButtonRole.NoRole)
msg.addButton(yes_btn, msg.ButtonRole.YesRole)
msg.buttonClicked.connect(lambda x: action(_id) if x.text().lower().strip() == "yes" else do_nothing())
msg.exec()
def __init__(self):
super().__init__()
self.logger=logging.getLogger()
handler=QLogger(parent=self,update_log_text_signal=self.update_log_text_signal)
self.update_log_text_signal.connect(handler.widget.appendPlainText)
handler.widget.textChanged.connect(handler.scroll_widget_to_bottom)
self.logger.setLevel(logging.INFO)
# Here set the log level you want
handler.setFormatter(logging.Formatter(fmt="%(asctime)s-%(levelname)s-%(message)s",datefmt="%Y-%m-%d %H:%M:%S"))
self.logger.addHandler(handler)
button=QPushButton("&Test")
button.clicked.connect(self.button_handler)
button.setDefault(True)
layout_=QVBoxLayout()
layout_.addWidget(handler.widget)
layout_.addWidget(button)
self.setLayout(layout_)
class UI(QWidget):
update_signal = pyqtSignal(str)
show_qr_signal = pyqtSignal(bytes)
finish_signal = pyqtSignal()
close_qr_signal = pyqtSignal()
def __init__(self):
super().__init__()
self.logger = logging.getLogger(__name__)
formatter = logging.Formatter(
fmt="%(asctime)s-%(levelname)s-%(message)s",
datefmt="%Y-%m-%d %H:%M:%S")
filehandler = logging.FileHandler(filename="logs.log",
mode="w",
encoding="utf-8")
handler = QLogger(update_signal=self.update_signal)
handler.setLevel(logging.INFO)
filehandler.setLevel(logging.INFO)
self.logger.setLevel(logging.INFO)
if os.path.exists("config.json") == False:
self.gen_conf()
with open(file="config.json", mode="r",
encoding="utf-8") as conf_reader:
conf = json.loads(conf_reader.read())
debug = bool(conf["debug"])
if debug == True:
handler.setLevel(logging.DEBUG)
filehandler.setLevel(logging.DEBUG)
self.logger.setLevel(logging.DEBUG)
handler.setFormatter(formatter)
filehandler.setFormatter(formatter)
self.logger.addHandler(handler)
self.logger.addHandler(filehandler)
self.logger.debug("当前调试状态：%s" % debug)
self.resize(1024, 768)
self.setWindowOpacity(0.9)
self.setAttribute(Qt.WidgetAttribute.WA_TranslucentBackground)
self.setWindowFlag(Qt.WindowFlags.FramelessWindowHint)
self.setAutoFillBackground(True)
self.work = Work(show_qr_signal=self.show_qr_signal,
finish_signal=self.finish_signal,
close_qr_signal=self.close_qr_signal)
self.work_thread = QThread()
self.work.moveToThread(self.work_thread)
self.main_layout = QGridLayout()
self.setLayout(self.main_layout)
self.title = QLabel("ChinaUniOnlineGUI")
self.title.setStyleSheet(
"QLabel{border:none;border-radius:5px;background:transparent;color:#9AD3BC;font-size:60px;}"
)
self.title.setAlignment(Qt.Alignment.AlignCenter)
handler.widget.setStyleSheet(
"QPlainTextEdit{font-family:Microsoft YaHei;background:#F3EAC2;border:none;border-radius:5px;}QScrollBar:vertical,QScrollBar::handle:vertical{background:#F3EAC2;border:none;border-radius:8px;width:16px;}QScrollBar::handle:vertical:hover{background:#F5B461;}QScrollBar::add-page:vertical,QScrollBar::sub-page:vertical{background:#FFFDF9;border:none;border-radius:8px;width:16px;}QScrollBar::down-arrow:vertical,QScrollBar::up-arrow:vertical{background:#F5B461;border:none;border-radius:8px;width:16px;height:16px;}QScrollBar::sub-line:vertical,QScrollBar::add-line:vertical{background:transparent;border:none;}"
)
self.control = QVBoxLayout()
self.control_close = QPushButton()
self.control_close.setToolTip("关闭")
self.control_close.setStyleSheet(
"QPushButton{background:#FFE3ED;border-radius:5px;border:none;}QPushButton:hover{background:#EC524B;}"
)
self.contron_max = QPushButton()
if self.isMaximized() == False:
self.contron_max.setToolTip("最大化")
else:
self.contron_max.setToolTip("还原")
self.contron_max.setStyleSheet(
"QPushButton{background:#FFFDF9;border-radius:5px;border:none;}QPushButton:hover{background:#F5B461;}"
)
self.control_min = QPushButton()
self.control_min.setToolTip("最小化")
self.control_min.setStyleSheet(
"QPushButton{background:#BEEBE9;border-radius:5px;border:none;}QPushButton:hover{background:#F3EAC2;}"
)
self.start_button = QPushButton("开始(&S)")
self.start_button.setStyleSheet(
"QPushButton{background:#9BE3DE;border:none;border-radius:5px;font-size:20px;font-family:DengXian;}QPushButton:hover{background:#9AD3BC;}"
)
self.start_button.setToolTip("开始")
self.start_button.setFixedSize(120, 60)
self.start_button.setDefault(True)
setting_button = QPushButton("设置")
setting_button.setToolTip("设置")
setting_button.setFixedSize(60, 60)
setting_button.setStyleSheet(
"QPushButton{background:#9BE3DE;border:none;border-radius:5px;font-size:20px;font-family:DengXian;}QPushButton:hover{background:#9AD3BC;}"
)
setting_button.clicked.connect(self.setting_callback)
start = QHBoxLayout()
start.addWidget(self.start_button, 2)
start.addWidget(setting_button, 1)
self.control_close.clicked.connect(self.close)
self.control_min.clicked.connect(self.min_callback)
self.contron_max.clicked.connect(self.max_callback)
self.start_button.clicked.connect(self.start_callback)
self.work_thread.started.connect(self.work.start)
self.finish_signal.connect(self.finish_callback)
self.close_qr_signal.connect(self.close_qr)
self.control.addWidget(self.control_min)
self.control.addWidget(self.contron_max)
self.control.addWidget(self.control_close)
self.main_layout.addLayout(self.control, 0, 0)
self.main_layout.addWidget(self.title, 0, 1)
self.main_layout.addLayout(start, 0, 2)
self.main_layout.addWidget(handler.widget, 1, 1)
self.update_signal.connect(handler.widget.appendPlainText)
handler.widget.textChanged.connect(handler.scroll_widget_to_bottom)
self.show_qr_signal.connect(self.show_qr)
self.logger.debug("已初始化UI")
def min_callback(self):
if self.isMinimized() == False:
self.showMinimized()
def max_callback(self):
if self.isMaximized() == False:
self.showMaximized()
self.contron_max.setToolTip("还原")
else:
self.showNormal()
self.contron_max.setToolTip("最大化")
def start_callback(self):
self.start_time = time.time()
self.work_thread.start()
self.start_button.setEnabled(False)
self.start_button.setText("执行中...")
def finish_callback(self):
self.start_button.setEnabled(True)
self.start_button.setText("开始")
passed_time = time.time() - self.start_time
mins, secs = divmod(passed_time, 60)
hours, mins = divmod(mins, 60)
self.logger.info("执行完成，共计用时 {:0>2d}:{:0>2d}:{:0>2d}".format(
int(hours), int(mins), int(secs)))
def show_qr(self, qr: bytes):
title_label = QLabel("请使用微信扫描小程序码完成登陆")
title_label.setStyleSheet(
"QLabel{color:#ffe3ed;border:none;background-color:transparent;border-radius:5px;}"
)
title_label.setAlignment(Qt.Alignment.AlignCenter)
title_label.setFixedHeight(20)
qr_label = QLabel()
pixmap = QPixmap()
pixmap.loadFromData(qr)
qr_label.setPixmap(pixmap)
qr_label.setStyleSheet(
"QLabel{color:#ffe3ed;border:none;background-color:transparent;border-radius:5px;}"
)
layout_ = QVBoxLayout()
layout_.addWidget(title_label, 1)
layout_.addWidget(qr_label, 9)
self.qr_dialog = QWidget(self)
self.qr_dialog.setLayout(layout_)
self.main_layout.addWidget(self.qr_dialog, 1, 1,
Qt.Alignment.AlignCenter)
self.qr_dialog.show()
def close_qr(self):
self.qr_dialog.close()
def setting_callback(self):
setting = SettingWindow(parent=self)
setting.setStyleSheet(
"QDialog{border:none;border-radius:5px;background:#F3EAC2;}")
setting.show()
def gen_conf(self):
default_conf = {
"debug": False,
"hero": {
"title": "英雄篇",
"enabled": True,
"times": 1
},
"revival": {
"title": "复兴篇",
"enabled": True,
"times": 1
},
"creation": {
"title": "创新篇",
"enabled": True,
"times": 1
},
"belief": {
"title": "信念篇",
"enabled": True,
"times": 1
},
"limit_time": {
"title": "限时赛",
"enabled": True,
"times": 1
},
"rob": {
"title": "抢十赛",
"enabled": True,
"times": 1
}
}
with open(file="config.json", mode="w",
encoding="utf-8") as conf_writer:
conf_writer.write(
json.dumps(default_conf,
indent=4,
sort_keys=True,
ensure_ascii=False))
self.logger.info("已生成默认配置文件")
def mousePressEvent(self, event: QMouseEvent):
self.logger.debug("触发鼠标按压事件")
super().mousePressEvent(event)
self.setFocus()
self.m_flag = True
if event.button() == Qt.MouseButtons.LeftButton and self.isMaximized(
) == False and self.hasFocus() == True:
self.old_pos = event.globalPosition() #获取鼠标相对窗口的位置
self.logger.debug("已获取鼠标位置")
self.setCursor(QtGui.QCursor(
Qt.CursorShape.SizeAllCursor)) #更改鼠标图标
def mouseMoveEvent(self, event: QMouseEvent):
self.logger.debug("触发鼠标移动事件")
super().mouseMoveEvent(event)
if self.m_flag == True:
delta_x = int(event.globalPosition().x() - self.old_pos.x())
delta_y = int(event.globalPosition().y() - self.old_pos.y())
self.move(self.x() + delta_x, self.y() + delta_y) #更改窗口位置
self.logger.debug("已更改窗口位置")
self.old_pos = event.globalPosition()
def mouseReleaseEvent(self, event: QMouseEvent):
self.logger.debug("触发鼠标释放事件")
super().mouseReleaseEvent(event)
self.m_flag = False
self.setCursor(QtGui.QCursor(Qt.CursorShape.ArrowCursor))
class YTdownloader(QWidget):
def __init__(self):
super().__init__()
# setup some flags
self.isFetching = False
self.isDownloading = False
# default output path
self.outputPath = f'{QDir.homePath()}/videos'
# setup some window specific things
self.setWindowTitle('YouTube Downloader')
self.setWindowIcon(QIcon('assets/yt-icon.ico'))
self.setFixedSize(705, 343)
# parent layout
layout = QVBoxLayout()
layout.setContentsMargins(15, 15, 15, 10)
self.setLayout(layout)
# top bar layout
topBar = QHBoxLayout()
# detail section
detailSec = QHBoxLayout()
metaSec = QVBoxLayout()
# download section
downloadSec = QHBoxLayout()
downloadBtn = QVBoxLayout()
# output path link button
self.outputBtn = QPushButton('📂 Output Path')
self.outputBtn.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.outputBtn.setToolTip(self.outputPath)
self.outputBtn.clicked.connect(self.setOutputPath)
# status bar
self.statusBar = QStatusBar()
# message box
self.message = QMessageBox()
# setting up widgets
self.urlBox = QLineEdit()
self.urlBox.setFocusPolicy(Qt.FocusPolicy.ClickFocus or Qt.FocusPolicy.NoFocus)
self.urlBox.setPlaceholderText('🔍 Enter or paste video URL...')
self.button = QPushButton('Get')
self.button.setDefault(True)
self.button.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.button.clicked.connect(self.getDetails)
# thumbnail
pixmap = QPixmap('assets\placeholder.jpg')
self.thumb = QLabel()
self.thumb.setFixedSize(250, 141)
self.thumb.setScaledContents(True)
self.thumb.setPixmap(pixmap)
# detail widgets
self.title = QLabel('Title: ')
self.author = QLabel('Author: ')
self.length = QLabel('Duration: ')
self.publish_date = QLabel('Published: ')
# progress bar
self.progress_bar = QProgressBar()
# download options
self.download = QComboBox()
self.download.setPlaceholderText('Download Video')
self.download.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.download.activated.connect(lambda: self.getContent(0))
self.download.setEnabled(False)
# download audio button
self.download_audio = QPushButton('Download Audio')
self.download_audio.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.download_audio.clicked.connect(lambda: self.getContent(1))
self.download_audio.setEnabled(False)
# add widgets and layouts
topBar.addWidget(self.urlBox)
topBar.addWidget(self.button)
# detail section
metaSec.addWidget(self.title)
metaSec.addWidget(self.author)
metaSec.addWidget(self.length)
metaSec.addWidget(self.publish_date)
detailSec.addWidget(self.thumb)
detailSec.addSpacing(20)
detailSec.addLayout(metaSec)
# download section
downloadBtn.addWidget(self.download)
downloadBtn.addWidget(self.download_audio)
downloadSec.addWidget(self.progress_bar)
downloadSec.addSpacing(10)
downloadSec.addLayout(downloadBtn)
# status bar
self.statusBar.setSizeGripEnabled(False)
self.statusBar.addPermanentWidget(self.outputBtn)
# add content to parent layout
layout.addLayout(topBar)
layout.addSpacing(20)
layout.addLayout(detailSec)
layout.addSpacing(5)
layout.addLayout(downloadSec)
layout.addWidget(self.statusBar)
# setup a connection thread to keep checking internet connectivity
self.connection = ConnectionThread()
self.connection.start()
# catch the connection response signal
self.connection.con_response.connect(self.connection_slot)
# connection slot
def connection_slot(self, status):
curMsg = self.statusBar.currentMessage()
# connection succeeded
if status:
if curMsg == '🔴 Disconnected':
self.statusBar.showMessage('🟢 Connection restored!', 3000)
elif curMsg != '🟢 Connected':
self.statusBar.showMessage('🟢 Connected')
# connection failed
elif curMsg == '🟢 Connected':
self.statusBar.showMessage('🔴 Connection interrupted!', 3000)
elif curMsg != '🔴 Disconnected':
self.statusBar.showMessage('🔴 Disconnected')
# set output path slot
def setOutputPath(self):
# update the output path
path = str(QFileDialog.getExistingDirectory(self, "Select Output Directory"))
if path:
self.outputPath = path
# update tooltip
self.outputBtn.setToolTip(path)
# get button slot
def getDetails(self):
curMsg = self.statusBar.currentMessage()
if curMsg == '🔴 Disconnected' or curMsg == '🔴 Connection interrupted!':
self.message.critical(
self,
'Error',
'Connection failed!\nAre you sure you\'re connected to the internet ? '
)
elif self.button.text() == 'Get':
self.button.setText('Stop')
# indicate progress bar as busy
self.progress_bar.setRange(0, 0)
# set fetching flag
self.isFetching = True
# setup a worker thread to keep UI responsive
self.worker = WorkerThread(self.urlBox.text())
self.worker.start()
# catch the finished signal
self.worker.finished.connect(self.finished_slot)
# catch the response signal
self.worker.worker_response.connect(self.response_slot)
# catch the error signal
self.worker.worker_err_response.connect(self.err_slot)
elif self.button.text() == 'Stop':
if self.isFetching:
# stop worker thread
self.worker.terminate()
# set back the button text
self.button.setText('Get')
elif self.isDownloading:
# stop download thread
self.download_thread.terminate()
# show the warning message
self.message.information(
self,
'Interrupted',
'Download interrupted!\nThe process was aborted while the file was being downloaded... '
)
# reset pogress bar
self.progress_bar.reset()
# download options slot
def getContent(self, id):
if self.isFetching:
# show the warning message
self.message.warning(
self,
'Warning',
'Please wait!\nWait while the details are being fetched... '
)
else:
# disable the download options
self.download.setDisabled(True)
self.download_audio.setDisabled(True)
# set downloading flag
self.isDownloading = True
# set button to stop
self.button.setText('Stop')
# setup download thread
if id == 0:
self.download_thread = DownloadThread(self.yt, self.download.currentText()[:4], self.outputPath)
else:
self.download_thread = DownloadThread(self.yt, 'audio', self.outputPath)
# start the thread
self.download_thread.start()
# catch the finished signal
self.download_thread.finished.connect(self.download_finished_slot)
# catch the response signal
self.download_thread.download_response.connect(self.download_response_slot)
# catch the complete signal
self.download_thread.download_complete.connect(self.download_complete_slot)
# catch the error signal
self.download_thread.download_err.connect(self.download_err_slot)
# finished slot
def finished_slot(self):
# remove progress bar busy indication
self.progress_bar.setRange(0, 100)
# unset fetching flag
self.isFetching = False
# response slot
def response_slot(self, res):
# set back the button text
self.button.setText('Get')
# save the yt object for speeding up download
self.yt = res[0]
# set the actual thumbnail of requested video
self.thumb.setPixmap(res[1])
# slice the title if it is more than the limit
if len(res[2]) > 50:
self.title.setText(f'Title: {res[2][:50]}...')
else:
self.title.setText(f'Title: {res[2]}')
# set leftover details
self.author.setText(f'Author: {res[3]}')
self.length.setText(f'Duration: {timedelta(seconds=res[4])}')
self.publish_date.setText(f'Published: {res[5].strftime("%d/%m/%Y")}')
# clear any previous items if any
self.download.clear()
# add resolutions as items to the download button and enable them
self.download.addItems([item for item in res[6]])
self.download.setDisabled(False)
self.download_audio.setDisabled(False)
# error slot
def err_slot(self):
# show the warning message
self.message.warning(
self,
'Warning',
'Something went wrong!\nProbably a broken link or some restricted content... '
)
# set back the button text
self.button.setText('Get')
# download finished slot
def download_finished_slot(self):
# set back the button text
self.button.setText('Get')
# now enable the download options
self.download.setDisabled(False)
self.download_audio.setDisabled(False)
# unset downloading flag
self.isDownloading = False
# reset pogress bar
self.progress_bar.reset()
# download response slot
def download_response_slot(self, per):
# update progress bar
self.progress_bar.setValue(per)
# adjust the font color to maintain the contrast
if per > 52:
self.progress_bar.setStyleSheet('QProgressBar { color: #fff }')
else:
self.progress_bar.setStyleSheet('QProgressBar { color: #000 }')
# download complete slot
def download_complete_slot(self, location):
# use native separators
location = QDir.toNativeSeparators(location)
# show the success message
if self.message.information(
self,
'Downloaded',
f'Download complete!\nFile was successfully downloaded to :\n{location}\n\nOpen the downloaded file now ?',
QMessageBox.StandardButtons.Open,
QMessageBox.StandardButtons.Cancel
) is QMessageBox.StandardButtons.Open: subprocess.Popen(f'explorer /select,{location}')
# download error slot
def download_err_slot(self):
# show the error message
self.message.critical(
self,
'Error',
'Error!\nSomething unusual happened and was unable to download...'
)
