sqltablemodel_example

class Sql_Table_Model(QSqlTableModel):
    def __init__(self, parent=None, db=None):
        QSqlTableModel.__init__(self, parent, db)
        '''Qt::ItemFlags flags(const QModelIndex &index) const override {return QSqlTableModel::flags(index) | Qt::ItemIsDragEnabled;}'''
        self.setTable(db.connectionName())
        self.select()
        self.removeColumn(0) # Remove the ID column
        self.rename_headers()
        self.fetch_all_rows()
        self.setEditStrategy(QSqlTableModel.OnManualSubmit)
        #self.rekordbox_compare = False


    def flags(self, index):
        """Reimplemented to set editable and movable status."""
        flags = (Qt.ItemIsSelectable | Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
        return flags


    def mimeData(self, indexes):
        if not indexes:
            return None

        files = set()
        for idx in indexes:
            row = idx.row()
            fpath = self.index(row, _COL_PATH).data()
            files.add(fpath)
        files = list(files)
        mime_data = QMimeData()
        mime_data.setUrls(files[0])
        mime_data.setText(files[0])

        b = bytearray()
        b.extend(map(ord, files[0]))
        mime_data.setData('text/plain', QByteArray(b))
        return mime_data