diff options
Diffstat (limited to 'filestreemodel.cpp')
| -rw-r--r-- | filestreemodel.cpp | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/filestreemodel.cpp b/filestreemodel.cpp index 3dc5dc1..38e86b8 100644 --- a/filestreemodel.cpp +++ b/filestreemodel.cpp @@ -16,6 +16,7 @@ #include "helper.h" FilesTreeModel::FilesTreeModel(QStringList &headers, QObject *parent) : SmTreeModel(headers, parent){ + //database setup mDb = QSqlDatabase::database("treedb"); mUpdateDvdQuery = new QSqlQuery(mDb); mUpdateDvdQuery->prepare("UPDATE files SET idvd = :dvd WHERE ifiles_id = :id"); @@ -27,6 +28,16 @@ FilesTreeModel::FilesTreeModel(QStringList &headers, QObject *parent) : SmTreeMo mFilesQuery->prepare("SELECT tfilename, cmd5sum FROM files WHERE iseriespart_id = :id"); mDeleteFileQuery = new QSqlQuery(mDb); mDeleteFileQuery->prepare("DELETE FROM files WHERE ifiles_id = :id"); + + //file types + mFileTypes.insert(1, "Movie"); + mFileTypes.insert(2, tr("Front cover")); + mFileTypes.insert(3, tr("Back cover")); + mFileTypes.insert(4, tr("General cover")); + + //cover types + mCoverTypes = mFileTypes; + mCoverTypes.remove(1); } FilesTreeModel::~FilesTreeModel(){ @@ -75,7 +86,7 @@ QVariant FilesTreeModel::data(const QModelIndex &index, int role) const{ SmTreeItem *item = static_cast<SmTreeItem*>(index.internalPointer()); if(role == Qt::DisplayRole){ - if(index.column()) + //if(index.column()) if(index.column() == DvdNo){ if(index.data(FileTypeRole) == QVariant()){ return QVariant(); @@ -92,6 +103,12 @@ QVariant FilesTreeModel::data(const QModelIndex &index, int role) const{ return QVariant(); } } + if(index.column() == Quality){ + if(item->data(Quality).toInt() < 1){ + return QVariant(); + } + return item->data(Quality); + } return item->data(index.column()); } if(role == Qt::EditRole){ |