1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/*
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version
2 of the License, or (at your option) any later version.
*/
#include <QSqlDatabase>
#include <QSqlQuery>
#include "archivebrowsermodel.h"
#include "smtreeitem.h"
#include "helper.h"
ArchiveBrowserModel::ArchiveBrowserModel(const QStringList &headers, QObject *parent) : SmTreeModel(headers, parent), mNumFields(7) {
mDb = QSqlDatabase::database("treedb");
populate();
}
QVariant ArchiveBrowserModel::data(const QModelIndex &index, int role) const {
SmTreeItem *item = itemAt(index);
int col = index.column();
if(role == Qt::DisplayRole && index.column() == 0){
return item->data(Name);
}
if(role == NameRole){
return item->data(Name);
}
if(role == GenericIdRole){
return item->data(GenericId);
}
if(role == NodeTypeRole){
return item->data(NodeType);
}
if(role == TotalSizeRole){
return item->data(TotalSize);
}
if(role == QualityRole){
return item->data(Quality);
}
if(role == FileTypeRole){
return item->data(FileType);
}
if(role == FullPathRole){
return item->data(FullPath);
}
if(role == Qt::ForegroundRole){
if(col == TotalSize){
qint64 s = item->data(TotalSize).toDouble();
QColor retval = Qt::green;
if(s > Q_INT64_C(1024 * 1024 * 1024 * 2)){
retval = Qt::cyan;
}
if(s > Q_INT64_C(1024 * 1024 * 1024 * 4)){
retval = Qt::red;
}
return retval;
}
if(col == Quality){
int q = item->data(Quality).toInt();
QColor retval = Qt::green;
if(q < 8){
retval = Qt::red;
}
return retval;
}
}
if(role == Qt::TextAlignmentRole){
if(col == TotalSize || col == Quality){
int retval = Qt::AlignRight | Qt::AlignVCenter;
return retval;
}
}
return SmTreeModel::data(index, role);
}
Qt::ItemFlags ArchiveBrowserModel::flags(const QModelIndex &index) const{
SmTreeItem *item = itemAt(index);
int nt = item->data(NodeType).toInt();
if(nt == FileNode){
return Qt::ItemIsEnabled;
}
return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
}
void ArchiveBrowserModel::populate(){
SmTreeItem *rootItem = new SmTreeItem(mNumFields);
QSqlQuery localFilesQ(mDb);
localFilesQ.prepare("SELECT tfilename, bisize, sifiletype, siquality, cmd5sum FROM files WHERE iseriespart_id = :sid ORDER BY sifiletype");
QSqlQuery localQ = QSqlQuery("SELECT DISTINCT(series.iseries_id), tseries_name, seriesparts.iseriespart, seriesparts.tsubtitle, seriesparts.iseriesparts_id FROM series LEFT JOIN seriesparts ON series.iseries_id = seriesparts.iseries_id LEFT JOIN files ON seriesparts.iseriesparts_id = files.iseriespart_id WHERE files.sifiletype = 1 AND files.idvd < 1 ORDER BY tseries_name ASC", mDb);
while(localQ.next()){
QList<QVariant> serPartData;
QString name;
if(localQ.value(2).toInt() > 0){
name = QString("%1 %2").arg(localQ.value(1).toString()).arg(QString::number(localQ.value(2).toInt()));
}else{
name = QString("%1 - %2").arg(localQ.value(1).toString()).arg(localQ.value(3).toString());
}
serPartData << name << localQ.value(4) << SeriesPartNode << QVariant() << QVariant() << QVariant() << QString();
SmTreeItem *seriesItem = new SmTreeItem(serPartData, rootItem);
rootItem->appendChild(seriesItem);
localFilesQ.bindValue(":sid", localQ.value(4));
localFilesQ.exec();
qint64 totalSize = 0;
int quality = -1;
while(localFilesQ.next()){
QList<QVariant> fileData;
fileData << localFilesQ.value(0) << QVariant() << FileNode << localFilesQ.value(1) << localFilesQ.value(3) << localFilesQ.value(2) << Helper::createArchivePath(localFilesQ.value(0).toString(), localFilesQ.value(4).toString());
totalSize += localFilesQ.value(1).toDouble();
if(localFilesQ.value(2) == 1){ //this is a movie file, no need for another enum
quality = localFilesQ.value(3).toInt();
}
SmTreeItem *fileItem = new SmTreeItem(fileData, seriesItem);
seriesItem->appendChild(fileItem);
}
seriesItem->setData(Quality, quality);
seriesItem->setData(TotalSize, totalSize);
}
setRoot(rootItem);
}
|
