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

ability to check caps of a backend against a string (not only a class)

Browse source
This commit is contained in:
Romain Bignon 2010-08-11 14:58:35 +02:00
commit 7b0c91275f
3 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
weboob/tools/backend.py
View file

@ -21,6 +21,7 @@ import os
from threading import RLock
from logging import debug
from weboob.capabilities.cap import ICap
__all__ = ['BaseBackend', 'ObjectNotSupported']
@ -170,9 +171,15 @@ class BaseBackend(object):
return self.BROWSER(*args, **kwargs)
def iter_caps(self):
for cap in self.__class__.__bases__:
if issubclass(cap, ICap) and cap != ICap:
yield cap
def has_caps(self, *caps):
for c in caps:
if isinstance(self, c):
if (isinstance(c, (unicode,str)) and c in [cap.__name__ for cap in self.iter_caps()]) or \
isinstance(self, c):
return True
return False