import io import os from typing import Callable from logging import getLogger import PIL from PIL import Image from PyQt5.QtCore import Qt, QCoreApplication, QRect, QSize, QPoint, pyqtSignal from PyQt5.QtGui import QMovie, QPixmap, QFontMetrics from PyQt5.QtWidgets import QLayout, QStyle, QSizePolicy, QLabel, QFileDialog, QHBoxLayout, QWidget, QPushButton, \ QStyleOptionTab, QStylePainter, QTabBar, QLineEdit, QToolButton, QTabWidget from qtawesome import icon from rare import resources_path, cache_dir from rare.utils.qt_requests import QtRequestManager logger = getLogger("ExtraWidgets") class FlowLayout(QLayout): def __init__(self, parent=None, margin=-1, hspacing=-1, vspacing=-1): super(FlowLayout, self).__init__(parent) self._hspacing = hspacing self._vspacing = vspacing self._items = [] self.setContentsMargins(margin, margin, margin, margin) def __del__(self): del self._items[:] def addItem(self, item): self._items.append(item) def horizontalSpacing(self): if self._hspacing >= 0: return self._hspacing else: return self.smartSpacing( QStyle.PM_LayoutHorizontalSpacing) def verticalSpacing(self): if self._vspacing >= 0: return self._vspacing else: return self.smartSpacing( QStyle.PM_LayoutVerticalSpacing) def count(self): return len(self._items) def itemAt(self, index): if 0 <= index < len(self._items): return self._items[index] def takeAt(self, index): if 0 <= index < len(self._items): return self._items.pop(index) def expandingDirections(self): return Qt.Orientations(0) def hasHeightForWidth(self): return True def heightForWidth(self, width): return self.doLayout(QRect(0, 0, width, 0), True) def setGeometry(self, rect): super(FlowLayout, self).setGeometry(rect) self.doLayout(rect, False) def sizeHint(self): return self.minimumSize() def minimumSize(self): size = QSize() for item in self._items: size = size.expandedTo(item.minimumSize()) left, top, right, bottom = self.getContentsMargins() size += QSize(left + right, top + bottom) return size def doLayout(self, rect, testonly): left, top, right, bottom = self.getContentsMargins() effective = rect.adjusted(+left, +top, -right, -bottom) x = effective.x() y = effective.y() lineheight = 0 for item in self._items: widget = item.widget() if not widget.isVisible(): continue hspace = self.horizontalSpacing() if hspace == -1: hspace = widget.style().layoutSpacing( QSizePolicy.PushButton, QSizePolicy.PushButton, Qt.Horizontal) vspace = self.verticalSpacing() if vspace == -1: vspace = widget.style().layoutSpacing( QSizePolicy.PushButton, QSizePolicy.PushButton, Qt.Vertical) nextX = x + item.sizeHint().width() + hspace if nextX - hspace > effective.right() and lineheight > 0: x = effective.x() y = y + lineheight + vspace nextX = x + item.sizeHint().width() + hspace lineheight = 0 if not testonly: item.setGeometry( QRect(QPoint(x, y), item.sizeHint())) x = nextX lineheight = max(lineheight, item.sizeHint().height()) return y + lineheight - rect.y() + bottom def smartSpacing(self, pm): parent = self.parent() if parent is None: return -1 elif parent.isWidgetType(): return parent.style().pixelMetric(pm, None, parent) else: return parent.spacing() class IndicatorLineEdit(QWidget): textChanged = pyqtSignal(str) is_valid = False def __init__(self, text: str = "", ph_text: str = "", edit_func: Callable[[str], tuple[bool, str]] = None, save_func: Callable[[str], None] = None, horiz_policy: QSizePolicy = QSizePolicy.Expanding, parent=None): super(IndicatorLineEdit, self).__init__(parent=parent) self.setObjectName("IndicatorTextEdit") self.layout = QHBoxLayout(self) self.layout.setObjectName("layout") self.layout.setContentsMargins(0, 0, 0, 0) self.line_edit = QLineEdit(self) self.line_edit.setObjectName("line_edit") self.line_edit.setPlaceholderText(ph_text) self.line_edit.setSizePolicy(horiz_policy, QSizePolicy.Fixed) self.layout.addWidget(self.line_edit) if edit_func is not None: self.indicator_label = QLabel() self.indicator_label.setPixmap(icon("ei.info-circle", color="gray").pixmap(16, 16)) self.indicator_label.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed) self.layout.addWidget(self.indicator_label) if not ph_text: _translate = QCoreApplication.translate self.line_edit.setPlaceholderText(_translate("PathEdit", "Default")) if text: self.line_edit.setText(text) self.edit_func = edit_func self.save_func = save_func self.line_edit.textChanged.connect(self.__edit) if self.edit_func is None: self.line_edit.textChanged.connect(self.__save) def text(self) -> str: return self.line_edit.text() def setText(self, text: str): self.line_edit.setText(text) def __indicator(self, res): color = "green" if res else "red" self.indicator_label.setPixmap(icon("ei.info-circle", color=color).pixmap(16, 16)) def __edit(self, text): if self.edit_func is not None: self.line_edit.blockSignals(True) self.is_valid, text = self.edit_func(text) self.line_edit.setText(text) self.line_edit.blockSignals(False) self.__indicator(self.is_valid) self.textChanged.emit(text) if self.is_valid: self.__save(text) def __save(self, text): if self.save_func is not None: self.save_func(text) class PathEdit(IndicatorLineEdit): def __init__(self, text: str = "", file_type: QFileDialog.FileType = QFileDialog.AnyFile, type_filter: str = "", name_filter: str = "", ph_text: str = "", edit_func: Callable[[str], tuple[bool, str]] = None, save_func: Callable[[str], None] = None, horiz_policy: QSizePolicy = QSizePolicy.Expanding, parent=None): super(PathEdit, self).__init__(text=text, ph_text=ph_text, edit_func=edit_func, save_func=save_func, horiz_policy=horiz_policy, parent=parent) self.setObjectName("PathEdit") self.line_edit.setMinimumSize(QSize(300, 0)) self.path_select = QToolButton(self) self.path_select.setObjectName("path_select") self.layout.addWidget(self.path_select) _translate = QCoreApplication.translate self.path_select.setText(_translate("PathEdit", "Browse...")) self.type_filter = type_filter self.name_filter = name_filter self.file_type = file_type self.path_select.clicked.connect(self.__set_path) def __set_path(self): dlg_path = self.line_edit.text() if not dlg_path: dlg_path = os.path.expanduser("~/") dlg = QFileDialog(self, self.tr("Choose path"), dlg_path) dlg.setFileMode(self.file_type) if self.type_filter: dlg.setFilter([self.type_filter]) if self.name_filter: dlg.setNameFilter(self.name_filter) if dlg.exec_(): names = dlg.selectedFiles() self.line_edit.setText(names[0]) class SideTabBar(QTabBar): def __init__(self, parent=None): super(SideTabBar, self).__init__(parent=parent) self.setObjectName("side_tab_bar") self.fm = QFontMetrics(self.font()) def tabSizeHint(self, index): # width = QTabBar.tabSizeHint(self, index).width() return QSize(200, self.fm.height() + 18) def paintEvent(self, event): painter = QStylePainter(self) opt = QStyleOptionTab() for i in range(self.count()): self.initStyleOption(opt, i) painter.drawControl(QStyle.CE_TabBarTabShape, opt) painter.save() s = opt.rect.size() s.transpose() r = QRect(QPoint(), s) r.moveCenter(opt.rect.center()) opt.rect = r c = self.tabRect(i).center() painter.translate(c) painter.rotate(90) painter.translate(-c) painter.drawControl(QStyle.CE_TabBarTabLabel, opt) painter.restore() class WaitingSpinner(QLabel): def __init__(self): super(WaitingSpinner, self).__init__() self.setStyleSheet(""" margin-left: auto; margin-right: auto; """) self.movie = QMovie(os.path.join(resources_path, "images", "loader.gif")) self.setMovie(self.movie) self.movie.start() class SelectViewWidget(QWidget): toggled = pyqtSignal() def __init__(self, icon_view: bool): super(SelectViewWidget, self).__init__() self.icon_view = icon_view self.setStyleSheet("""QPushButton{border: none; background-color: transparent}""") self.icon_view_button = QPushButton() self.list_view = QPushButton() if icon_view: self.icon_view_button.setIcon(icon("mdi.view-grid-outline", color="orange")) self.list_view.setIcon(icon("fa5s.list")) else: self.icon_view_button.setIcon(icon("mdi.view-grid-outline")) self.list_view.setIcon(icon("fa5s.list", color="orange")) self.icon_view_button.clicked.connect(self.icon) self.list_view.clicked.connect(self.list) self.layout = QHBoxLayout() self.layout.addWidget(self.icon_view_button) self.layout.addWidget(self.list_view) self.setLayout(self.layout) def isChecked(self): return self.icon_view def icon(self): self.icon_view_button.setIcon(icon("mdi.view-grid-outline", color="orange")) self.list_view.setIcon(icon("fa5s.list")) self.icon_view = False self.toggled.emit() def list(self): self.icon_view_button.setIcon(icon("mdi.view-grid-outline")) self.list_view.setIcon(icon("fa5s.list", color="orange")) self.icon_view = True self.toggled.emit() class ImageLabel(QLabel): image = None img_size = None name = str() def __init__(self): super(ImageLabel, self).__init__() self.path = cache_dir self.manager = QtRequestManager("bytes") def update_image(self, url, name="", size: tuple = (240, 320)): self.setFixedSize(*size) self.img_size = size self.name = name for c in r'<>?":|\/* ': self.name = self.name.replace(c, "") if self.img_size[0] > self.img_size[1]: name_extension = "wide" else: name_extension = "tall" self.name = f"{self.name}_{name_extension}.png" if not os.path.exists(os.path.join(self.path, self.name)): self.manager.get(url, self.image_ready) # self.request.finished.connect(self.image_ready) else: self.show_image() def image_ready(self, data): self.setPixmap(QPixmap()) try: image: Image.Image = Image.open(io.BytesIO(data)) except PIL.UnidentifiedImageError: return image = image.resize((self.img_size[:2])) byte_array = io.BytesIO() image.save(byte_array, format="PNG") # pixmap = QPixmap.fromImage(ImageQt(image)) pixmap = QPixmap() pixmap.loadFromData(byte_array.getvalue()) # pixmap = QPixmap.fromImage(ImageQt.ImageQt(image)) pixmap = pixmap.scaled(*self.img_size[:2], Qt.KeepAspectRatioByExpanding) self.setPixmap(pixmap) def show_image(self): self.image = QPixmap(os.path.join(self.path, self.name)).scaled(*self.img_size, transformMode=Qt.SmoothTransformation) self.setPixmap(self.image) class ButtonLineEdit(QLineEdit): buttonClicked = pyqtSignal() def __init__(self, icon_name, placeholder_text: str, parent=None): super(ButtonLineEdit, self).__init__(parent) self.button = QToolButton(self) self.button.setIcon(icon(icon_name, color="white")) self.button.setStyleSheet('border: 0px; padding: 0px;') self.button.setCursor(Qt.ArrowCursor) self.button.clicked.connect(self.buttonClicked.emit) self.setPlaceholderText(placeholder_text) frameWidth = self.style().pixelMetric(QStyle.PM_DefaultFrameWidth) buttonSize = self.button.sizeHint() self.setStyleSheet('QLineEdit {padding-right: %dpx; }' % (buttonSize.width() + frameWidth + 1)) self.setMinimumSize(max(self.minimumSizeHint().width(), buttonSize.width() + frameWidth * 2 + 2), max(self.minimumSizeHint().height(), buttonSize.height() + frameWidth * 2 + 2)) def resizeEvent(self, event): buttonSize = self.button.sizeHint() frameWidth = self.style().pixelMetric(QStyle.PM_DefaultFrameWidth) self.button.move(self.rect().right() - frameWidth - buttonSize.width(), (self.rect().bottom() - buttonSize.height() + 1) / 2) super(ButtonLineEdit, self).resizeEvent(event)