[vlille] adapt to browser2
This commit is contained in:
Bezleputh
parent
b03881de24
commit
22209d0020
3 changed files with 109 additions and 101 deletions
|
|
@ -18,7 +18,9 @@
|
|||
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
|
||||
from weboob.tools.browser import BasePage
|
||||
from weboob.tools.browser2.page import HTMLPage, XMLPage, method, ListElement, ItemElement, TableElement
|
||||
from weboob.tools.browser2.filters import CleanText, TableCell, Filter
|
||||
|
||||
from weboob.capabilities.gauge import Gauge, GaugeMeasure, GaugeSensor
|
||||
from weboob.capabilities.base import NotLoaded
|
||||
import datetime
|
||||
|
|
@ -27,86 +29,100 @@ import re
|
|||
__all__ = ['InfoStationPage', 'ListStationsPage']
|
||||
|
||||
|
||||
class InfoStationPage(BasePage):
|
||||
def _create_bikes_sensor(self, value, gauge_id, last_update, adresse):
|
||||
levelbikes = GaugeSensor(gauge_id + '-bikes')
|
||||
levelbikes.name = u'Bikes'
|
||||
levelbikes.address = u'%s' % adresse
|
||||
lastvalue = GaugeMeasure()
|
||||
lastvalue.level = float(value)
|
||||
lastvalue.date = last_update
|
||||
if lastvalue.level < 1:
|
||||
lastvalue.alarm = u'Empty station'
|
||||
levelbikes.lastvalue = lastvalue
|
||||
levelbikes.history = NotLoaded
|
||||
levelbikes.gaugeid = gauge_id
|
||||
return levelbikes
|
||||
|
||||
def _create_attach_sensor(self, value, gauge_id, last_update, adresse):
|
||||
levelattach = GaugeSensor(gauge_id + '-attach')
|
||||
levelattach.name = u'Attach'
|
||||
levelattach.address = u'%s' % adresse
|
||||
lastvalue = GaugeMeasure()
|
||||
if lastvalue.level < 1:
|
||||
lastvalue.alarm = u'Full station'
|
||||
lastvalue.level = float(value)
|
||||
lastvalue.date = last_update
|
||||
levelattach.lastvalue = lastvalue
|
||||
levelattach.history = NotLoaded
|
||||
levelattach.gaugeid = gauge_id
|
||||
return levelattach
|
||||
|
||||
def _create_status_sensor(self, value, gauge_id, last_update, adresse):
|
||||
levelstatus = GaugeSensor(gauge_id + '-status')
|
||||
levelstatus.name = u'Status'
|
||||
levelstatus.address = u'%s' % adresse
|
||||
lastvalue = GaugeMeasure()
|
||||
status = float(value)
|
||||
if status == 0:
|
||||
status = 1
|
||||
else:
|
||||
status = -1
|
||||
if lastvalue.level < 1:
|
||||
lastvalue.alarm = u'Not available station'
|
||||
lastvalue.level = float(status)
|
||||
lastvalue.date = last_update
|
||||
levelstatus.lastvalue = lastvalue
|
||||
levelstatus.history = NotLoaded
|
||||
levelstatus.gaugeid = gauge_id
|
||||
return levelstatus
|
||||
|
||||
def _get_last_update(self, last_update):
|
||||
class LastDateFilter(Filter):
|
||||
def filter(self, last_update):
|
||||
return datetime.datetime.now() - datetime.timedelta(seconds=int(re.match(r'\d+', last_update).group(0)))
|
||||
|
||||
def get_station_infos(self, gauge_id):
|
||||
|
||||
last_update = self._get_last_update(self.parser.select(self.document.getroot(), 'lastupd', 1).text)
|
||||
sensors = []
|
||||
class InfoStationPage(XMLPage):
|
||||
@method
|
||||
class get_station_infos(ListElement):
|
||||
item_xpath = "."
|
||||
|
||||
adresse = self.parser.select(self.document.getroot(), 'adress', 1).text
|
||||
class item(ItemElement):
|
||||
klass = Gauge
|
||||
|
||||
sensors.append(self._create_bikes_sensor(self.parser.select(self.document.getroot(), 'bikes', 1).text, gauge_id, last_update, adresse))
|
||||
def _create_bikes_sensor(self, value, gauge_id, last_update, adresse):
|
||||
levelbikes = GaugeSensor(gauge_id + '-bikes')
|
||||
levelbikes.name = u'Bikes'
|
||||
levelbikes.address = u'%s' % adresse
|
||||
lastvalue = GaugeMeasure()
|
||||
lastvalue.level = float(value)
|
||||
lastvalue.date = last_update
|
||||
if lastvalue.level < 1:
|
||||
lastvalue.alarm = u'Empty station'
|
||||
levelbikes.lastvalue = lastvalue
|
||||
levelbikes.history = NotLoaded
|
||||
levelbikes.gaugeid = gauge_id
|
||||
return levelbikes
|
||||
|
||||
sensors.append(self._create_attach_sensor(self.parser.select(self.document.getroot(), 'attachs', 1).text, gauge_id, last_update, adresse))
|
||||
def _create_attach_sensor(self, value, gauge_id, last_update, adresse):
|
||||
levelattach = GaugeSensor(gauge_id + '-attach')
|
||||
levelattach.name = u'Attach'
|
||||
levelattach.address = u'%s' % adresse
|
||||
lastvalue = GaugeMeasure()
|
||||
if lastvalue.level < 1:
|
||||
lastvalue.alarm = u'Full station'
|
||||
lastvalue.level = float(value)
|
||||
lastvalue.date = last_update
|
||||
levelattach.lastvalue = lastvalue
|
||||
levelattach.history = NotLoaded
|
||||
levelattach.gaugeid = gauge_id
|
||||
return levelattach
|
||||
|
||||
sensors.append(self._create_status_sensor(self.parser.select(self.document.getroot(), 'status', 1).text, gauge_id, last_update, adresse))
|
||||
def _create_status_sensor(self, value, gauge_id, last_update, adresse):
|
||||
levelstatus = GaugeSensor(gauge_id + '-status')
|
||||
levelstatus.name = u'Status'
|
||||
levelstatus.address = u'%s' % adresse
|
||||
lastvalue = GaugeMeasure()
|
||||
status = float(value)
|
||||
if status == 0:
|
||||
status = 1
|
||||
else:
|
||||
status = -1
|
||||
if lastvalue.level < 1:
|
||||
lastvalue.alarm = u'Not available station'
|
||||
lastvalue.level = float(status)
|
||||
lastvalue.date = last_update
|
||||
levelstatus.lastvalue = lastvalue
|
||||
levelstatus.history = NotLoaded
|
||||
levelstatus.gaugeid = gauge_id
|
||||
return levelstatus
|
||||
|
||||
return sensors
|
||||
def parse(self, el):
|
||||
self.obj = self.env['obj']
|
||||
|
||||
def obj_sensors(self):
|
||||
sensors = []
|
||||
last_update = LastDateFilter(CleanText('lastupd'))(self)
|
||||
adresse = CleanText('adress')(self)
|
||||
sensors.append(self._create_bikes_sensor(CleanText('bikes')(self),
|
||||
self.env['idgauge'],
|
||||
last_update, adresse))
|
||||
sensors.append(self._create_attach_sensor(CleanText('attachs')(self),
|
||||
self.env['idgauge'],
|
||||
last_update, adresse))
|
||||
sensors.append(self._create_status_sensor(CleanText('status')(self),
|
||||
self.env['idgauge'],
|
||||
last_update, adresse))
|
||||
return sensors
|
||||
|
||||
|
||||
class ListStationsPage(BasePage):
|
||||
def get_station_list(self):
|
||||
gauges = []
|
||||
class ListStationsPage(HTMLPage):
|
||||
@method
|
||||
class get_station_list(TableElement):
|
||||
item_xpath = "//table[@id='ctl00_Contenu_ListeStations1_ListViewStations_itemPlaceholderContainer']/tr"
|
||||
head_xpath = "//table[@id='ctl00_Contenu_ListeStations1_ListViewStations_itemPlaceholderContainer']/tr/th/@id"
|
||||
|
||||
trs = self.document.getroot().xpath('//table[@id="ctl00_Contenu_ListeStations1_ListViewStations_itemPlaceholderContainer"]/tr')
|
||||
col_id = 'ctl00_Contenu_ListeStations1_ListViewStations_Th1'
|
||||
col_name = 'ctl00_Contenu_ListeStations1_ListViewStations_Th2'
|
||||
col_city = 'ctl00_Contenu_ListeStations1_ListViewStations_Th9'
|
||||
|
||||
for tr in trs:
|
||||
if not ('id' in tr.attrib):
|
||||
tds = self.parser.select(tr, 'td/span', method='xpath')
|
||||
if len(tds) > 4:
|
||||
gauge = Gauge(int(tds[0].text))
|
||||
gauge.name = u'%s' % tds[1].text
|
||||
gauge.city = u'%s' % tds[3].text
|
||||
gauge.object = u'vLille'
|
||||
gauges.append(gauge)
|
||||
return gauges
|
||||
class item(ItemElement):
|
||||
klass = Gauge
|
||||
condition = lambda self: (len(self.el.xpath('td/span')) > 4 and not ('id' in self.el.attrib))
|
||||
|
||||
obj_id = CleanText(TableCell('id'))
|
||||
obj_name = CleanText(TableCell('name'))
|
||||
obj_city = CleanText(TableCell('city'))
|
||||
obj_object = u'vLille'
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue