Mark items in DbAnalyzerDialog

It's now possible to mark items in DbAnalyzerDialog with a different
color, when there are no covers or actor information available on the
net. I enhanced SmTreeItem to contain a QVariant::foregroundColor, and
made SmTreeModel return it when QModelIndex::data with role
Qt::ForegroundRole is called. Maybe that could be useful for other
things like favorites and such...

1 files changed, 142 insertions, 14 deletions

diff --git a/dbanalyzer.cpp b/dbanalyzer.cpp
index 4de88da..95bf5a9 100644
--- a/dbanalyzer.cpp
+++ b/dbanalyzer.cpp
@@ -15,12 +15,15 @@
 #include <QTableWidget>
 #include <QModelIndex>
 #include <QLabel>
+#include <QColor>
+#include <QMenu>
+#include <QAction>
 
 #include "dbanalyzer.h"
 #include "smtreemodel.h"
 #include "smtreeitem.h"
 
-DbAnalyzerDialog::DbAnalyzerDialog(QWidget *parent, Qt::WindowFlags f) : QDialog(parent, f){
+DbAnalyzerDialog::DbAnalyzerDialog(QWidget *parent, Qt::WindowFlags f) : QDialog(parent, f), mMarkMode(DbAnalyzer::NOMARKS), mCurrentView(0){
     //create tab widget
     mTab = new QTabWidget;
 
@@ -41,10 +44,12 @@ DbAnalyzerDialog::DbAnalyzerDialog(QWidget *parent, Qt::WindowFlags f) : QDialog
     mNoActorsV->setColumnHidden(5, true);
     mNoActorsV->setEditTriggers(QTreeView::NoEditTriggers);
     mNoActorsV->setSelectionBehavior(QAbstractItemView::SelectRows);
-    mNoActorsV->setSelectionMode(QAbstractItemView::SingleSelection);
+    mNoActorsV->setSelectionMode(QAbstractItemView::ExtendedSelection);
     mNoActorsV->setAlternatingRowColors(true);
     connect(mNoActorsV, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(noDataDoubleClicked(QModelIndex)));
     noActorsT->setLayout(noActorsL);
+    mCurrentView = mNoActorsV;
+    mMarkMode = DbAnalyzer::MARKS_ACTORS;
 
     //no covers
     QWidget *noCoversT = new QWidget;
@@ -60,7 +65,7 @@ DbAnalyzerDialog::DbAnalyzerDialog(QWidget *parent, Qt::WindowFlags f) : QDialog
     mNoCoversV->setColumnHidden(5, true);
     mNoCoversV->setEditTriggers(QTreeView::NoEditTriggers);
     mNoCoversV->setSelectionBehavior(QAbstractItemView::SelectRows);
-    mNoCoversV->setSelectionMode(QAbstractItemView::SingleSelection);
+    mNoCoversV->setSelectionMode(QAbstractItemView::ExtendedSelection);
     mNoCoversV->setAlternatingRowColors(true);
     connect(mNoCoversV, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(noDataDoubleClicked(QModelIndex)));
     noCoversT->setLayout(noCoversL);
@@ -102,9 +107,17 @@ DbAnalyzerDialog::DbAnalyzerDialog(QWidget *parent, Qt::WindowFlags f) : QDialog
     connect(mRefresh, SIGNAL(clicked()), this, SLOT(refresh()));
     mClose = new QPushButton(tr("Close"));
     connect(mClose, SIGNAL(clicked()), this, SLOT(accept()));
-    mDelete = new QPushButton(tr("Delete..."));
-    mDelete->setEnabled(false);
-    connect(mDelete, SIGNAL(clicked()), this, SLOT(deleteItems()));
+    mMark = new QPushButton(tr("Mark"));
+
+    //button menu
+    QMenu *markMenu = new QMenu(mMark);
+    QAction *markOkA = new QAction(tr("Set OK"), this);
+    QAction *deleteMarkA = new QAction(tr("Delete mark"), this);
+    markMenu->addAction(markOkA);
+    markMenu->addAction(deleteMarkA);
+    mMark->setMenu(markMenu);
+    connect(markOkA, SIGNAL(triggered()), this, SLOT(setMarks()));
+    connect(deleteMarkA, SIGNAL(triggered()), this, SLOT(deleteMarks()));
 
     //setup dialog
     mTab->addTab(noActorsT, tr("No Actors"));
@@ -112,13 +125,12 @@ DbAnalyzerDialog::DbAnalyzerDialog(QWidget *parent, Qt::WindowFlags f) : QDialog
     mTab->addTab(strayActorsT, tr("Stray actors"));
     mTab->addTab(strayGenresT, tr("Stray genres"));
 
-
     //buttons
     QHBoxLayout *buttonLayout = new QHBoxLayout;
     buttonLayout->addWidget(mCancel);
     buttonLayout->addWidget(mRefresh);
     buttonLayout->addStretch();
-    buttonLayout->addWidget(mDelete);
+    buttonLayout->addWidget(mMark);
     buttonLayout->addWidget(mClose);
 
     //totals
@@ -165,8 +177,8 @@ void DbAnalyzerDialog::analyzerFinished(){
     mCancel->setEnabled(false);
     mRefresh->setEnabled(true);
     mClose->setEnabled(true);
-    populate(mNoActorsV, mNoActorsM, mAnalyzer->noActors());
-    populate(mNoCoversV, mNoCoversM, mAnalyzer->noCovers());
+    populate(mNoActorsV, mNoActorsM, mAnalyzer->noActors(), mAnalyzer->actorMarks());
+    populate(mNoCoversV, mNoCoversM, mAnalyzer->noCovers(), mAnalyzer->coverMarks());
     populate(mStrayActorsV, mStrayActorsM, mAnalyzer->strayActors());
     populate(mStrayGenresV, mStrayGenresM, mAnalyzer->strayGenres());
     QString totalString = QString(tr("<b>Total items: %1</b>"));
@@ -220,11 +232,48 @@ void DbAnalyzerDialog::deleteItems(){
 }
 
 void DbAnalyzerDialog::tabChanged(int index){
-    mDelete->setEnabled(index > 1);
+    switch(index){
+        case 0:
+            mMarkMode = DbAnalyzer::MARKS_ACTORS;
+            mCurrentView = mNoActorsV;
+            break;
+        case 1:
+            mMarkMode = DbAnalyzer::MARKS_COVERS;
+            mCurrentView = mNoCoversV;
+            break;
+        case 2:
+            mMarkMode = DbAnalyzer::NOMARKS;
+            mCurrentView = mStrayActorsV;
+            break;
+        case 3:
+            mMarkMode = DbAnalyzer::NOMARKS;
+            mCurrentView = mStrayGenresV;
+            break;
+        default:
+            mMarkMode = DbAnalyzer::NOMARKS;
+            mCurrentView = 0;
+            break;
+    }
     mTotal->setText(mTotals.at(index));
 }
 
-void DbAnalyzerDialog::populate(QTreeView *view, SmTreeModel *model, const QList<QList<QVariant> > &data){
+void DbAnalyzerDialog::setMarks(){
+    if(mMarkMode != DbAnalyzer::NOMARKS){
+        const QList<int> curIds = currentIds();
+        mAnalyzer->setMarks(curIds, mMarkMode, 1);
+        refresh();
+    }
+}
+
+void DbAnalyzerDialog::deleteMarks(){
+    if(mMarkMode != DbAnalyzer::NOMARKS){
+        const QList<int> curIds = currentIds();
+        mAnalyzer->deleteMarks(curIds, mMarkMode);
+        refresh();
+    }
+}
+
+void DbAnalyzerDialog::populate(QTreeView *view, SmTreeModel *model, const QList<QList<QVariant> > &data, const QHash<int, QVariant> &marks){
     const int columns = data.first().count();
     if(columns == 0){
         return;
@@ -232,12 +281,31 @@ void DbAnalyzerDialog::populate(QTreeView *view, SmTreeModel *model, const QList
     SmTreeItem *root = new SmTreeItem(columns);
     foreach(QList<QVariant> l, data){
         SmTreeItem *child = new SmTreeItem(l, root);
+        if(!marks.isEmpty()){
+            int seriesId = l.at(4).toInt();
+            QColor fgColor = Qt::red;
+            if(marks.contains(seriesId) && marks.value(seriesId).toInt() > 0){
+                fgColor = Qt::green;
+            }
+            child->setForegroundColor(fgColor);
+        }
         root->appendChild(child);
     }
     model->setRoot(root);
     view->resizeColumnToContents(0);
 }
 
+const QList<int> DbAnalyzerDialog::currentIds() const {
+    QTreeView *curView = qobject_cast<QTreeView*>(mCurrentView);
+    Q_ASSERT(curView);
+    QModelIndexList curIdxs = curView->selectionModel()->selectedRows(4);
+    QList<int> retval;
+    foreach(QModelIndex i, curIdxs){
+        retval << i.data().toInt();
+    }
+    return retval;
+}
+
 DbAnalyzer::DbAnalyzer(QObject *parent) : QThread(parent), mCanceled(false), mStatus(Fail) {
     mDb = QSqlDatabase::cloneDatabase(QSqlDatabase::database("treedb"), "analyzerDb");
     mDb.open();
@@ -250,6 +318,12 @@ DbAnalyzer::DbAnalyzer(QObject *parent) : QThread(parent), mCanceled(false), mSt
     mStrayGenresQuery->prepare("SELECT genres.tgenrename, genres.igenres_id, COUNT(seriesparts_genremap.igenres_id) FROM genres LEFT JOIN seriesparts_genremap ON genres.igenres_id = seriesparts_genremap.igenres_id WHERE seriesparts_genremap.igenres_id IS NULL GROUP BY genres.igenres_id, genres.tgenrename ORDER BY genres.tgenrename");
     mNoCoverQuery = new QSqlQuery(mDb);
     mNoCoverQuery->prepare("SELECT series.tseries_name, seriesparts.iseriespart, seriesparts.tsubtitle, series.iseries_id, seriesparts.iseriesparts_id FROM files, series, seriesparts WHERE files.iseriespart_id = seriesparts.iseriesparts_id AND seriesparts.iseries_id = series.iseries_id GROUP BY files.iseriespart_id, series.tseries_name,seriesparts.iseriespart, seriesparts.tsubtitle, seriesparts.iseriesparts_id, series.iseries_id HAVING COUNT(iseriespart_id) = 1 ORDER BY series.tseries_name");
+    mMarksQuery = new QSqlQuery(mDb);
+    mMarksQuery->prepare("SELECT id, mark_id FROM marks where mark_reason = :reason");
+    mSetMarksQuery = new QSqlQuery(mDb);
+    mSetMarksQuery->prepare("INSERT INTO marks (id, mark_reason, mark_id) VALUES(:seriesid, :reason, :mark_id)");
+    mDeleteMarksQuery = new QSqlQuery(mDb);
+    mDeleteMarksQuery->prepare("DELETE FROM marks WHERE id = :id AND mark_reason = :reason");
 }
 
 DbAnalyzer::~DbAnalyzer(){
@@ -257,6 +331,9 @@ DbAnalyzer::~DbAnalyzer(){
     delete mNoCoverQuery;
     delete mStrayActorsQuery;
     delete mStrayGenresQuery;
+    delete mMarksQuery;
+    delete mSetMarksQuery;
+    delete mDeleteMarksQuery;
     mDb.close();
     mDb = QSqlDatabase();
     QSqlDatabase::removeDatabase("analyzerDb");
@@ -267,19 +344,49 @@ void DbAnalyzer::setCancel(bool canceled){
     mCanceled = canceled;
 }
 
+void DbAnalyzer::setMarks(const QList<int> &ids, int reason, int mark_id){
+    mDb.transaction();
+    mSetMarksQuery->bindValue(":reason", reason);
+    mSetMarksQuery->bindValue(":mark_id", mark_id);
+    foreach(int id, ids){
+        mSetMarksQuery->bindValue(":seriesid", id);
+        if(!mSetMarksQuery->exec()){
+            mDb.rollback();
+            return;
+        }
+    }
+    mDb.commit();
+}
+
+void DbAnalyzer::deleteMarks(const QList<int> &ids, int reason){
+    mDb.transaction();
+    mDeleteMarksQuery->bindValue(":reason", reason);
+    foreach(int id, ids){
+        mDeleteMarksQuery->bindValue(":id", id);
+        if(!mDeleteMarksQuery->exec()){
+            mDb.rollback();
+            return;
+        }
+    }
+    mDb.commit();
+}
+
 void DbAnalyzer::run(){
     mNoActorR = noDataCheck(mNoActorQuery);
+    mActorMarks = marks(MARKS_ACTORS);
     mNoCoverR = noDataCheck(mNoCoverQuery);
+    mCoverMarks = marks(MARKS_COVERS);
     mStrayActorR = strayCheck(mStrayActorsQuery);
     mStrayGenresR = strayCheck(mStrayGenresQuery);
 }
 
 const QList<QList<QVariant> > DbAnalyzer::noDataCheck(QSqlQuery *query){
     if(!query->exec()){
-        QMutexLocker m(&mStatusMutex);
-        mStatus = Fail;
+        setStatus(Fail);
         return QList<QList<QVariant> >();
     }
+
+    //read data
     QList<QList<QVariant> > retval;
     while(query->next()){
         if(mCanceled){
@@ -315,3 +422,24 @@ const QList<QList<QVariant> > DbAnalyzer::strayCheck(QSqlQuery *query){
     }
     return retval;
 }
+
+QHash<int, QVariant> DbAnalyzer::marks(int markType){
+    QHash<int, QVariant> retval;
+    if(markType == NOMARKS){
+        return retval;
+    }
+    mMarksQuery->bindValue(":reason", markType);
+    if(!mMarksQuery->exec()){
+        setStatus(Fail);
+        return retval;
+    }
+    while(mMarksQuery->next()){
+        retval.insert(mMarksQuery->value(0).toInt(), mMarksQuery->value(1));
+    }
+    return retval;
+}
+
+void DbAnalyzer::setStatus(int status){
+    QMutexLocker m(&mStatusMutex);
+    mStatus = status;
+}