# -*- coding:utf-8 -*-
import sys
import os
from PySide import QtGui, QtCore
RootDirPath = '###任意のディレクトリパス###'

class DraggableView(QtGui.QListView):
    IconSize = QtCore.QSize(128, 128) # アイコンサイズ
    def __init__(self, parent=None):
        super(DraggableView, self).__init__(parent)
        self.setWindowTitle('Draggable File View')
        self.resize(300, 600)
        self.setAlternatingRowColors(True)
        self.setSelectionMode(self.ExtendedSelection)
        self.start_pos = None

        # modelを作成し、アイテムを追加する。==================================
        model = QtGui.QStandardItemModel(0, 1)

        # 変数RootDirPathディレクトリ下のファイルを一覧する。
        for file in os.listdir(RootDirPath):
            item = QtGui.QStandardItem(file)
            # ファイルのフルパスはアイテムのdataの方に確保しておく。
            item.setData(os.path.join(RootDirPath, file)) 
            model.setItem(model.rowCount(), 0, item)

        self.setModel(model)
        # =====================================================================

        # itemSelectionModelを追加する。=======================================
        sel_model = QtGui.QItemSelectionModel(model)
        self.setSelectionModel(sel_model)
        # =====================================================================

    def mousePressEvent(self, event):
        u'''
        マウスをクリックした時の挙動をカスタマイズする。
        '''
        if event.buttons() == QtCore.Qt.RightButton:
            # マウスの右ボタンの時のみドラッグ処理の準備を行う。
            # マウスクリック時の場所を覚えておく。
            self.start_pos = event.pos()
            return
        # 右ボタン以外の場合は親クラスの処理を行う。
        super(DraggableView, self).mousePressEvent(event)

    def startToDrag(self):
        u'''
        ドラッグ操作を開始する。その後ドラッグの結果を返す。
        '''
        # 選択アイテムからファイルパスの一覧を作成する。=======================
        # ついでにQFileIconProviderクラスを使用してアイコンも取得する。
        provider = QtGui.QFileIconProvider()

        filelist = []
        icons = []
        for index in self.selectionModel().selectedIndexes():
            # QModelIndexであるindexから24行目でsetDataしたデータを取得する。
            local_filepath = index.data(QtCore.Qt.UserRole+1)
            icon = provider.icon(local_filepath)
            icons.append(icon.pixmap(self.IconSize))

            # 一度URL形式に変換してからfilelistに追加する。
            filelist.append(QtCore.QUrl.fromLocalFile(local_filepath))
        # =====================================================================

        # ファイルパスのリストをmimeDataにURLとしてセットする。
        mimedata = QtCore.QMimeData()
        mimedata.setUrls(filelist)

        # ドラッグオブジェクトを生成する。=====================================
        drag = QtGui.QDrag(self)
        # 先程作成したmimeDataをセットする。
        drag.setMimeData(mimedata)
        # 先で取得したアイコンの0番目を表示用アイコンとしてセットする。
        drag.setPixmap(icons[0])

        # ドラッグを開始し、その結果を返す。
        return drag.exec_(QtCore.Qt.CopyAction|QtCore.Qt.MoveAction)
        # =====================================================================

    def removeSelectedIndexes(self):
        '''
        選択されたアイテムをビューから削除する。
        '''
        model = self.model()
        indexes = self.selectionModel().selectedIndexes()
        indexes.sort()
        indexes.reverse()
        for index in indexes:
            model.removeRow(index.row())

    def mouseMoveEvent(self, event):
        u'''
        マウスをドラッグしている際の挙動をカスタマイズする。
        '''
        if event.buttons() != QtCore.Qt.RightButton:
            # マウスの右ボタン以外の場合は親クラスの処理を行い終了する。
            super(DraggableView, self).mouseMoveEvent(event)
            return

        if (
            (event.pos() - self.start_pos).manhattanLength()
            < QtGui.qApp.startDragDistance()
        ):
            # マウスをクリックした位置から一定の距離以内の場合は何もしない。
            return

        dragged_result = self.startToDrag()
        if dragged_result == QtCore.Qt.MoveAction:
            # ドラッグの処理が移動だった場合、リストから選択アイテムを削除する。
            self.removeSelectedIndexes()

if __name__ == '__main__':
    app = QtGui.QApplication(sys.argv)
    view = DraggableView()
    view.show()
    
    sys.exit(app.exec_())

スポンサーサイト