class MainUi(QWidget):
def __init__(self):
super(MainUi, self).__init__()
self.setFixedSize(600,500)
self.setWindowTitle("妹子图爬虫工具 version: 1.0.0 ")
self.download_progressbar = QProgressBar()
self.download_progressbar.setAlignment(QtCore.Qt.Alignment.AlignCenter)#文字居中
self.download_progressbar.setStyleSheet(".QProgressBar::chunk { background-color: red;}")#背景
self.download_progressbar.setValue(100)
label01 = QLabel("下载URL:")
label02 = QLabel("下载目录:")
self.url_input = QLineEdit()
self.url_input.setText("https://www.mzitu.com/221746")
self.url_input.setContentsMargins(0,0,0,0)
self.download_dir = QLineEdit()
self.download_dir.setContentsMargins(0,0,0,0)
self.start_btn = QPushButton("开始爬虫")
self.start_btn.setFixedHeight(50)
self.start_btn.setContentsMargins(0,0,0,0)
inputlayout = QGridLayout()
inputlayout.addWidget(label01, 0, 0) #第0行 0列
inputlayout.addWidget(label02, 1, 0)
inputlayout.addWidget(self.url_input, 0, 1)
inputlayout.addWidget(self.download_dir, 1, 1)
inputlayout.addWidget(self.start_btn, 0, 2, 2,1,QtCore.Qt.Alignment.AlignRight) #起始行,起始列, 占行数，占列数
inputlayout.setColumnStretch(0, 1) #设置每一列比例
inputlayout.setColumnStretch(1, 10)
inputlayout.setColumnStretch(2, 1)
vlayout = QVBoxLayout()
vlayout.addLayout(inputlayout)
vlayout.addWidget(self.download_progressbar)
self.frame = QFrame()
self.frame.setFixedHeight(400)
vlayout.addWidget(self.frame)
vlayout.addStretch()
inputlayout.setContentsMargins(0,0,0,0)
vlayout01 = QVBoxLayout()
self.frame.setLayout(vlayout01)
self.qtablewidget = QTableWidget(1,3)
self.qtablewidget.setHorizontalHeaderLabels(['目录','下载图片总数目', '删除'])
vlayout01.addWidget(self.qtablewidget)
self.qtablewidget.setColumnWidth(0, 358) # 将第0列的单元格，设置成300宽度
self.qtablewidget.setColumnWidth(1, 100 ) # 将第0列的单元格，设置成50宽度
self.qtablewidget.verticalHeader().setVisible(False) #隐藏水平表头
#self.qtablewidget.setDisabled(True) #设置不可编辑
self.setLayout(vlayout)
self.current_index = 0
def tableupdate(self,dir, pic_num, pushbtn ) -> None:
self.qtablewidget.setRowCount(self.current_index+1) #设置行
diritem = QTableWidgetItem(str(dir))
self.qtablewidget.setItem( self.current_index, 0, diritem)
pic_numitem = QTableWidgetItem(str(pic_num))
self.qtablewidget.setItem( self.current_index, 1, pic_numitem)
self.qtablewidget.setCellWidget( self.current_index, 2, pushbtn )
self.current_index += 1
class MainWidget(QWidget):
def __init__(self):
super(MainWidget, self).__init__()
self.resize(500, 600)
self.setWindowTitle("喜马拉雅下载 by[Zero] " + __VERSION__)
self.mainlayout = QVBoxLayout()
self.setLayout(self.mainlayout)
self.groupbox = QGroupBox("选择类型")
self.groupbox.setFixedHeight(50)
hlayout = QHBoxLayout(self.groupbox)
self.signal_m4a = QRadioButton("单个下载")
self.mut_m4a = QRadioButton("专辑下载")
self.vip_signal_m4a = QRadioButton("VIP单个下载")
self.vip_m4a = QRadioButton("VIP专辑下载")
hlayout.addWidget(self.signal_m4a)
hlayout.addWidget(self.mut_m4a)
hlayout.addWidget(self.vip_signal_m4a)
hlayout.addWidget(self.vip_m4a)
self.mainlayout.addWidget(self.groupbox)
frame01 = QFrame(self)
child_layout = QVBoxLayout()
print(self.width())
label01 = QLabel("链 接", self)
label02 = QLabel("下载目录", self)
self.url_lineedit = QLineEdit(self)
self.dir_lineedit = QLineEdit(self)
hlayout01 = QHBoxLayout()
hlayout01.addWidget(label01, 1)
hlayout01.addWidget(self.url_lineedit, 9)
hlayout02 = QHBoxLayout()
hlayout02.addWidget(label02, 1)
hlayout02.addWidget(self.dir_lineedit, 9)
child_layout.addLayout(hlayout01)
child_layout.addLayout(hlayout02)
child_layout.setContentsMargins(
5, 0, 5, 0) #(int left， int top， int right， int bottom)
frame01.setLayout(child_layout)
self.download_progressbar = QProgressBar()
self.download_progressbar.setAlignment(
QtCore.Qt.Alignment.AlignCenter) #文字居中
self.download_progressbar.setValue(88)
self.download_btn = QPushButton("开始下载")
self.show_plaintextedit = QPlainTextEdit()
self.show_plaintextedit.setMinimumHeight(400)
self.mainlayout.addWidget(frame01)
self.mainlayout.addWidget(self.download_progressbar)
self.mainlayout.addWidget(self.download_btn)
self.mainlayout.addWidget(self.show_plaintextedit)
self.mainlayout.addStretch()
### 设置stylesheet
self.download_btn.setStyleSheet(
'QPushButton:pressed{ text-align: center;background-color:red;}')
def exerciseLayout(self, index):
hLayout = QHBoxLayout()
setSpinBox = QSpinBox()
setSpinBox.setRange(1, 100)
setSpinBox.setValue(5)
setSpinBox.setMaximumWidth(80)
startButton = QPushButton(self.classifyExercises.exercises[index].name)
startButton.setMaximumWidth(120)
assignedKey = QPushButton(
self.classifyExercises.exercises[index].assigned_key[0])
assignedKey.setStyleSheet('background-color: grey; color: white;')
assignedKey.setMaximumWidth(50)
assignedKey.setEnabled(False)
progress = QProgressBar()
progress.setMaximumWidth(150)
progress.setValue(randint(1, 100))
progress.setAlignment(QtCore.Qt.Alignment.AlignCenter)
latency = QLabel(str(randint(50, 250)) + 'ms')
avgLatency = QLabel(str(randint(50, 250)) + 'ms')
hLayout.addWidget(setSpinBox)
hLayout.setAlignment(setSpinBox, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(startButton)
hLayout.setAlignment(startButton, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(assignedKey)
hLayout.setAlignment(assignedKey, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(progress)
hLayout.setAlignment(progress, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(latency)
hLayout.setAlignment(latency, QtCore.Qt.Alignment.AlignHCenter)
hLayout.addWidget(avgLatency)
hLayout.setAlignment(avgLatency, QtCore.Qt.Alignment.AlignHCenter)
hLayout.setContentsMargins(0, 10, 0, 10)
self.exerciseLayouts.append(hLayout)
return hLayout
class ConfigDialog(QDialog):
def __init__(self, parent=None):
super(ConfigDialog, self).__init__(parent)
self.classifyExercises = parent.classifyExercises
self.setFixedSize(500, 400)
self.setWindowTitle("Model Configurations")
self.epochValue = QLabel()
self.vbox = QVBoxLayout()
self.label_maximum = QLabel()
self.label_minimum = QLabel()
self.slider_hbox = QHBoxLayout()
self.slider_vbox = QVBoxLayout()
self.batchSizeMenu = QComboBox()
self.properties = QFormLayout()
self.epochSlider = Slider(orientation=Qt.Orientations.Horizontal)
self.trainButton = QPushButton('Train Model')
self.resultButton = QPushButton('Show result image')
self.progress = QProgressBar()
self.batchSizeMenu.addItems(['2', '4', '8', '16', '32', '64', '128'])
self.batchSizeMenu.setCurrentIndex(3)
self.batchSizeMenu.setMaximumWidth(100)
self.initSlider()
self.properties.addRow('Batch Size', self.batchSizeMenu)
self.resultButton.setEnabled(False)
self.actionsLayout = QHBoxLayout()
self.actionsLayout.addWidget(self.trainButton)
self.actionsLayout.addWidget(self.resultButton)
self.optionsLayout = QVBoxLayout()
self.optionsLayout.addWidget(QLabel('Model properties'))
self.optionsLayout.addLayout(self.vbox)
self.optionsLayout.addLayout(self.properties)
self.optionsLayout.addLayout(self.actionsLayout)
self.progress.setAlignment(QtCore.Qt.Alignment.AlignCenter)
self.optionsLayout.addWidget(self.progress)
# self.options_layout.addWidget(self.label)
# self.options_layout.addWidget(self.list_widget)
self.setLayout(self.optionsLayout)
self.trainThread = TrainThread(self.classifyExercises)
self.connections()
print("init config")
def initSlider(self):
self.epochValue.setAlignment(QtCore.Qt.Alignment.AlignHCenter)
self.epochSlider.setMinimum(2)
self.epochSlider.setMaximum(10)
self.epochSlider.setTickInterval(1)
# self.epochSlider.setInterval(1)
# self.epochSlider.setValue(8) # no idea why, but 8 is the middle somehow
self.epochSlider.setSliderPosition(6)
self.epochSlider.setTickPosition(QSlider.TickPosition.TicksBelow)
self.label_minimum.setNum(self.epochSlider.minimum().real * 50)
self.label_minimum.setAlignment(QtCore.Qt.Alignment.AlignLeft)
self.label_maximum.setNum(self.epochSlider.maximum().real * 50)
self.label_maximum.setAlignment(QtCore.Qt.Alignment.AlignRight)
self.epochSlider.minimumChanged.connect(self.label_minimum.setNum)
self.epochSlider.maximumChanged.connect(self.label_maximum.setNum)
self.slider_hbox.addWidget(self.label_minimum,
QtCore.Qt.Alignment.AlignLeft)
self.slider_hbox.addWidget(self.epochValue)
self.slider_hbox.addWidget(self.label_maximum,
QtCore.Qt.Alignment.AlignRight)
self.slider_vbox.addWidget(self.epochSlider)
self.slider_vbox.addLayout(self.slider_hbox)
# self.slider_vbox.addStretch()
self.vbox.addLayout(self.slider_vbox)
def connections(self):
self.batchSizeMenu.currentIndexChanged.connect(
self.onBatchSizeSelected)
self.resultButton.clicked.connect(self.onResultClicked)
self.trainButton.clicked.connect(self.onTrainClicked)
self.epochSlider.valueChanged.connect(self.updateEpochValue)
self.trainThread.taskFinished.connect(self.onFinished)
def updateEpochValue(self, num):
print(num)
epochs = num * 50
self.epochValue.setNum(epochs)
self.classifyExercises.epochs = epochs
def onBatchSizeSelected(self, ind):
self.classifyExercises.training_batch_size = int(
self.batchSizeMenu.currentText())
def onResultClicked(self):
print("open image")
self.classifyExercises.DisplayResults()
def onTrainClicked(self):
if self.classifyExercises.subject is not None:
if self.classifyExercises.DataAvailable():
if self.resultButton.isEnabled:
self.resultButton.setEnabled(False)
self.trainThread.start()
self.progress.setRange(0, 0)
else:
CustomMessage.showDialog(
"Message", "Calibrate for patient to obtain data.",
QMessageBox.StandardButtons.Ok)
else:
CustomMessage.showDialog(
"Message", "You must either select or enter a subject name.",
QMessageBox.StandardButtons.Ok)
print("Subject is none!")
def onFinished(self):
# Stop the progress
self.progress.setRange(0, 1)
self.progress.setValue(1)
CustomMessage.showDialog("Message", "Training model finished!",
QMessageBox.StandardButtons.Ok)
self.resultButton.setEnabled(True)