from PyQt5.QtWidgets import QPushButton
from PyQt5.QtGui import QPixmap, QIcon
from PyQt5.QtCore import QSize


class DropButton(QPushButton):

    def __init__(self, parent):
        super(DropButton, self).__init__(parent)

        self.presetChooser = None

        self.preset = None
        self.setFixedSize(64, 64)
        self.setIconSize(QSize(64, 64))

    def selectPreset(self):
        self.preset = self.presetChooser.currentPreset().name()
        current_preset = self.presetChooser.currentPreset()
        self.setIcon(QIcon(QPixmap.fromImage(current_preset.image())))