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
|
/*
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.
*/
#ifndef MOVIEMODEL_H
#define MOVIEMODEL_H
#include <QAbstractItemModel>
#include <QVariant>
class QSqlQuery;
class CoverItem;
#include "movieitem.h"
class MovieModel : public QAbstractItemModel {
Q_OBJECT
Q_ENUMS(CustomRoles)
public:
enum CustomRoles { Md5Role = Qt::UserRole + 1};
MovieModel(QObject *parent = 0);
~MovieModel();
QModelIndex index(int row, int column, const QModelIndex &parent) const;
QModelIndex parent(const QModelIndex &) const { return QModelIndex(); };
int columnCount(const QModelIndex &) const { return MovieItem::NumRows; };
int rowCount(const QModelIndex &) const { return mItems.size(); };
QVariant data(const QModelIndex &index, int role) const;
Qt::ItemFlags flags(const QModelIndex &index) const;
QVariant headerData(int section, Qt::Orientation o, int role) const;
bool insertRows(int row, int count, const QModelIndex &);
bool removeRows(int row, int count, const QModelIndex &);
bool setRow(const QModelIndex &idx, const QList<QVariant> &data, const QList<QVariant> &actors, const QList<CoverItem> &covers);
bool setDataAt(const QModelIndex &idx, const QVariant &data);
void setActors(int id, const QList<QVariant> &actors);
void setCovers(int id, const QList<CoverItem> &covers);
void addMovie(const QList<QVariant> &data, const QList<QVariant> &actors, const QList<CoverItem> &covers);
void removeMovie(const QModelIndex &idx);
private:
void populate();
QList<MovieItem*> mItems;
QList<QVariant> mHeaders;
QList<QSqlQuery*> mColumnQueries;
QSqlQuery *mInsertQuery;
QSqlQuery *mDeleteQuery;
QSqlQuery *mDeleteActorsForMovie;
QSqlQuery *mInsertActorsForMovie;
QSqlQuery *mDeleteCovers;
QSqlQuery *mInsertCovers;
};
#endif
|
