Implemented add covers

Finally we can add covers to already existsing series. Seems to work,
but I encountered a crash when merging series while testing...
Unfortunately I don't know the cause.

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){