nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add the 'travel' frontend

Browse source
This commit is contained in:
Romain Bignon 2010-03-23 13:59:14 +01:00
commit 3f6d62f34a
3 changed files with 160 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

23
weboob/frontends/travel/__init__.py Normal file
View file

@ -0,0 +1,23 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# vim: ft=python et softtabstop=4 cinoptions=4 shiftwidth=4 ts=4 ai
"""
Copyright(C) 2010 Romain Bignon
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, version 3 of the License.
This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""
from .application import Application

109
weboob/frontends/travel/application.py Normal file
View file

@ -0,0 +1,109 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# vim: ft=python et softtabstop=4 cinoptions=4 shiftwidth=4 ts=4 ai
"""
Copyright(C) 2010 Romain Bignon
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, version 3 of the License.
This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""
import sys
from types import MethodType
from weboob import Weboob
from weboob.capabilities.travel import ICapTravel
from weboob.tools.application import BaseApplication
class Application(BaseApplication):
APPNAME = 'travel'
CONFIG = {}
def main(self, argv):
#if not self.config:
# print >>sys.stderr, "Error: %s is not configured yet. Please call 'weboob.travel -c'" % argv[0]
# print >>sys.stderr, "Also, you need to use 'weboobcfg' to set backend configs"
# return -1
self.weboob.load_modules(ICapTravel)
if len(argv) == 1:
print >>sys.stderr, "Usage: %s <command> [args ...]" % argv[0]
return -1
return self.command(argv[1], *argv[2:])
def getMethods(self, prefix):
services = {}
for attrname in dir(self):
if not attrname.startswith(prefix):
continue
attr = getattr(self, attrname)
if not isinstance(attr, MethodType):
continue
name = attrname[len(prefix):]
services[name] = attr
return services
def command(self, command, *args):
commands = self.getMethods('command_')
if not command in commands:
print >>sys.stderr, "No such command: %s" % command
self.command_help()
return 1
try:
return commands[command](*args)
except TypeError, e:
try:
print >>sys.stderr, "Command %s takes %s arguments" % (command, int(str(e).split(' ')[3]) - 1)
except:
print >>sys.stderr, '%s' % e
return 1
def command_help(self):
print 'Available commands are:'
print ' stations <pattern> Search stations'
print ' departures <station> List all departures on a special station'
def command_stations(self, pattern):
print ".-----------------.----------------------------------------."
print '| ID | Name |'
print '+-----------------+----------------------------------------+'
count = 0
for name, backend, in self.weboob.iter_backends():
for station in backend.iter_station_search(pattern):
print '| %-15s | %-38s |' % (station.id, station.name)
count += 1
print "+-----------------'----------------------------------------+"
print "| %3d stations listed |" % count
print "'----------------------------------------------------------'"
def command_departures(self, station):
print ".-----.-----------.-------.-----------------------.-------.--------------------."
print "| ID | Type | Time | Arrival | Late | Info |"
print "+-----+-----------+-------+-----------------------+-------+--------------------+"
count = 0
for name, backend, in self.weboob.iter_backends():
for departure in backend.iter_station_departures(station):
print u"| %3d | %-9s | %5s | %-21s | %5s | %-18s |" % (departure.id,
departure.type,
departure.time.strftime("%H:%M"),
departure.arrival_station,
departure.late and departure.late.strftime("%H:%M") or '',
departure.information.replace('\n', '').strip())
count += 1
print "+-----'-----------'-------'-----------------------'-------'--------------------+"
print "| %3d departures listed |" % count
print "'------------------------------------------------------------------------------'"

28
weboob/frontends/travel/scripts/travel Executable file
View file

@ -0,0 +1,28 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# vim: ft=python et softtabstop=4 cinoptions=4 shiftwidth=4 ts=4 ai
"""
Copyright(C) 2010 Romain Bignon
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, version 3 of the License.
This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""
import sys
from weboob.frontends.travel import Application
if __name__ == '__main__':
app = Application()
sys.exit(app.main(sys.argv))