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

Add SensorNotFound exception

Browse source
This commit is contained in:
Florent 2012-11-08 20:22:45 +01:00 • committed by Romain Bignon
commit c838b91fb2
2 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
modules/sachsen/backend.py
View file

@ -21,7 +21,7 @@
from __future__ import with_statement from __future__ import with_statement
from .browser import SachsenBrowser from .browser import SachsenBrowser
from weboob.capabilities.gauge import ICapGauge, GaugeSensor, Gauge from weboob.capabilities.gauge import ICapGauge, GaugeSensor, Gauge, SensorNotFound
from weboob.tools.backend import BaseBackend from weboob.tools.backend import BaseBackend
@ -75,9 +75,13 @@ class SachsenLevelBackend(BaseBackend, ICapGauge):
def iter_gauge_history(self, sensor): def iter_gauge_history(self, sensor):
if not isinstance(sensor, GaugeSensor): if not isinstance(sensor, GaugeSensor):
sensor = self._get_sensor_by_id(sensor) sensor = self._get_sensor_by_id(sensor)
if sensor is None:
raise SensorNotFound()
return self.browser.iter_history(sensor) return self.browser.iter_history(sensor)
def get_last_measure(self, sensor): def get_last_measure(self, sensor):
if not isinstance(sensor, GaugeSensor): if not isinstance(sensor, GaugeSensor):
sensor = self._get_sensor_by_id(sensor) sensor = self._get_sensor_by_id(sensor)
if sensor is None:
raise SensorNotFound()
return sensor.lastvalue return sensor.lastvalue

10
weboob/capabilities/gauge.py
View file

@ -18,10 +18,16 @@
# along with weboob. If not, see <http://www.gnu.org/licenses/>. # along with weboob. If not, see <http://www.gnu.org/licenses/>.
from .base import IBaseCap, CapBaseObject, StringField, FloatField, DateField, Field from .base import IBaseCap, CapBaseObject, StringField, FloatField, DateField, Field, UserError
__all__ = ['Gauge', 'GaugeSensor', 'GaugeMeasure', 'ICapGauge'] __all__ = ['Gauge', 'GaugeSensor', 'GaugeMeasure', 'ICapGauge', 'SensorNotFound']
class SensorNotFound(UserError):
"""
Not found a sensor
"""
class Gauge(CapBaseObject): class Gauge(CapBaseObject):