class GenerateMazeGroupView(QWidget):
onMazeSpecChosen = pyqtSignal(MazeGenerationSpecification)
__simplyConnectedCheckbox: QCheckBox
__mazeSizePicker: XYPicker
def __init__(
self,
parent: Optional[QWidget] = None,
*args: Tuple[Any, Any],
**kwargs: Tuple[Any, Any],
) -> None:
"""
Grouped controls box for generating mazes
"""
super(GenerateMazeGroupView, self).__init__(parent=parent,
*args,
**kwargs)
self.setContentsMargins(0, 0, 0, 0)
layout = QVBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
groupbox = QGroupBox("Generate Maze")
layout.addWidget(groupbox)
vbox = QFormLayout()
groupbox.setLayout(vbox)
self.__mazeSizePicker = XYPicker(
minimum=XY(2, 2),
maximum=XY(250, 250),
initialValue=XY(2, 2),
parent=self,
)
self.__simplyConnectedCheckbox = QCheckBox()
self.__simplyConnectedCheckbox.setChecked(True)
generateButton = QPushButton("Generate")
generateButton.clicked.connect(
self.__onGenerateButtonPressed) # type: ignore
vbox.addRow("Size", self.__mazeSizePicker)
vbox.addRow("Simply Connected", self.__simplyConnectedCheckbox)
vbox.addRow(generateButton)
self.setLayout(layout)
def __onGenerateButtonPressed(self) -> None:
self.onMazeSpecChosen.emit(
MazeGenerationSpecification(
self.__mazeSizePicker.getValues(),
self.__simplyConnectedCheckbox.isChecked(),
), ),
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.setFixedSize(800, 410)
self.setWindowTitle("PyLX16A Servo Testing Software")
self.port_selection_box = QComboBox(self)
self.port_selection_box.setFixedSize(200, 27)
self.port_selection_box.move(30, 65)
port_selection_box_label = QLabel("Select Port:", self)
port_selection_box_label.move(30, 35)
self.port_selection_box_refresh_button = QPushButton("Refresh", self)
self.port_selection_box_refresh_button.setFixedSize(60, 23)
self.port_selection_box_refresh_button.move(170, 38)
self.id_selection_box = QListWidget(self)
self.id_selection_box.setFixedSize(200, 200)
self.id_selection_box.move(30, 135)
id_selection_box_label = QLabel("Connected Servos:", self)
id_selection_box_label.setFixedWidth(200)
id_selection_box_label.move(30, 105)
self.id_selection_box_refresh_button = QPushButton("Refresh", self)
self.id_selection_box_refresh_button.setFixedSize(60, 23)
self.id_selection_box_refresh_button.move(170, 108)
self.set_id_line_edit = QLineEdit(self)
self.set_id_line_edit.setFixedSize(50, 27)
self.set_id_line_edit.move(80, 355)
set_id_line_edit_label = QLabel("Set ID:", self)
set_id_line_edit_label.move(30, 355)
set_id_line_edit_label.setFixedSize(50, 27)
self.set_id_button = QPushButton("Change ID!", self)
self.set_id_button.setFixedSize(85, 27)
self.set_id_button.move(145, 355)
self.position_slider = QSlider(Qt.Orientation.Horizontal, self)
self.position_slider.setMinimum(0)
self.position_slider.setMaximum(240)
self.position_slider.setFixedWidth(200)
self.position_slider.move(300, 55)
self.position_slider_readout = QLabel("0.00°", self)
self.position_slider_readout.setFixedWidth(50)
self.position_slider_readout.move(450, 30)
self.position_slider_readout.setAlignment(
Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignVCenter)
position_slider_label = QLabel("Angle (degrees):", self)
position_slider_label.move(300, 30)
self.position_offset_slider = QSlider(Qt.Orientation.Horizontal, self)
self.position_offset_slider.setMinimum(-30)
self.position_offset_slider.setMaximum(30)
self.position_offset_slider.setFixedWidth(200)
self.position_offset_slider.move(300, 125)
self.position_offset_slider_readout = QLabel("0.00°", self)
self.position_offset_slider_readout.setFixedWidth(50)
self.position_offset_slider_readout.move(450, 100)
self.position_offset_slider_readout.setAlignment(
Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignVCenter)
position_offset_slider_label = QLabel("Angle offset (degrees):", self)
position_offset_slider_label.setFixedWidth(200)
position_offset_slider_label.move(300, 100)
self.angle_lower_limit_textentry = QLineEdit(self)
self.angle_lower_limit_textentry.setFixedWidth(50)
self.angle_lower_limit_textentry.move(450, 175)
self.angle_lower_limit_textentry.setValidator(
QIntValidator(0, 240, self))
self.angle_upper_limit_textentry = QLineEdit(self)
self.angle_upper_limit_textentry.setFixedWidth(50)
self.angle_upper_limit_textentry.move(450, 210)
self.angle_upper_limit_textentry.setValidator(
QIntValidator(0, 240, self))
self.angle_lower_limit_textentry_label = QLabel(
"Lower Limit (degrees):", self)
self.angle_lower_limit_textentry_label.move(300, 175)
self.angle_lower_limit_textentry_label.setFixedWidth(150)
self.angle_upper_limit_textentry_label = QLabel(
"Upper Limit (degrees):", self)
self.angle_upper_limit_textentry_label.move(300, 210)
self.angle_upper_limit_textentry_label.setFixedWidth(150)
self.vin_lower_limit_textentry = QLineEdit(self)
self.vin_lower_limit_textentry.setFixedWidth(50)
self.vin_lower_limit_textentry.move(450, 265)
self.vin_lower_limit_textentry.setValidator(
QIntValidator(4500, 12000, self))
self.vin_upper_limit_textentry = QLineEdit(self)
self.vin_upper_limit_textentry.setFixedWidth(50)
self.vin_upper_limit_textentry.move(450, 300)
self.vin_upper_limit_textentry.setValidator(
QIntValidator(4500, 12000, self))
self.vin_lower_limit_textentry_label = QLabel(
"Voltage Lower Limit (mV):", self)
self.vin_lower_limit_textentry_label.move(300, 265)
self.vin_lower_limit_textentry_label.setFixedWidth(150)
self.vin_upper_limit_textentry_label = QLabel(
"Voltage Upper Limit (mV):", self)
self.vin_upper_limit_textentry_label.move(300, 300)
self.vin_upper_limit_textentry_label.setFixedWidth(150)
self.temp_limit_textentry = QLineEdit(self)
self.temp_limit_textentry.setFixedWidth(50)
self.temp_limit_textentry.move(450, 355)
self.temp_limit_textentry.setValidator(QIntValidator(50, 100, self))
self.temp_limit_textentry_label = QLabel("Temp Limit (°C):", self)
self.temp_limit_textentry_label.move(300, 355)
self.temp_limit_textentry_label.setFixedWidth(150)
self.servo_mode_radio_button = QRadioButton("Servo Mode", self)
self.servo_mode_radio_button.move(565, 50)
self.motor_mode_radio_button = QRadioButton("Motor Mode", self)
self.motor_mode_radio_button.move(565, 75)
self.motor_speed_slider = QSlider(Qt.Orientation.Horizontal, self)
self.motor_speed_slider.setMinimum(-1000)
self.motor_speed_slider.setMaximum(1000)
self.motor_speed_slider.setFixedWidth(200)
self.motor_speed_slider.move(565, 125)
motor_speed_slider_label = QLabel("Motor Speed:", self)
motor_speed_slider_label.move(565, 100)
self.torque_enabled_checkbox = QCheckBox("Torque Enabled", self)
self.torque_enabled_checkbox.move(565, 175)
self.torque_enabled_checkbox.setFixedWidth(200)
self.led_enabled_checkbox = QCheckBox("LED Enabled", self)
self.led_enabled_checkbox.move(565, 210)
self.led_enabled_checkbox.setFixedWidth(200)
self.led_over_temp_checkbox = QCheckBox("LED Over Temperature", self)
self.led_over_temp_checkbox.move(565, 258)
self.led_over_temp_checkbox.setFixedWidth(200)
self.led_over_voltage_checkbox = QCheckBox("LED Over Voltage", self)
self.led_over_voltage_checkbox.move(565, 283)
self.led_over_voltage_checkbox.setFixedWidth(200)
self.led_rotor_locked_checkbox = QCheckBox("LED Rotor Locked", self)
self.led_rotor_locked_checkbox.move(565, 308)
self.led_rotor_locked_checkbox.setFixedWidth(200)
self.physical_position_readout = QLabel("--°", self)
self.physical_position_readout.move(565, 367)
self.physical_position_readout.setFixedWidth(200)
self.physical_position_readout_label = QLabel("Position", self)
self.physical_position_readout_label.move(565, 347)
self.temperature_readout = QLabel("-- °C", self)
self.temperature_readout.move(635, 367)
self.temperature_readout.setFixedWidth(200)
self.temperature_readout_label = QLabel("Temperature", self)
self.temperature_readout_label.move(635, 347)
self.voltage_readout = QLabel("-- V", self)
self.voltage_readout.move(730, 367)
self.voltage_readout.setFixedWidth(200)
self.voltage_readout_label = QLabel("Voltage", self)
self.voltage_readout_label.move(730, 347)
self.readout_update_timer = QTimer(self)
self.readout_update_timer.timeout.connect(self.update_readouts)
self.readout_update_timer.start(250)
self.active_servo: LX16A = None
self.position_slider.setValue(0)
self.position_offset_slider.setValue(0)
self.motor_speed_slider.setValue(0)
self.id_selection_box_refresh_button.setEnabled(False)
self.disable_widgets()
self.port_selection_box.currentTextChanged.connect(
self.port_selection_box_changed)
self.port_selection_box_refresh_button.clicked.connect(
self.port_refresh_button_clicked)
self.id_selection_box.currentTextChanged.connect(
self.id_selection_box_changed)
self.id_selection_box_refresh_button.clicked.connect(
self.id_refresh_button_clicked)
self.set_id_button.pressed.connect(self.id_updated)
self.position_slider.sliderMoved.connect(self.position_slider_updated)
self.position_offset_slider.sliderMoved.connect(
self.position_offset_slider_updated)
self.angle_lower_limit_textentry.textChanged.connect(
self.angle_lower_limit_updated)
self.angle_upper_limit_textentry.textChanged.connect(
self.angle_upper_limit_updated)
self.vin_lower_limit_textentry.textChanged.connect(
self.vin_lower_limit_updated)
self.vin_upper_limit_textentry.textChanged.connect(
self.vin_upper_limit_updated)
self.temp_limit_textentry.textChanged.connect(self.temp_limit_updated)
self.servo_mode_radio_button.toggled.connect(
self.servo_mode_radio_button_toggled)
self.motor_mode_radio_button.toggled.connect(
self.motor_mode_radio_button_toggled)
self.motor_speed_slider.valueChanged.connect(
self.motor_speed_slider_updated)
self.torque_enabled_checkbox.stateChanged.connect(
self.torque_enabled_checkbox_toggled)
self.led_enabled_checkbox.stateChanged.connect(
self.led_enabled_checkbox_toggled)
self.led_over_temp_checkbox.stateChanged.connect(
self.led_error_triggers_checkbox_toggled)
self.led_over_voltage_checkbox.stateChanged.connect(
self.led_error_triggers_checkbox_toggled)
self.led_rotor_locked_checkbox.stateChanged.connect(
self.led_error_triggers_checkbox_toggled)
self.scan_for_ports()
def disable_widgets(self):
self.set_id_line_edit.setEnabled(False)
self.position_slider.setEnabled(False)
self.position_offset_slider.setEnabled(False)
self.angle_lower_limit_textentry.setEnabled(False)
self.angle_upper_limit_textentry.setEnabled(False)
self.vin_lower_limit_textentry.setEnabled(False)
self.vin_upper_limit_textentry.setEnabled(False)
self.temp_limit_textentry.setEnabled(False)
self.servo_mode_radio_button.setEnabled(False)
self.motor_mode_radio_button.setEnabled(False)
self.motor_speed_slider.setEnabled(False)
self.torque_enabled_checkbox.setEnabled(False)
self.led_enabled_checkbox.setEnabled(False)
self.led_over_temp_checkbox.setEnabled(False)
self.led_over_voltage_checkbox.setEnabled(False)
self.led_rotor_locked_checkbox.setEnabled(False)
def enable_widgets(self):
self.set_id_line_edit.setEnabled(True)
self.position_slider.setEnabled(True)
self.position_offset_slider.setEnabled(True)
self.angle_lower_limit_textentry.setEnabled(True)
self.angle_upper_limit_textentry.setEnabled(True)
self.vin_lower_limit_textentry.setEnabled(True)
self.vin_upper_limit_textentry.setEnabled(True)
self.temp_limit_textentry.setEnabled(True)
self.servo_mode_radio_button.setEnabled(True)
self.motor_mode_radio_button.setEnabled(True)
self.motor_speed_slider.setEnabled(True)
self.torque_enabled_checkbox.setEnabled(True)
self.led_enabled_checkbox.setEnabled(True)
self.led_over_temp_checkbox.setEnabled(True)
self.led_over_voltage_checkbox.setEnabled(True)
self.led_rotor_locked_checkbox.setEnabled(True)
def clear_servo(self):
self.active_servo = None
@catch_disconnection
def set_servo_id(self, id_):
if not id_.isdigit():
return
self.active_servo = LX16A(int(id_))
self.active_servo.enable_torque()
self.position_slider.setValue(
int(self.active_servo.get_physical_angle()))
self.position_slider_readout.setText(
f"{int(self.active_servo.get_physical_angle() * 25 / 6) * 6 / 25:0.2f}°"
)
self.position_offset_slider.setValue(
int(self.active_servo.get_angle_offset()))
self.position_offset_slider_readout.setText(
f"{int(self.active_servo.get_angle_offset() * 25 / 6) * 6 / 25:0.2f}°"
)
self.angle_lower_limit_textentry.setText(
str(int(self.active_servo.get_angle_limits()[0])))
self.angle_upper_limit_textentry.setText(
str(int(self.active_servo.get_angle_limits()[1])))
self.vin_lower_limit_textentry.setText(
str(self.active_servo.get_vin_limits()[0]))
self.vin_upper_limit_textentry.setText(
str(self.active_servo.get_vin_limits()[1]))
self.temp_limit_textentry.setText(
str(self.active_servo.get_temp_limit()))
self.motor_speed_slider.setValue(self.active_servo.get_motor_speed(
) if self.active_servo.is_motor_mode() else 0)
if self.active_servo.is_motor_mode():
self.motor_mode_radio_button.setChecked(True)
else:
self.servo_mode_radio_button.setChecked(True)
self.motor_speed_slider.setEnabled(self.active_servo.is_motor_mode())
self.torque_enabled_checkbox.setChecked(
self.active_servo.is_torque_enabled())
self.led_enabled_checkbox.setChecked(
self.active_servo.is_led_power_on())
self.led_over_temp_checkbox.setChecked(
self.active_servo.get_led_error_triggers()[0])
self.led_over_voltage_checkbox.setChecked(
self.active_servo.get_led_error_triggers()[1])
self.led_rotor_locked_checkbox.setChecked(
self.active_servo.get_led_error_triggers()[2])
@catch_disconnection
def scan_for_servos(self, port):
self.setCursor(Qt.CursorShape.WaitCursor)
LX16A.initialize(port)
self.id_selection_box.clear()
for i in range(0, 254):
try:
servo = LX16A(i)
self.id_selection_box.addItem(str(i))
except:
pass
self.setCursor(Qt.CursorShape.ArrowCursor)
@catch_disconnection
def scan_for_ports(self):
ports = serial.tools.list_ports.comports()
for port in ports:
self.port_selection_box.addItem(port.device)
@catch_disconnection
def update_readouts(self):
if self.active_servo is None:
return
try:
self.physical_position_readout.setText(
f"{self.active_servo.get_physical_angle():0.2f}°")
self.temperature_readout.setText(
f"{self.active_servo.get_temp()} °C")
self.voltage_readout.setText(
f"{self.active_servo.get_vin() / 1000} V")
except (ServoTimeoutError, ServoChecksumError):
pass
@catch_disconnection
def id_updated(self):
new_id = self.set_id_line_edit.text()
try:
servo = LX16A(int(new_id))
except ServoTimeoutError:
# Meaning this ID is not taken
self.active_servo.set_id(int(new_id))
self.id_selection_box.item(
self.id_selection_box.currentRow()).setText(new_id)
return
QMessageBox.warning(None, "Error", "ID already taken")
@catch_disconnection
def position_slider_updated(self, pos):
if float(self.voltage_readout.text()[:-2]) < 5:
QMessageBox.warning(
None,
"Error",
"The voltage going through the servo is too low. Is your battery powered on?",
)
return
self.active_servo.move(pos)
self.position_slider_readout.setText(
f"{int(pos * 25 / 6) * 6 / 25:0.2f}°")
@catch_disconnection
def position_offset_slider_updated(self, pos):
self.active_servo.set_angle_offset(pos)
self.position_offset_slider_readout.setText(
f"{int(pos * 25 / 6) * 6 / 25:0.2f}°")
@catch_disconnection
def angle_lower_limit_updated(self, text):
if (QIntValidator(0, 240, self).validate(text, 0) !=
QIntValidator.State.Acceptable):
return
if int(text) > int(self.angle_upper_limit_textentry.text()):
return
self.active_servo.set_angle_limits(
int(text), int(self.angle_upper_limit_textentry.text()))
@catch_disconnection
def angle_upper_limit_updated(self, text):
if (QIntValidator(0, 240, self).validate(text, 0) !=
QIntValidator.State.Acceptable):
return
if int(text) < int(self.angle_lower_limit_textentry.text()):
return
self.active_servo.set_angle_limits(
int(self.angle_lower_limit_textentry.text()), int(text))
@catch_disconnection
def vin_lower_limit_updated(self, text):
if (QIntValidator(4500, 12000, self).validate(text, 0) !=
QIntValidator.State.Acceptable):
return
if int(text) > int(self.vin_upper_limit_textentry.text()):
return
self.active_servo.set_vin_limits(
int(text), int(self.vin_upper_limit_textentry.text()))
@catch_disconnection
def vin_upper_limit_updated(self, text):
if (QIntValidator(4500, 12000, self).validate(text, 0) !=
QIntValidator.State.Acceptable):
return
if int(text) < int(self.vin_lower_limit_textentry.text()):
return
self.active_servo.set_vin_limits(
int(self.vin_lower_limit_textentry.text()), int(text))
@catch_disconnection
def temp_limit_updated(self, text):
if (QIntValidator(50, 100, self).validate(text, 0) !=
QIntValidator.State.Acceptable):
return
self.active_servo.set_temp_limit(int(text))
@catch_disconnection
def servo_mode_radio_button_toggled(self, checked):
if checked:
self.active_servo.servo_mode()
self.motor_speed_slider.setEnabled(False)
self.position_slider.setEnabled(True)
self.position_offset_slider.setEnabled(True)
else:
self.active_servo.motor_mode(int(self.motor_speed_slider.value()))
self.motor_speed_slider.setEnabled(True)
self.position_slider.setEnabled(False)
self.position_offset_slider.setEnabled(False)
@catch_disconnection
def motor_mode_radio_button_toggled(self, checked):
if checked:
self.active_servo.motor_mode(int(self.motor_speed_slider.value()))
self.motor_speed_slider.setEnabled(True)
self.position_slider.setEnabled(False)
self.position_offset_slider.setEnabled(False)
else:
self.active_servo.servo_mode()
self.motor_speed_slider.setEnabled(False)
self.position_slider.setEnabled(True)
self.position_offset_slider.setEnabled(True)
@catch_disconnection
def motor_speed_slider_updated(self, pos):
self.active_servo.motor_mode(pos)
@catch_disconnection
def torque_enabled_checkbox_toggled(self, checked):
if checked:
self.active_servo.enable_torque()
else:
self.active_servo.disable_torque()
self.position_slider.setEnabled(checked)
self.position_offset_slider.setEnabled(checked)
self.servo_mode_radio_button.setEnabled(checked)
self.motor_mode_radio_button.setEnabled(checked)
self.motor_speed_slider.setEnabled(checked)
@catch_disconnection
def led_enabled_checkbox_toggled(self, checked):
if checked:
self.active_servo.led_power_on()
else:
self.active_servo.led_power_off()
@catch_disconnection
def led_error_triggers_checkbox_toggled(self):
self.active_servo.set_led_error_triggers(
self.led_over_voltage_checkbox.isChecked(),
self.led_over_temp_checkbox.isChecked(),
self.led_rotor_locked_checkbox.isChecked(),
)
@catch_disconnection
def port_refresh_button_clicked(self, value):
self.id_selection_box_refresh_button.setEnabled(False)
self.disable_widgets()
self.port_selection_box.clear()
self.id_selection_box.clear()
self.scan_for_ports()
@catch_disconnection
def id_refresh_button_clicked(self, value):
self.disable_widgets()
self.id_selection_box.clear()
self.scan_for_servos(self.port_selection_box.currentText())
@catch_disconnection
def port_selection_box_changed(self, text):
if text == "":
return
self.id_selection_box_refresh_button.setEnabled(True)
self.disable_widgets()
self.id_selection_box.clear()
self.clear_servo()
self.scan_for_servos(text)
@catch_disconnection
def id_selection_box_changed(self, text):
if text == "":
return
self.enable_widgets()
self.set_servo_id(text)
class VentanaPrincipal(QMainWindow):
def __init__(self):
QMainWindow.__init__(self)
self.initGui()
def initGui(self):
self.setWindowTitle('Pizza & Ingredientes.')
self.setFixedSize(400, 450)
self.lbl_precio_base = QLabel('Precio base Pizza: $15', self)
self.lbl_precio_base.move(50, 50)
self.lbl_precio_base.setFixedWidth(300)
self.lbl_adicionales = QLabel('Ingredientes Adicionales.', self)
self.lbl_adicionales.move(80, 100)
self.lbl_adicionales.setFixedWidth(300)
self.cbx_1 = QCheckBox('Queso Mozzarela. $1', self)
self.cbx_1.move(80, 130)
self.cbx_1.setFixedWidth(250)
self.cbx_1.stateChanged.connect(self.calcular_total)
self.cbx_2 = QCheckBox('Pepperoni. $2', self)
self.cbx_2.move(80, 160)
self.cbx_2.setFixedWidth(250)
self.cbx_2.stateChanged.connect(self.calcular_total)
self.cbx_3 = QCheckBox('Tocineta. $3', self)
self.cbx_3.move(80, 190)
self.cbx_3.setFixedWidth(250)
self.cbx_3.stateChanged.connect(self.calcular_total)
self.lbl_total = QLabel('Valor del pedido:', self)
self.lbl_total.move(50, 250)
self.lbl_total.setFixedWidth(300)
self.lbl_resultado = QLabel(self)
self.lbl_resultado.move(50, 280)
self.lbl_resultado.setFixedSize(300, 100)
def calcular_total(self):
pizza = 15
adicion = 0
if self.cbx_1.isChecked():
adicion += 1
if self.cbx_2.isChecked():
adicion += 2
if self.cbx_3.isChecked():
adicion += 3
total = float(adicion + pizza)
self.lbl_resultado.setText(f'Pizza: USD ${float(pizza)}\nAdicionales: USD ${float(adicion)}\n\nEl Precio Total de la Pizza es de USD ${total}')
class UIComicInfoWidget(QWidget):
load_chapter_list_signa = QtCore.pyqtSignal(ChapterInfo)
load_download_task_signa = QtCore.pyqtSignal(DownloadTask)
def __init__(self, comic_info: ComicInfo, down_v_box_layout: QVBoxLayout):
super().__init__()
self.comic_info = comic_info
self.down_v_box_layout = down_v_box_layout
self.img_label = QLabel(self)
self.img_label.setScaledContents(True)
img = QImage.fromData(comic_info.cover)
w, h = image_resize(comic_info.cover, width=200)
self.img_label.resize(QtCore.QSize(w, h))
self.img_label.setGeometry(10, 10, w, h)
self.img_label.setPixmap(QPixmap.fromImage(img))
# self.img_label.setPixmap(QtGui.QPixmap("/Users/bo/my/tmp/老夫子2/第1卷/1.jpg"))
self.title = QLabel(self)
self.title.setGeometry(220, 10, 100, 40)
title_font = QtGui.QFont()
title_font.setPointSize(16)
title_font.setBold(True)
title_font.setUnderline(True)
self.title.setFont(title_font)
self.title.setText(comic_info.title)
self.title.setWordWrap(True)
info_font = QtGui.QFont()
info_font.setPointSize(14)
# 作者
self.author = QLabel(self)
self.author.setText("作者 : " + comic_info.author)
self.author.setGeometry(220, 50, 150, 40)
self.author.setWordWrap(True)
self.author.setFont(info_font)
# 状态
self.status = QLabel(self)
self.status.setText("更新状态 : " + comic_info.status)
self.status.setGeometry(220, 90, 150, 40)
self.status.setFont(info_font)
# 热度
self.heat = QLabel(self)
self.heat.setText("热度 : " + str(comic_info.heat))
self.heat.setGeometry(220, 130, 150, 40)
self.heat.setFont(info_font)
# 类型
self.tip = QLabel(self)
self.tip.setText("类型 : " + comic_info.tip)
self.tip.setGeometry(220, 170, 150, 40)
self.tip.setWordWrap(True)
self.tip.setFont(info_font)
# web
self.domain = QLabel(self)
self.domain.setText(f"查看原网页 : {comic_info.domain}")
self.domain.setText(f'查看原网页 : <a href="{comic_info.url}">{comic_info.domain}</a>')
self.domain.setGeometry(220, 210, 150, 40)
self.domain.setOpenExternalLinks(True)
self.domain.setFont(info_font)
# 描述
self.describe = QLabel(self)
self.describe.setText(" " + comic_info.describe)
self.describe.setGeometry(10, 320, 350, 330)
self.describe.setWordWrap(True)
# 对齐方式
self.describe.setAlignment(
QtCore.Qt.AlignmentFlag.AlignLeading | QtCore.Qt.AlignmentFlag.AlignLeft | QtCore.Qt.AlignmentFlag.AlignTop)
# 章节列表,创建一个区域
self.searchHBoxLayout = QHBoxLayout()
# self.searchHBoxLayout.addSpacing()
self.searchGroupBox = QGroupBox()
self.searchGroupBox.setLayout(self.searchHBoxLayout)
self.searchScroll = QScrollArea(self)
self.searchScroll.setGeometry(370, 10, 574, 590)
self.searchScroll.setWidget(self.searchGroupBox)
self.searchScroll.setWidgetResizable(True)
# 全选
self.check_all = QCheckBox(self)
self.check_all.setText("全选")
self.check_all.setGeometry(700, 610, 100, 20)
self.check_all.stateChanged.connect(self.check_all_fun)
# 下载
self.down_button = QPushButton(self)
self.down_button.setText("下载")
self.down_button.setGeometry(780, 605, 50, 30)
self.down_button.clicked.connect(self.download_button_click)
self.load_chapter_list_signa.connect(self.load_chapter)
self.load_download_task_signa.connect(self.download_callback)
# 调用对应的service的接口,获取章节列表
constant.SERVICE.chapter(comic_info, self.load_chapter_list_signa.emit)
i = 0
searchVBoxLayout: QVBoxLayout
check_box_list: List[QCheckBox] = []
def check_all_fun(self):
for check_box in self.check_box_list:
check_box.setChecked(self.check_all.isChecked())
def download_callback(self, task: DownloadTask):
widget = DownLoadTaskWidget(task)
self.down_v_box_layout.addWidget(widget)
def download_button_click(self):
flag = False
for check_box in self.check_box_list:
if check_box.isChecked():
constant.SERVICE.parse_image(self.comic_info, check_box.property("chapter_info"),
self.load_download_task_signa.emit)
if not flag:
QMessageBox.information(self, "下载通知", "正在解析选中章节", QMessageBox.StandardButton.Yes)
flag = True
def load_chapter(self, chapter_info: ChapterInfo):
if self.i % 26 == 0:
self.searchVBoxLayout = QVBoxLayout()
self.searchVBoxLayout.setAlignment(QtCore.Qt.AlignmentFlag.AlignTop) # 对齐方式,研究了3个小时 o(╥﹏╥)o
self.searchHBoxLayout.addLayout(self.searchVBoxLayout)
check_box = QCheckBox()
self.check_box_list.append(check_box)
check_box.setText(chapter_info.title)
check_box.setProperty("chapter_info", chapter_info)
task = constant.downloaded_task_map.get(chapter_info.url)
if task and task.status == -1:
check_box.setStyleSheet('color:red')
check_box.setChecked(True)
self.searchVBoxLayout.addWidget(check_box)
self.i += 1
class Window(QDialog):
def __init__(self):
super(Window, self).__init__()
self.groupBox = QGroupBox(
"What is your favorite programming Language ?")
self.groupBox.setFont(QFont("Sanserif", 13))
self.checkBox1 = QCheckBox("Python")
self.checkBox1.setIcon(QIcon(""))
self.checkBox1.setWhatsThis("This is a checkbox of Python")
self.checkBox2 = QCheckBox("C++")
self.checkBox2.setIcon(QIcon(""))
self.checkBox1.setWhatsThis("This is a checkbox of C++")
self.checkBox3 = QCheckBox("Java")
self.checkBox3.setIcon(QIcon(""))
self.checkBox1.setWhatsThis("This is a checkbox of Java")
self.label = QLabel("You Have Selected FootBall")
self.title = " "
self.left = 100
self.top = 200
self.width = 400
self.height = 300
self.InitWindow()
def InitWindow(self):
self.setWindowTitle(self.title)
self.setGeometry(self.left, self.top, self.width, self.height)
self.createCheckBox()
vbox = QVBoxLayout()
vbox.addWidget(self.groupBox)
vbox.addWidget(self.label)
self.setLayout(vbox)
self.show()
def createCheckBox(self):
hboxLayout = QHBoxLayout()
self.checkBox1.toggled.connect(self.onCheckBox_Toggled)
hboxLayout.addWidget(self.checkBox1)
self.checkBox2.toggled.connect(self.onCheckBox_Toggled)
hboxLayout.addWidget(self.checkBox2)
self.checkBox3.toggled.connect(self.onCheckBox_Toggled)
hboxLayout.addWidget(self.checkBox3)
self.groupBox.setLayout(hboxLayout)
def onCheckBox_Toggled(self):
if self.checkBox1.isChecked():
self.label.setText("You Have Select : " + self.checkBox1.text())
if self.checkBox2.isChecked():
self.label.setText("You Have Select : " + self.checkBox2.text())
if self.checkBox3.isChecked():
self.label.setText("You Have Select : " + self.checkBox3.text())
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.setWindowTitle("📸 Random Image Tools V1.2 🔨")
self.image_list = []
# ToDo: Add a "rename?" flag. Like rename image files
# to store the directory path for later use
self.directory_path_name = "/"
# Set up the layouts
layer_one = QHBoxLayout() # Select a folder, selected directory
layer_one_and_a_half = QHBoxLayout() # Selected directory contents
layer_two = QHBoxLayout() # Second line of buttons
layer_two_vertical_one = QVBoxLayout() # Store the first column w/checkbox and "Convert"
layer_two_vertical_two = QVBoxLayout() # Store the second column w/checkbox and "Open in File Browser"
layer_three = QHBoxLayout() # Conversion process state
vertical_layout_parent = QVBoxLayout()
# Parent widget
widget = QWidget()
# Displays selected directory
self.directory_label = QLabel()
self.directory_label.setText("Directory to be worked on will show here ")
self.directory_label.show()
# Displays "Select folder" button
self.select_a_folder_button = QPushButton()
self.select_a_folder_button.setText("Select a folder:")
self.select_a_folder_button.clicked.connect(self.select_folder_prompt)
self.select_a_folder_button.show()
# Displays the image contents of the selected folder
self.image_paths_list_widget = QListWidget()
self.image_paths_list_widget.show()
# Displays button to initiate image conversion
self.convert_to_png_button = QPushButton()
self.convert_to_png_button.setText("Convert to PNG")
self.convert_to_png_button.clicked.connect(self.convert_folder_to_png)
self.convert_to_png_button.show()
# Check boxes for "Create new folder for PNGs" and "Delete original files after converting"
self.create_new_folder_checkbox = QCheckBox()
self.create_new_folder_checkbox.setText("Create new folder to store converted PNG's?")
self.create_new_folder_checkbox.show()
self.delete_original_files_checkbox = QCheckBox()
self.delete_original_files_checkbox.setText("Delete original files after converting them to PNG?")
self.create_new_folder_checkbox.show()
# Displays button to open selected directory in the file browser
self.show_folder_button = QPushButton()
self.show_folder_button.setText("Open selected folder in file browser")
self.show_folder_button.clicked.connect(self.open_folder)
self.show_folder_button.show()
# Displays label when conversion is finished, and the corresponding progress bar
self.conversion_finished_or_error_label = QLabel()
self.conversion_finished_or_error_label.setText("👀 waiting for you to press \"Convert to PNG\" ")
# Put the find folder button and folder selected button together
layer_one.addWidget(self.select_a_folder_button)
layer_one.addWidget(self.directory_label)
# Image paths of selected folder
layer_one_and_a_half.addWidget(self.image_paths_list_widget)
# Put the convert button and open-in-finder button together
layer_two_vertical_one.addWidget(self.convert_to_png_button)
layer_two_vertical_one.addWidget(self.delete_original_files_checkbox)
layer_two.addLayout(layer_two_vertical_one)
layer_two_vertical_two.addWidget(self.show_folder_button)
layer_two_vertical_two.addWidget(self.create_new_folder_checkbox)
layer_two.addLayout(layer_two_vertical_two)
# Label and progress bar
layer_three.addWidget(self.conversion_finished_or_error_label)
layer_three.setAlignment(Qt.AlignmentFlag.AlignHCenter)
# Put the "convert to png" button beneath
vertical_layout_parent.addLayout(layer_one)
vertical_layout_parent.addLayout(layer_one_and_a_half)
vertical_layout_parent.addLayout(layer_two)
vertical_layout_parent.addLayout(layer_three)
widget.setLayout(vertical_layout_parent)
self.setCentralWidget(widget)
# Prompts user to select a folder, stores the given folder path and displays chosen path to user
def select_folder_prompt(self):
# Clear self.image_list and QListWidget to prepare for newly selected folder.
self.image_list.clear()
self.image_paths_list_widget.clear()
# Append a "/" otherwise it will mix the folder name and containing image file together
directory = str(QFileDialog.getExistingDirectory(self, "Select Directory")) + "/"
# Update QLabel to new directory, and store it in self for future use
self.directory_label.setText(directory)
self.directory_path_name = directory
# Update self.image_list field
image_list = self.scan_for_jpg_file_paths()
self.image_list = image_list
# Populated the QListWindow() with the update self.image_list field
self.image_paths_list_widget.addItems(self.image_list)
# Given a path name, will open it in the Folder browser app
def open_folder(self):
subprocess.call(["open", "-R", self.directory_path_name])
# Given the current state of the directory_path_name folder, will scan for image files in that folder
def scan_for_jpg_file_paths(self):
image_list = []
for root, dirs, files in os.walk(self.directory_path_name, topdown=True):
for filename in files:
if '.jpeg' or '.jpg' or '.webp' or '.gif' or '.icns' in filename:
if '.png' not in filename:
absolute_path = self.directory_path_name + filename
# Avoid adding duplicates
if absolute_path not in image_list:
image_list.append(absolute_path)
return image_list
# Given a non-empty folder path, converts all jpg images in it to png.
# Todo: Store png images in a new folder?
# ToDo: Add a "Delete images after converting?"
# ToDo: Add functionality for checkboxes
# Ok so that the QListWidget can update, I'm going to have to get the image_list before calling this function.
# In other words, image_list is a field in the MainWindow subclass, and is update after selecting the folder, and
# called before calling this function.
def convert_folder_to_png(self):
self.conversion_finished_or_error_label.setText("Converting")
# Progress bar depends on independent variable, length of image list = x
if len(self.image_list) > 0:
self.convert_images_to_png()
self.conversion_finished_or_error_label.setText("Conversion finished")
if len(self.image_list) <= 0:
self.conversion_finished_or_error_label.setText("There are no image files in this folder")
# This will be called after a folder has been selected and the "Convert to PNG" button has been pressed
def convert_images_to_png(self):
self.conversion_finished_or_error_label.setText("...")
# Convert images
# ToDo : Put the checkbox logic here
# Check if user wanted images stored in a new folder
# Check if user wanted to delete original images after conversion
# Placeholder for converted_PNG folder
user_wanted_png_in_new_folder = self.create_new_folder_checkbox.isChecked()
user_wanted_unconverted_image_deleted = self.delete_original_files_checkbox.isChecked()
converted_png_folder_name = ""
# Create folder if user wanted converted images stored in a new folder
if user_wanted_png_in_new_folder:
converted_png_folder_name = self.directory_path_name + "Converted PNG Files/"
os.mkdir(converted_png_folder_name)
for image_path in self.image_list:
# Get absolute path
absolute_image_path = os.path.abspath(image_path)
if ".DS_Store" not in absolute_image_path: # Mac adds .DS_Store, just a way to ignore these pesky files
unconverted_image = Image.open(absolute_image_path) # Image object
if user_wanted_png_in_new_folder:
# Get just the image name and extension
name_and_extension_of_image = extract_image_name_and_extension(absolute_image_path)
# Append image name and extension to the converted image folder path
image_path_stored_in_converted_png_folder = converted_png_folder_name + name_and_extension_of_image
# Store in converted image folder path
unconverted_image.save(rename_image_path_to_png(image_path_stored_in_converted_png_folder))
else:
unconverted_image.save(rename_image_path_to_png(absolute_image_path))
if user_wanted_unconverted_image_deleted:
os.remove(absolute_image_path)
