from PySide2 import QtCore, QtGui, QtWidgets, QtUiTools
from PySide2.QtCore import Qt, QObject, SIGNAL

from maya import cmds

def load_ui():
    return QtUiTools.QUiLoader().load(ui_path)

from PySide2 import QtCore, QtGui, QtWidgets, QtUiTools
from PySide2.QtCore import Qt, QObject, SIGNAL

from maya import cmds

def load_ui():
    return QtUiTools.QUiLoader().load(ui_path)

class Assembly(QtWidgets.QMainWindow):
    def __init__(self, parent=None):
        super(Assembly, self).__init__(parent=parent)

        self.__init_ui__()

    def __init_ui__(self):
        self.ui = load_ui()

        self.tree_widget = self.ui.tree_widget
        self.tree_widget.setDragDropMode(QtWidgets.QAbstractItemView.InternalMove)
        self.tree_widget.setAcceptDrops(True)

        self.ui.show()

    def dragEnterEvent(self, event):
        if event.mimeData().hasUrls:
            event.accept()
            print "------"
        else:
            print "dragEnter"

    def dropEvent(self, event):
        if event.mimeData().hasUrls:
            print event.mimeData().urls()
            event.setDropAction(QtCore.Qt.MoveAction)
            event.accept()

            print "bomb dropped"
            qPoint = event.pos()
            print qPoint

        else:
            event.ignore()