HYTOS

# -*- coding: utf-8 -*-

""" This is library module """

import sys

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from PyQt5.QtSvg import *

from AppDocData import AppDocData, MessageType

class LibraryItemWidget(QWidget):

    COLUMN = 3

    def __init__(self, symbol_tree_widget, parent=None):

        super().__init__(parent)

        try:

            self.setAcceptDrops(True)

            self._symbol_tree_widget = symbol_tree_widget

            self.layout = QGridLayout()

            self.setLayout(self.layout)

            app_doc_data = AppDocData.instance()

            self.favorites = app_doc_data.get_favorite_libraries()

            for idx, symbol in enumerate(self._symbol_tree_widget.symbols):

                matches = [favorite for favorite in self.favorites if symbol.getUid() == favorite['Symbol_UID']]

                if matches:

                    self.layout.addWidget(LibraryItem(symbol), int(idx / LibraryItemWidget.COLUMN),

                                          idx % LibraryItemWidget.COLUMN)

        except Exception as ex:

            from App import App

            message = 'error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename,

                                                           sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

    def remove_library(self, item):

        """remove library from widget and delete it from database"""

        matches = [idx for idx in range(self.layout.count()) if self.layout.itemAt(idx).widget() is item]

        if matches:

            app_doc_data = AppDocData.instance()

            with app_doc_data.project.database.connect() as conn:

                try:

                    cursor = conn.cursor()

                    sql = 'delete from Libraries where Symbol_UID=?'

                    param = (str(item.symbol.getUid()), )

                    cursor.execute(sql, param)

                    conn.commit()

                    layout_item = self.layout.takeAt(matches[0])

                    self.layout.removeWidget(item)

                    item.deleteLater()

                    del item

                    self.layout.removeItem(layout_item)

                    del layout_item

                    self.layout.update()

                except Exception as ex:

                    print('error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename,

                                                               sys.exc_info()[-1].tb_lineno))

    def add_library(self, item):

        """add library to widget and save it"""

        import os

        import uuid

        row = int((self.layout.count()) / LibraryItemWidget.COLUMN)

        col = (self.layout.count()) % LibraryItemWidget.COLUMN

        self.layout.addWidget(item, row, col)

        app_doc_data = AppDocData.instance()

        with app_doc_data.project.database.connect() as conn:

            try:

                cursor = conn.cursor()

                sql = 'insert into Libraries(UID,User,Symbol_UID) values(?,?,?)'

                param = (str(uuid.uuid4()), os.environ['COMPUTERNAME'].upper(), item.symbol.getUid())

                cursor.execute(sql, param)

                conn.commit()

            except Exception as ex:

                print('error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename,

                                                          sys.exc_info()[-1].tb_lineno))

    def dragEnterEvent(self, event: QDragEnterEvent) -> None:

        if issubclass(type(event.source()), QTreeWidget) and event.mimeData().hasFormat('text/plain'):

            # check if symbol is already registered

            for idx in range(self.layout.count()):

                widget = self.layout.itemAt(idx).widget()

                id = f"{event.mimeData().tag.getType()}.{event.mimeData().tag.getName()}"

                if widget.id == id:

                    event.ignore()

                    return

            # up to here

            event.accept()

        else:

            event.ignore()

    def dropEvent(self, event: QDropEvent) -> None:

        if hasattr(event.mimeData(), 'tag'):

            item = LibraryItem(event.mimeData().tag)

            self.add_library(item)

            event.acceptProposedAction()

class LibraryItem(QFrame):

    def __init__(self, symbol):

        QFrame.__init__(self)

        self.id = f"{symbol.getType()}.{symbol.getName()}"

        self._symbol = symbol

        # An Icon and a label below

        icon = QLabel()

        icon.resize(symbol.pixmap.width(), symbol.pixmap.height())

        icon.setPixmap(symbol.pixmap)

        layout = QGridLayout()

        layout.addWidget(icon, 0, 0, Qt.AlignHCenter)

        title = QLabel(symbol.getName())

        font = title.font()

        font.setPixelSize(10)

        title.setFont(font)

        layout.addWidget(title, 1, 0, Qt.AlignTop | Qt.AlignHCenter)

        self.setLayout(layout)

        self.setMaximumSize(96, 96)

    @property

    def symbol(self):

        return self._symbol

    def keyPressEvent(self, event: QKeyEvent) -> None:

        if event.key() == Qt.Key_Delete:

            self.parentWidget().remove_library(self)

            return

        super(LibraryItem, self).keyPressEvent(event)

    """

    Drag and drop management

    """

    def enterEvent(self, event):

        self.setFrameStyle(QFrame.Panel | QFrame.Sunken)

        self.setCursor(Qt.OpenHandCursor)

        self.setFocus()

    def leaveEvent(self, event):

        self.setFrameStyle(QFrame.NoFrame)

        self.setCursor(Qt.ArrowCursor)

    def mousePressEvent(self, mouseEvent):

        self.setCursor(Qt.ClosedHandCursor)

    def mouseReleaseEvent(self, mouseEvent):

        self.setCursor(Qt.OpenHandCursor)

    def mouseMoveEvent(self, event):

        """Drag-n-Drop on left click"""

        if event.buttons() != Qt.LeftButton:

            return

        mime = QMimeData()

        mime.setText(self._symbol.getName())

        drag = QDrag(self)

        drag.setMimeData(mime)

        originalPoint = self._symbol.getOriginalPoint()

        drag.setHotSpot(QPoint(int(float(originalPoint.split(",")[0])), int(float(originalPoint.split(",")[1]))))

        drag.setPixmap(self._symbol.pixmap)

        drag.exec(Qt.CopyAction)