git.maemo.org Git - quandoparte/blob - application/stationschedulemodel.cpp

   1 /*
   2
   3 Copyright (C) 2011 Luciano Montanaro <mikelima@cirulla.net>
   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 #include "stationschedulemodel.h"
  23
  24 #include "dataprovider.h"
  25
  26 #include <QDebug>
  27 #include <QWebElement>
  28 #include <QWebFrame>
  29 #include <QWebPage>
  30
  31 StationScheduleModel::StationScheduleModel(const QString &name, QObject *parent) :
  32     QStringListModel(parent),
  33     m_name(name)
  34
  35 {
  36     DataProvider *provider = DataProvider::instance();
  37
  38     connect(provider, SIGNAL(stationScheduleReady(QByteArray,QUrl)),
  39             this, SLOT(parse(QByteArray,QUrl)));
  40 }
  41
  42 QString & StationScheduleModel::name()
  43 {
  44     return m_name;
  45 }
  46
  47 void StationScheduleModel::setName(const QString &name)
  48 {
  49     if (name != m_name) {
  50         m_name = name;
  51         emit nameChanged();
  52     }
  53 }
  54
  55 void StationScheduleModel::parse(const QByteArray &htmlReply, const QUrl &baseUrl)
  56 {
  57     Q_UNUSED(baseUrl);
  58     qDebug() << "--- start of query result --- cut here ------";
  59     qDebug() << QString::fromUtf8(htmlReply.constData());
  60     qDebug() << "--- end of query result ----- cut here ------";
  61
  62     emit layoutAboutToBeChanged();
  63     QWebPage page;
  64     page.mainFrame()->setContent(htmlReply, "text/html", baseUrl);
  65     QWebElement doc = page.mainFrame()->documentElement();
  66
  67     // Find the first div
  68     QWebElement current = doc.findFirst("div");
  69
  70     QStringList departures, arrivals;
  71     qDebug() << "skipping to the departures";
  72     // Skip to the first div of class corpocentrale, which contains the first
  73     // departure-related contents
  74     while (!current.classes().contains("corpocentrale")) {
  75         current = current.nextSibling();
  76         qDebug() << "skipping to the next element";
  77         if (current.isNull())
  78             break;
  79     }
  80     // Mark every div as a departure class element; the next corpocentrale
  81     // marks the start of the arrivals section
  82     qDebug() << "marking departures";
  83     do {
  84         if (current.classes().contains("bloccorisultato")) {
  85             departures << current.toPlainText();
  86         }
  87         current.addClass("departures");
  88         current = current.nextSibling();
  89         qDebug() << "marking as departures";
  90         if (current.isNull())
  91             break;
  92     } while (!current.classes().contains("corpocentrale"));
  93
  94     // Mark everything as an arrival, until reaching the footer
  95     while (!current.classes().contains("footer")) {
  96         if (current.classes().contains("bloccorisultato")) {
  97             arrivals << current.toPlainText();
  98         }
  99         current.addClass("arrivals");
 100         current = current.nextSibling();
 101         qDebug() << "marking as arrival";
 102         if (current.isNull())
 103             break;
 104     }
 105
 106     setStringList(departures);
 107     qDebug() << "departures list contain:";
 108     qDebug() << departures;
 109     qDebug() << "arrivals list contain:";
 110     qDebug() << arrivals;
 111     emit  layoutChanged();
 112 }
 113
 114 void StationScheduleModel::fetch(const QString &name)
 115 {
 116     DataProvider *provider = DataProvider::instance();
 117
 118     provider->fetchStationSchedule(name);
 119     setName(name);
 120 }