121 lines
2.2 KiB
C++
121 lines
2.2 KiB
C++
#include "TablesTableModel.h"
|
|
#include "PgDatabaseCatalog.h"
|
|
#include "PgClass.h"
|
|
#include "PgClassContainer.h"
|
|
#include "Pgsql_declare.h"
|
|
|
|
TablesTableModel::TablesTableModel(QObject *parent)
|
|
: BaseTableModel(parent)
|
|
{
|
|
|
|
}
|
|
|
|
void TablesTableModel::setCatalog(std::shared_ptr<const PgDatabaseCatalog> cat)
|
|
{
|
|
beginResetModel();
|
|
m_catalog = cat;
|
|
|
|
// Later afscheiden naar filter functie
|
|
auto classes = cat->classes();
|
|
|
|
// How many?
|
|
int n = 0;
|
|
for (const auto &e : *classes)
|
|
if (e.kind == RelKind::Table) ++n;
|
|
|
|
m_tables.clear();
|
|
m_tables.reserve(n); // reserve space
|
|
for (const auto &e : *classes) {
|
|
if (e.kind == RelKind::Table) {
|
|
m_tables.push_back(e);
|
|
}
|
|
}
|
|
|
|
endResetModel();
|
|
}
|
|
|
|
|
|
QVariant TablesTableModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
QVariant v;
|
|
if (orientation == Qt::Horizontal) {
|
|
if (role == Qt::DisplayRole) {
|
|
switch (section) {
|
|
case NameCol:
|
|
v = tr("Name");
|
|
break;
|
|
case OwnerCol:
|
|
v = tr("Owner");
|
|
break;
|
|
case TablespaceCol:
|
|
v = tr("Tablespace");
|
|
break;
|
|
case OptionsCol:
|
|
v = tr("Options");
|
|
break;
|
|
case AclCol:
|
|
v = tr("ACL");
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return v;
|
|
}
|
|
|
|
// Basic functionality:
|
|
int TablesTableModel::rowCount(const QModelIndex &) const
|
|
{
|
|
return m_tables.size();
|
|
}
|
|
|
|
int TablesTableModel::columnCount(const QModelIndex &) const
|
|
{
|
|
return colCount;
|
|
}
|
|
|
|
Oid TablesTableModel::getType(int column) const
|
|
{
|
|
Oid oid;
|
|
switch (column) {
|
|
case TablespaceCol:
|
|
case OwnerCol:
|
|
case NameCol:
|
|
case OptionsCol:
|
|
case AclCol:
|
|
default:
|
|
oid = Pgsql::VARCHAROID;
|
|
}
|
|
return oid;
|
|
}
|
|
|
|
QVariant TablesTableModel::getData(const QModelIndex &index) const
|
|
{
|
|
QVariant v;
|
|
const auto &t = m_tables[index.row()];
|
|
switch (index.column()) {
|
|
case NameCol:
|
|
v = formatTableName(t);
|
|
break;
|
|
case OwnerCol:
|
|
v = getRoleDisplayString(*m_catalog, t.owner);
|
|
break;
|
|
case TablespaceCol:
|
|
v = getTablespaceDisplayString(*m_catalog, t.tablespace);
|
|
break;
|
|
case OptionsCol:
|
|
v = t.options;
|
|
break;
|
|
case AclCol:
|
|
v = t.acl;
|
|
break;
|
|
}
|
|
|
|
return v;
|
|
}
|
|
|
|
QString TablesTableModel::formatTableName(const PgClass &cls) const
|
|
{
|
|
const char * format = "%2 (%1)";
|
|
QString ns_name = getNamespaceDisplayString(*m_catalog, cls.relnamespace);
|
|
return QString(format).arg(ns_name).arg(cls.name);
|
|
}
|