287 lines
10 KiB
Python
287 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2010-2011 Julien Hébert
|
|
#
|
|
# This file is part of weboob.
|
|
#
|
|
# weboob is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# weboob is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
from weboob.capabilities.travel import ICapTravel
|
|
from weboob.tools.application.base import BaseApplication
|
|
from logging import warning
|
|
|
|
import gtk
|
|
|
|
class FakeConic(object):
|
|
STATUS_CONNECTED = None
|
|
STATUS_DISCONNECTED = None
|
|
CONNECT_FLAG_NONE = None
|
|
def Connection(self):
|
|
raise NotImplementedError()
|
|
try:
|
|
import hildon
|
|
except ImportError:
|
|
toolkit = gtk
|
|
else:
|
|
toolkit = hildon
|
|
|
|
try :
|
|
import conic
|
|
except ImportError:
|
|
warning("conic is not found")
|
|
conic = FakeConic()
|
|
|
|
|
|
from logging import debug
|
|
|
|
|
|
__all__ = ['Masstransit']
|
|
|
|
class MasstransitHildon():
|
|
"hildon interface"
|
|
|
|
def connect_event(self, connection, event=None, c=None, d=None):
|
|
debug("DBUS-DEBUG a: %s, b:%s, c:%s,d: %s" % (connection, event, c, d))
|
|
status = event.get_status()
|
|
if status == conic.STATUS_CONNECTED:
|
|
self.connected = True
|
|
if self.touch_selector_entry_filled == False:
|
|
debug("connected, now fill")
|
|
self.fill_touch_selector_entry()
|
|
if self.refresh_in_progress:
|
|
self.refresh()
|
|
elif status == conic.STATUS_DISCONNECTED:
|
|
self.connected = False
|
|
|
|
def __init__(self, weboob):
|
|
self.touch_selector_entry_filled = False
|
|
self.refresh_in_progress = False
|
|
self.connected = False
|
|
self.weboob = weboob
|
|
try :
|
|
self.connection = conic.Connection()
|
|
self.connection.connect("connection-event", self.connect_event)
|
|
self.connection.set_property("automatic-connection-events", True)
|
|
self.connection.request_connection(conic.CONNECT_FLAG_NONE)
|
|
except NotImplementedError:
|
|
pass
|
|
|
|
horizontal_box = gtk.HBox()
|
|
self.main_window = toolkit.Window()
|
|
try :
|
|
self.refresh_button = toolkit.Button(
|
|
gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT,
|
|
hildon.BUTTON_ARRANGEMENT_HORIZONTAL,
|
|
"Actualiser"
|
|
)
|
|
self.retour_button = hildon.Button(
|
|
gtk.HILDON_SIZE_AUTO_WIDTH | gtk.HILDON_SIZE_FINGER_HEIGHT,
|
|
hildon.BUTTON_ARRANGEMENT_HORIZONTAL,
|
|
"Retour"
|
|
)
|
|
self.combo_source = hildon.TouchSelectorEntry(text=True)
|
|
self.combo_dest = hildon.TouchSelectorEntry(text=True)
|
|
self.picker_button_source = hildon.PickerButton(
|
|
gtk.HILDON_SIZE_AUTO,
|
|
hildon.BUTTON_ARRANGEMENT_VERTICAL)
|
|
|
|
self.picker_button_dest = hildon.PickerButton(
|
|
gtk.HILDON_SIZE_AUTO,
|
|
hildon.BUTTON_ARRANGEMENT_VERTICAL
|
|
)
|
|
self.picker_button_source.set_sensitive(False)
|
|
self.picker_button_dest.set_sensitive(False)
|
|
|
|
self.picker_button_source.set_title("Gare de Depart")
|
|
self.picker_button_dest.set_title("Gare d'arrivee")
|
|
|
|
self.picker_button_source.set_selector(self.combo_source)
|
|
self.picker_button_dest.set_selector(self.combo_dest)
|
|
horizontal_box.pack_start(self.picker_button_source)
|
|
horizontal_box.pack_start(self.picker_button_dest)
|
|
|
|
except AttributeError:
|
|
self.refresh_button = gtk.Button("Actualiser")
|
|
self.retour_button = gtk.Button("Retour")
|
|
self.combo_source = gtk.combo_box_entry_new_text()
|
|
self.combo_dest = gtk.combo_box_entry_new_text()
|
|
horizontal_box.pack_start(self.combo_source)
|
|
horizontal_box.pack_start(self.combo_dest)
|
|
|
|
self.main_window.set_title("Horaires des Prochains Trains")
|
|
self.main_window.connect("destroy", self.on_main_window_destroy)
|
|
|
|
self.refresh_button.connect("clicked", self.on_refresh_button_clicked)
|
|
|
|
self.retour_button.set_sensitive(False)
|
|
self.retour_button.connect("clicked", self.on_retour_button_clicked)
|
|
|
|
self.treestore = gtk.TreeStore(str, str, str, str, str)
|
|
treeview = gtk.TreeView(self.treestore)
|
|
|
|
|
|
treeview.append_column(
|
|
gtk.TreeViewColumn(
|
|
'Train',
|
|
gtk.CellRendererText(),
|
|
text=0
|
|
))
|
|
|
|
treeview.append_column(
|
|
gtk.TreeViewColumn(
|
|
'Horaire',
|
|
gtk.CellRendererText(),
|
|
text=1
|
|
))
|
|
|
|
treeview.append_column(
|
|
gtk.TreeViewColumn(
|
|
'Destination',
|
|
gtk.CellRendererText(),
|
|
text=2
|
|
))
|
|
treeview.append_column(
|
|
gtk.TreeViewColumn(
|
|
'Voie',
|
|
gtk.CellRendererText(),
|
|
text=3
|
|
))
|
|
treeview.append_column(
|
|
gtk.TreeViewColumn(
|
|
'Information',
|
|
gtk.CellRendererText(),
|
|
text=4
|
|
))
|
|
|
|
|
|
|
|
vertical_box = gtk.VBox()
|
|
vertical_box.pack_start(horizontal_box)
|
|
horizontal_box.pack_start(self.retour_button)
|
|
vertical_box.pack_start(treeview)
|
|
vertical_box.pack_start(self.refresh_button)
|
|
|
|
self.main_window.add(vertical_box)
|
|
self.main_window.show_all()
|
|
self.fill_touch_selector_entry()
|
|
|
|
if toolkit != gtk:
|
|
self.picker_button_source.connect("value-changed",
|
|
self.check_station_input,
|
|
self.picker_button_source)
|
|
self.picker_button_dest.connect("value-changed",
|
|
self.check_station_input,
|
|
self.picker_button_dest)
|
|
|
|
def fill_touch_selector_entry(self):
|
|
liste = []
|
|
|
|
for backend in self.weboob.iter_backends():
|
|
for station in backend.iter_station_search(""):
|
|
liste.append(station.name.capitalize())
|
|
|
|
liste.sort()
|
|
|
|
for station in liste:
|
|
self.combo_source.append_text(station)
|
|
self.combo_dest.append_text(station)
|
|
|
|
self.touch_selector_entry_filled = True
|
|
if toolkit != gtk:
|
|
self.picker_button_source.set_sensitive(True)
|
|
|
|
def on_main_window_destroy(self, widget):
|
|
"exit application at the window close"
|
|
gtk.main_quit()
|
|
|
|
def on_main_window_show(self, param):
|
|
self.fill_touch_selector_entry()
|
|
|
|
def on_retour_button_clicked(self, widget):
|
|
"the button is clicked"
|
|
debug("on_retour_button_clicked")
|
|
self.refresh_in_progress = True
|
|
col_source = self.combo_source.get_active(0)
|
|
col_dest = self.combo_dest.get_active(0)
|
|
self.combo_source.set_active(0, col_dest)
|
|
self.combo_dest.set_active(0, col_source)
|
|
self.refresh()
|
|
|
|
def on_refresh_button_clicked(self, widget):
|
|
"the refresh button is clicked"
|
|
debug("on_refresh_button_clicked")
|
|
self.refresh_in_progress = True
|
|
try :
|
|
self.connection.request_connection(conic.CONNECT_FLAG_NONE)
|
|
except AttributeError:
|
|
if isinstance(conic, FakeConic):
|
|
self.refresh()
|
|
else:
|
|
raise
|
|
|
|
def check_station_input(self, widget, user_data):
|
|
if self.combo_source.get_current_text() is None :
|
|
self.picker_button_dest.set_sensitive(False)
|
|
self.refresh_button.set_sensitive(False)
|
|
self.retour_button.set_sensitive(False)
|
|
else:
|
|
self.picker_button_dest.set_sensitive(True)
|
|
if self.combo_dest.get_current_text() is None:
|
|
self.refresh_button.set_sensitive(False)
|
|
self.retour_button.set_sensitive(False)
|
|
else:
|
|
self.refresh_button.set_sensitive(True)
|
|
self.retour_button.set_sensitive(True)
|
|
|
|
def refresh(self):
|
|
"update departures"
|
|
banner = hildon.hildon_banner_show_information(self.main_window, "", "Chargement en cours")
|
|
banner.set_timeout(10000)
|
|
hildon.hildon_gtk_window_set_progress_indicator(self.main_window, 1)
|
|
self.treestore.clear()
|
|
try :
|
|
source_text = self.combo_source.get_current_text()
|
|
dest_text = self.combo_dest.get_current_text()
|
|
except AttributeError:
|
|
source_text = self.combo_source.child.get_text()
|
|
dest_text = self.combo_dest.child.get_text()
|
|
for backend in self.weboob.iter_backends():
|
|
for station in backend.iter_station_search(source_text):
|
|
for arrival in \
|
|
backend.iter_station_search(dest_text):
|
|
for departure in \
|
|
backend.iter_station_departures(station.id, arrival.id):
|
|
self.treestore.append(None,
|
|
[departure.type,
|
|
departure.time,
|
|
departure.arrival_station,
|
|
departure.plateform,
|
|
departure.information])
|
|
|
|
self.refresh_in_progress = False
|
|
banner.set_timeout(1)
|
|
hildon.hildon_gtk_window_set_progress_indicator(self.main_window, 0)
|
|
|
|
|
|
class Masstransit(BaseApplication):
|
|
"Application Class"
|
|
APPNAME = 'masstransit'
|
|
VERSION = '0.8'
|
|
COPYRIGHT = 'Copyright(C) 2010-2011 Julien Hébert'
|
|
|
|
def main(self, argv):
|
|
self.load_backends(ICapTravel)
|
|
MasstransitHildon(self.weboob)
|
|
gtk.main()
|