Drag and drop (needs fix)

Author: rayzchen

drag.ui

@@ -0,0 +1,97 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>Form</class>
+ <widget class="QWidget" name="Form">
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>400</width>
+    <height>300</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>Form</string>
+  </property>
+  <widget class="QTreeWidget" name="treeWidget">
+   <property name="geometry">
+    <rect>
+     <x>40</x>
+     <y>30</y>
+     <width>256</width>
+     <height>192</height>
+    </rect>
+   </property>
+   <property name="frameShadow">
+    <enum>QFrame::Sunken</enum>
+   </property>
+   <property name="lineWidth">
+    <number>1</number>
+   </property>
+   <property name="dragEnabled">
+    <bool>true</bool>
+   </property>
+   <property name="dragDropOverwriteMode">
+    <bool>false</bool>
+   </property>
+   <property name="dragDropMode">
+    <enum>QAbstractItemView::InternalMove</enum>
+   </property>
+   <property name="selectionMode">
+    <enum>QAbstractItemView::ExtendedSelection</enum>
+   </property>
+   <column>
+    <property name="text">
+     <string>1</string>
+    </property>
+   </column>
+   <item>
+    <property name="text">
+     <string>a</string>
+    </property>
+    <item>
+     <property name="text">
+      <string>aa</string>
+     </property>
+    </item>
+    <item>
+     <property name="text">
+      <string>ab</string>
+     </property>
+    </item>
+   </item>
+   <item>
+    <property name="text">
+     <string>b</string>
+    </property>
+    <item>
+     <property name="text">
+      <string>ba</string>
+     </property>
+    </item>
+    <item>
+     <property name="text">
+      <string>bb</string>
+     </property>
+    </item>
+   </item>
+   <item>
+    <property name="text">
+     <string>c</string>
+    </property>
+    <item>
+     <property name="text">
+      <string>ca</string>
+     </property>
+    </item>
+    <item>
+     <property name="text">
+      <string>cb</string>
+     </property>
+    </item>
+   </item>
+  </widget>
+ </widget>
+ <resources/>
+ <connections/>
+</ui>

editor/views.py

@@ -1,13 +1,14 @@
 import os
 import pyunity as pyu
-# from PyQt5.QtCore import QItemSelectionModel, QModelIndex
+from PyQt5.QtCore import QItemSelectionModel, QModelIndex
 from PyQt5.QtCore import Qt
 from PyQt5.QtGui import QIcon
 from PyQt5.QtWidgets import *
 
 class HierarchyItem(QTreeWidgetItem):
     def __init__(self, gameObject):
         super(HierarchyItem, self).__init__()
+        # self.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsDropEnabled | Qt.ItemIsDragEnabled)
         self.setText(0, gameObject.name)
         self.name = gameObject.name
         self.gameObject = gameObject
@@ -155,6 +156,10 @@ def __init__(self, parent):
         super(CustomTreeWidget, self).__init__(parent)
         self.setSelectionMode(QAbstractItemView.ExtendedSelection)
         self.header().setVisible(False)
+        self.setDragEnabled(True)
+        # self.viewport().setAcceptDrops(True)
+        self.setDropIndicatorShown(True)
+        self.setDragDropMode(QAbstractItemView.InternalMove)
         self.setIndentation(10)
 
     def selectAll(self):
@@ -165,9 +170,8 @@ def selectAll(self):
 
     # def mousePressEvent(self, event):
     #     item = self.indexAt(event.pos())
-    #     selected = self.selectionModel().isSelected(item)
     #     super(CustomTreeWidget, self).mousePressEvent(event)
-    #     if (item.row() == -1 and item.column() == -1) or selected:
+    #     if item.row() == -1 and item.column() == -1:
     #         self.clearSelection()
     #         self.selectionModel().setCurrentIndex(QModelIndex(), QItemSelectionModel.Select)