mywidgets.py

from PyQt5 import QtCore, QtWidgets, QtGui
import images_rc


class CaptionWidget(QtWidgets.QWidget):
    def __init__(self, movewindow, parent=None):
        super(CaptionWidget, self).__init__(parent)
        self.setMouseTracking(True)
        self.movewindow = movewindow
        self.mx = 0
        self.my = 0
        self.mousedown = False
        self.caption = ''

    def mousePressEvent(self, event):
        if self.movewindow is None:
            return
        if event.buttons() == QtCore.Qt.LeftButton:
            self.mx = event.globalX()
            self.my = event.globalY()
            self.l = self.movewindow.pos().x()
            self.t = self.movewindow.pos().y()
            self.mousedown = True

    def mouseReleaseEvent(self, event):
        self.mousedown = False

    def mouseMoveEvent(self, event):
        if self.mousedown:
            x = event.globalX()
            y = event.globalY()
            t = y-self.my+self.t
            l = x-self.mx+self.l
            self.movewindow.do_move(l, t)

    def paintEvent(self, event):
        if len(self.caption) == 0:
            super(MyWidget, self).paintEvent(event)
            return
        painter = QtGui.QPainter(self)
        painter.setPen(QtGui.QColor(220, 220, 255))
        painter.drawText(event.rect().adjusted(1, 1, 1, 1), QtCore.Qt.AlignCenter, self.caption)


class VerSizeWidget(QtWidgets.QWidget):

    def __init__(self, parent=None):
        super(VerSizeWidget, self).__init__(parent)
        self.setMouseTracking(True)
        self.parent = parent
        self.mx = 0
        self.my = 0
        self.mousedown = False

    def mousePressEvent(self, event):
        if self.parent is None:
            return
        if event.buttons() == QtCore.Qt.LeftButton:
            self.mx = event.globalX()
            self.my = event.globalY()
            self.h = self.parent.parent().size().height()
            self.w = self.parent.parent().size().width()
            self.mousedown = True

    def mouseReleaseEvent(self, event):
        self.mousedown = False

    def mouseMoveEvent(self, event):
        if self.mousedown:
            x = event.globalX()
            y = event.globalY()
            h = y-self.my+self.h
            self.parent.parent().resize(QtCore.QSize(self.w, h))

class MouseWidget(QtWidgets.QLabel):
    onMousePress = QtCore.pyqtSignal(QtGui.QMouseEvent)
    onMouseRelease = QtCore.pyqtSignal(QtGui.QMouseEvent)
    onMouseMove = QtCore.pyqtSignal(QtGui.QMouseEvent)

    def __init__(self, parent=None):
        super(MouseWidget, self).__init__(parent)
        self.setMouseTracking(True)
        self.md = False
        self.setText('')

    def mousePressEvent(self, event):
        if event.buttons() == QtCore.Qt.LeftButton:
            self.md = True
            self.onMousePress.emit(event)

    def mouseReleaseEvent(self, event):
        if self.md:
            self.md = False
            self.onMouseRelease.emit(event)

    def mouseMoveEvent(self, event):
        if self.md:
            self.onMouseMove.emit(event)


class MyLabel(QtWidgets.QLabel):
    onClick = QtCore.pyqtSignal()

    def __init__(self, styles, parent):
        super(MyLabel, self).__init__(parent)
        self.styles = styles if styles != None else {}
        self.checkable = False
        self.checked = False
        if 'default' in self.styles:
            self.setStyleSheet(self.styles['default'])
        self.leftbtn = False

    def mousePressEvent(self, event):
        if event.buttons() == QtCore.Qt.LeftButton:
            self.leftbtn = True

            if 'pressed' in self.styles:
                self.setStyleSheet(self.styles['pressed'])
            self.raise_()
        else:
            self.leftbtn = False

    def mouseReleaseEvent(self, event):
        if self.leftbtn:
            self.leftbtn = False

            x, y = event.pos().x(), event.pos().y()
            if x > 0 and y > 0 and x < self.width() and y < self.height():
                self.onClick.emit()
                if self.checkable:
                    self.checked = not self.checked

            if self.checked and 'checked' in self.styles:
                self.setStyleSheet(self.styles['checked'])
            elif 'default' in self.styles:
                self.setStyleSheet(self.styles['default'])

    def updatestyle(self):
        style = ''
        if self.leftbtn:
            if 'pressed' in self.styles:
                style = self.styles['pressed']
        else:
            if self.checked and 'checked' in self.styles:
                style = self.styles['checked']
            elif 'default' in self.styles:
                style = self.styles['default']
        self.setStyleSheet(style)

class MyCheckBox(QtWidgets.QLabel):
    toggled = QtCore.pyqtSignal()

    def __init__(self, parent = None):
        super(MyCheckBox,self).__init__(parent)
        self.setMinimumSize(QtCore.QSize(18, 18))
        self.setMaximumSize(QtCore.QSize(18, 18))
        self.setChecked(False)

    def mousePressEvent(self, event):
        if event.buttons() == QtCore.Qt.LeftButton:
            self.setChecked(not self.checked)
            self.toggled.emit()
        super(MyCheckBox, self).mousePressEvent(event)

    def isChecked(self):
        return self.checked

    def setChecked(self, value):
        self.checked = value

        if self.checked:
            self.setStyleSheet('QLabel {background: url(:/images/checked.png) no-repeat;}')
        else:
            self.setStyleSheet('QLabel {background: url(:/images/unchecked.png) no-repeat;}')

class MyComboBox(QtWidgets.QComboBox):
    keyPressed = QtCore.pyqtSignal(QtGui.QKeyEvent)

    def keyPressEvent(self, event):
        if event.key() == 32:
            self.keyPressed.emit(event)
        else:
            super(MyComboBox, self).keyPressEvent(event)

class MyListView(QtWidgets.QListView):
    keyPressed = QtCore.pyqtSignal(QtGui.QKeyEvent)

    def keyPressEvent(self, event):
        k = event.key()
        if k == 32 or k == 16777220:
            self.keyPressed.emit(event)
        else:
            super(MyListView, self).keyPressEvent(event)