#include "querywizardwidget.h"#include <QStringList>QueryWizardWidget::QueryWizardWidget(QWidget *parent) : QWidget(parent){ rootCombo = new QComboBox; rootModel = new ESqlQueryModel; // отображение таблиц из комбобокса queryModel = new ESqlQueryModel; treeView = new QTreeView; // --- main --- QVBoxLayout *mainLayout = new QVBoxLayout; mainLayout->addWidget(rootCombo); mainLayout->addWidget(treeView); setLayout(mainLayout); // --- init --- updateData(); // выбор таблицы из комбобокса connect(rootCombo,SIGNAL(currentIndexChanged(QString)),this,SLOT(slot(QString)));}//----------------qww_TreeItem::qww_TreeItem(const QList<QVariant> &data, qww_TreeItem *parent){ parentItem = parent; itemData = data;}qww_TreeItem::~qww_TreeItem(){ qDeleteAll(childItems);}void qww_TreeItem::appendChild(qww_TreeItem *item){ childItems.append(item);}qww_TreeItem *qww_TreeItem::child(int row){ return childItems.value(row);}int qww_TreeItem::childCount() const{ return childItems.count();}int qww_TreeItem::columnCount() const{ return itemData.count();}QVariant qww_TreeItem::data(int column) const{ return itemData.value(column);}qww_TreeItem *qww_TreeItem::parent(){ return parentItem;}int qww_TreeItem::row() const{ if (parentItem) return parentItem->childItems.indexOf(const_cast<qww_TreeItem*>(this)); return 0;}//--------------qww_TreeModel::qww_TreeModel(const QString &data, QObject *parent) : QAbstractItemModel(parent){ QList<QVariant> rootData; //rootData << "Title" << "Summary"; rootItem = new qww_TreeItem(rootData); setupModelData(data.split(QString("\n")), rootItem);}qww_TreeModel::~qww_TreeModel(){ delete rootItem;}int qww_TreeModel::columnCount(const QModelIndex &parent) const{ if (parent.isValid()) return static_cast<qww_TreeItem*>(parent.internalPointer())->columnCount(); else return rootItem->columnCount();}QVariant qww_TreeModel::data(const QModelIndex &index, int role) const{ if (!index.isValid()) return QVariant(); if (role != Qt::DisplayRole) return QVariant(); qww_TreeItem *item = static_cast<qww_TreeItem*>(index.internalPointer()); return item->data(index.column());}Qt::ItemFlags qww_TreeModel::flags(const QModelIndex &index) const{ if (!index.isValid()) return 0; return Qt::ItemIsEnabled | Qt::ItemIsSelectable;}QVariant qww_TreeModel::headerData(int section, Qt::Orientation orientation, int role) const{ if (orientation == Qt::Horizontal && role == Qt::DisplayRole) return rootItem->data(section); return QVariant();}QModelIndex qww_TreeModel::index(int row, int column, const QModelIndex &parent)const{ if (!hasIndex(row, column, parent)) return QModelIndex(); qww_TreeItem *parentItem; if (!parent.isValid()) parentItem = rootItem; else parentItem = static_cast<qww_TreeItem*>(parent.internalPointer()); qww_TreeItem *childItem = parentItem->child(row); if (childItem) return createIndex(row, column, childItem); else return QModelIndex();}QModelIndex qww_TreeModel::parent(const QModelIndex &index) const{ if (!index.isValid()) return QModelIndex(); qww_TreeItem *childItem = static_cast<qww_TreeItem*>(index.internalPointer()); qww_TreeItem *parentItem = childItem->parent(); if (parentItem == rootItem) return QModelIndex(); return createIndex(parentItem->row(), 0, parentItem);}int qww_TreeModel::rowCount(const QModelIndex &parent) const{ qww_TreeItem *parentItem; if (parent.column() > 0) return 0; if (!parent.isValid()) parentItem = rootItem; else parentItem = static_cast<qww_TreeItem*>(parent.internalPointer()); return parentItem->childCount();}void qww_TreeModel::setupModelData(const QStringList &lines, qww_TreeItem *parent){ QList<qww_TreeItem*> parents; QList<int> indentations; parents << parent; indentations << 0; int number = 0; while (number < lines.count()) { int position = 0; while (position < lines[number].length()) { if (lines[number].mid(position, 1) != " ") break; position++; } QString lineData = lines[number].mid(position).trimmed(); if (!lineData.isEmpty()) { // Read the column data from the rest of the line. QStringList columnStrings = lineData.split("\t", QString::SkipEmptyParts); QList<QVariant> columnData; for (int column = 0; column < columnStrings.count(); ++column) columnData << columnStrings[column]; if (position > indentations.last()) { // The last child of the current parent is now the new parent // unless the current parent has no children. if (parents.last()->childCount() > 0) { parents << parents.last()->child(parents.last()->childCount()-1); indentations << position; } } else { while (position < indentations.last() && parents.count() > 0) { parents.pop_back(); indentations.pop_back(); } } // Append a new item to the current parent's list of children. parents.last()->appendChild(new qww_TreeItem(columnData, parents.last())); } number++; }}void QueryWizardWidget::slot(QString str){ QFile file(":/default.txt"); file.open(QIODevice::ReadOnly); qww_TreeModel model(file.readAll()); file.close(); treeView->setModel(&model);}void QueryWizardWidget::updateData(){ rootModel->setQuery("SELECT schemaname||'.'||tablename AS tdesc" " FROM pg_tables" " WHERE NOT(tableowner='postgres')" " ORDER BY tdesc ASC;"); rootCombo->setModel(rootModel); rootCombo->setModelColumn(0);}
qww_TreeModel::qww_TreeModel(const QString &data, QObject *parent) : QAbstractItemModel(parent){ QList<QVariant> rootData; //rootData << "Title" << "Summary"; rootItem = new qww_TreeItem(rootData); setupModelData(data.split(QString("\n")), rootItem);}