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

Change sachsen module to use ICapGauge

Browse source
This commit is contained in:
Florent 2012-11-08 20:16:19 +01:00 • committed by Romain Bignon
commit 8e6b0fba4c
4 changed files with 111 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

115
modules/sachsen/pages.py
View file

@ -19,8 +19,8 @@
from datetime import datetime, date, time
from weboob.tools.browser import BasePage
from weboob.capabilities.gauge import Gauge, GaugeMeasure
from weboob.capabilities.base import NotAvailable
from weboob.capabilities.gauge import Gauge, GaugeMeasure, GaugeSensor
from weboob.capabilities.base import NotAvailable, NotLoaded
__all__ = ['ListPage', 'HistoryPage']
@ -32,74 +32,87 @@ class ListPage(BasePage):
data = pegel.attrib['onmouseover'].strip('pegelein(').strip(')').replace(",'", ",").split("',")
gauge = Gauge(int(data[7]))
gauge.name = unicode(data[0].strip("'"))
gauge.river = unicode(data[1])
gauge.city = gauge.name.split(' ')[0] # TODO: real regexp to remove the number
gauge.object = unicode(data[1])
sensors = []
try:
lastdate = date(*reversed([int(x) for x in data[2].split(' ')[0].split(".")]))
lasttime = time(*[int(x) for x in data[2].split(' ')[1].split(":")])
gauge.lastdate = datetime.combine(lastdate, lasttime)
lastdate = datetime.combine(lastdate, lasttime)
except:
gauge.lastdate = NotAvailable
try:
gauge.level = float(data[3])
except:
gauge.level = NotAvailable
try:
gauge.flow = float(data[4])
except:
gauge.flow = NotAvailable
lastdate = NotAvailable
bildforecast = data[5]
if bildforecast == "pf_gerade.png":
gauge.forecast = u"stable"
forecast = u"stable"
elif bildforecast == "pf_unten.png":
gauge.forecast = u"Go down"
forecast = u"Go down"
elif bildforecast == "pf_oben.png":
gauge.forecast = u"Go up"
forecast = u"Go up"
else:
gauge.forecast = NotAvailable
forecast = NotAvailable
try:
level = float(data[3])
levelsensor = GaugeSensor(gauge.id + "-level")
levelsensor.name = u"Level"
# TODO levelsensor.unit =
levelsensor.forecast = forecast
lastvalue = GaugeMeasure()
lastvalue.level = level
lastvalue.date = lastdate
# TODO lastvalue.alarm =
levelsensor.lastvalue = lastvalue
levelsensor.history = NotLoaded
levelsensor.gaugeid = gauge.id
sensors.append(levelsensor)
except:
pass
try:
flow = float(data[4])
flowsensor = GaugeSensor(gauge.id + "-flow")
flowsensor.name = u"Flow"
# TODO flowsensor.unit =
flowsensor.forecast = forecast
lastvalue = GaugeMeasure()
lastvalue.level = flow
lastvalue.date = lastdate
# TODO lastvalue.alarm =
flowsensor.lastvalue = lastvalue
flowsensor.history = NotLoaded
flowsensor.gaugeid = gauge.id
sensors.append(flowsensor)
except:
pass
gauge.sensors = sensors
yield gauge
class HistoryPage(BasePage):
def iter_history(self):
def iter_history(self, sensor):
table = self.document.getroot().cssselect('table[width="215"]')
first = True
for line in table[0].cssselect("tr"):
if first:
first = False
continue
lines = table[0].cssselect("tr")
lines.pop(0) # remove header
lines.pop(0) # remove first value (already in lastvalue)
for line in lines:
history = GaugeMeasure()
leveldate = date(*reversed([int(x) for x in line[0].text_content().split(' ')[0].split(".")]))
leveltime = time(*[int(x) for x in line[0].text_content().split(' ')[1].split(":")])
history.date = datetime.combine(leveldate, leveltime)
try:
history.level = float(line[1].text_content())
except:
history.level = NotAvailable
try:
history.flow = float(line[2].text_content())
except:
history.flow = NotAvailable
if sensor.name == u"Level":
try:
history.level = float(line[1].text_content())
except:
history.level = NotAvailable
elif sensor.name == u"Flow":
try:
history.level = float(line[2].text_content())
except:
history.level = NotAvailable
# TODO: history.alarm
yield history
def first_value(self, table, index):
first = NotAvailable
for lignes in table[0].cssselect("tr"):
try:
valeur = float(lignes[index].text_content())
if (valeur > 1.0):
first = valeur
return first
except:
continue
return first
def last_seen(self):
tables = self.document.getroot().cssselect('table[width="215"]')
gauge = GaugeMeasure()
gauge.level = self.first_value(tables, 1)
gauge.flow = self.first_value(tables, 2)
return gauge