Changed station list so that it also fetches arrivals data
[quandoparte] / application / stationschedulemodel.h
1 /*
2
3 Copyright (C) 2011 mikelima
4
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
9
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with this program; see the file COPYING.  If not, write to
17 the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18 Boston, MA 02110-1301, USA.
19
20 */
21
22 #ifndef STATIONSCHEDULEMODEL_H
23 #define STATIONSCHEDULEMODEL_H
24
25 #include "stationscheduleitem.h"
26
27 #include <QObject>
28 #include <QAbstractListModel>
29 #include <QUrl>
30
31 class StationScheduleModel : public QAbstractListModel
32 {
33     Q_OBJECT
34     Q_PROPERTY(QString name READ name WRITE setName NOTIFY nameChanged)
35
36     enum StationRoles {
37         TrainRole = Qt::UserRole +1,
38         DepartureStationRole,
39         DepartureTimeRole,
40         ArrivalStationRole,
41         ArrivalTimeRole,
42         DetailsUrlRole,
43         DelayRole,
44         DelayClassRole
45     };
46
47 public:
48     explicit StationScheduleModel(const QString &name = "", QObject *parent = 0);
49
50     QString &name();
51     void setName(const QString &name);
52
53     int rowCount(const QModelIndex &parent) const;
54     QVariant data(const QModelIndex &index, int role) const;
55
56 signals:
57     void nameChanged();
58
59 public slots:
60     void fetch(const QString &name);
61
62 private slots:
63     void parse(const QByteArray &htmlReply, const QUrl &baseUrl);
64
65 private:
66     QString m_name;
67     QList<StationScheduleItem> m_departureSchedules;
68     QList<StationScheduleItem> m_arrivalSchedules;
69 };
70
71 #endif // STATIONSCHEDULEMODEL_H