Talk:Sort and Filter a QSqlQueryModel

Реализация класса QSortFilterSqlQueryModel сортируемой модели не позволяет сортировать таблицы полученные с помощью запросов типа SELECT field1, field2 FROM table1 WHERE field1=? Приведенная ниже реализация позволяет редактировать/устанавливать verticalHeader и сортировать модели от любых запросов.

1. ifndef QSqlQueryModelEx_h
2. define QSqlQueryModelEx_h

1. include <QSqlQueryModel>
2. include <QSqlQuery>
3. include <QMap>

class QSqlQueryModelEx : public QSqlQueryModel {

   Q_OBJECT

public:

   QMap<int, QString> vertical_header;

   QSqlQueryModelEx(QObject *parent = 0):
       QSqlQueryModel(parent) { }

   QVariant QSqlQueryModelEx::headerData(int section, Qt::Orientation orientation, int role) const {
       if(orientation == Qt::Vertical && role == Qt::DisplayRole) {
           return QVariant(vertical_header[section]);
       } else {
           return QSqlQueryModel::headerData(section, orientation, role);
       }
   }

   bool setHeaderData(int section, Qt::Orientation orientation, const QVariant &value, int role = Qt::EditRole) {
       if(QSqlQueryModel::setHeaderData(section, orientation, value, role)) {
           emit QSqlQueryModel::headerDataChanged(orientation, section, section);
           return true;
       } else {
           if (orientation == Qt::Vertical) {
               vertical_header[section] = value.toString();
               emit QSqlQueryModel::headerDataChanged(orientation, section, section);
           }
           return false;
       }
   }

   void setQuery(const QSqlQuery & query) {
       __query__ = query;
       str_filter.clear();
       flag_filter = Qt::MatchStartsWith;
       sort_column = -1;
       sort_order = Qt::AscendingOrder;
       return QSqlQueryModel::setQuery(query);
   }

public slots:

   void select() {
       if (__query__.lastQuery().isEmpty()) return;
       QString str_query = __query__.lastQuery();
       if (!str_filter.isEmpty() && !column_filter.isEmpty()) {
           //...filtration
           //...filtration
           //...filtration
       }
       if (sort_column >= 0) {
           QString str_order;
           if (str_query.contains("order by ")) {
               str_query = str_query.left(str_query.indexOf("order by"));
           } else {
               str_order += " ";
           }
           str_order += "ORDER BY " + QString::number(sort_column+1) + " " + ((sort_order == Qt::AscendingOrder) ? "ASC" : "DESC");
           str_query.append(str_order);
       }
       QSqlQuery temp = QSqlQuery(str_query);
       if (__query__.boundValues().size()) {
           foreach (const QVariant& value, __query__.boundValues()) {
               temp.addBindValue(value);
           }
           temp.exec();
       }
       QSqlQueryModel::setQuery(temp);
   }

   void setSort(int column, Qt::SortOrder order) {
       sort_column = column;
       sort_order = order;
   }

   void sort(int column, Qt::SortOrder order) {
       if (sort_column != column || sort_order != order) {
           setSort(column, order);
           select();
       }
   }

   void setFilterColumn(const QString& string_column) {
       column_filter = string_column;
   }

   void setFilter(const QString& string_filter) {
       str_filter = string_filter;
   }

   void setFilterFlags(const Qt::MatchFlag flags) {
       flag_filter = flags;
   }

   void filter(const QString& string_filter) {
       if (str_filter != string_filter) {
           setFilter(string_filter);
           select();
       }
   }

private:

   QSqlQuery __query__;
   Qt::MatchFlag flag_filter;
   QString str_filter, column_filter;
   int sort_column;
   Qt::SortOrder sort_order;

};

1. endif // QSqlQueryModelEx_h