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

Fix some collecton validation

Browse source 
And add restrict_level helper method.
This commit is contained in:
Laurent Bachelier 2012-03-11 15:25:16 +01:00
commit 1430b40bc5
7 changed files with 11 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

3
modules/nova/backend.py
View file

@ -64,8 +64,7 @@ class NovaBackend(BaseBackend, ICapRadio, ICapCollection):
def iter_resources(self, objs, split_path):
if Radio in objs:
if len(split_path) > 0:
raise CollectionNotFound(split_path)
self._restrict_level(split_path)
for id in self._RADIOS.iterkeys():
yield self.get_radio(id)

3
modules/ouifm/backend.py
View file

@ -48,8 +48,7 @@ class OuiFMBackend(BaseBackend, ICapRadio, ICapCollection):
def iter_resources(self, objs, split_path):
if Radio in objs:
if len(split_path) > 0:
raise CollectionNotFound(split_path)
self._restrict_level(split_path)
for id in self._RADIOS.iterkeys():
yield self.get_radio(id)

2
modules/radiofrance/backend.py
View file

@ -112,7 +112,7 @@ class RadioFranceBackend(BaseBackend, ICapRadio, ICapCollection, ICapVideo):
for _id in sorted(self._RADIOS.iterkeys()):
if not _id.startswith('fb'):
yield self.get_radio(_id)
yield Collection(['francebleu'], 'France Bleu')
yield Collection([u'francebleu'], u'France Bleu')
else:
raise CollectionNotFound(split_path)

6
modules/redmine/backend.py
View file

@ -96,17 +96,15 @@ class RedmineBackend(BaseBackend, ICapContent, ICapBugTracker, ICapCollection):
############# CapCollection ###################################################
def iter_resources(self, objs, split_path):
if Project in objs or Issue in objs:
self._restrict_level(split_path, 1)
if len(split_path) == 0:
return [Collection([project.id], project.name)
for project in self.iter_projects()]
if len(split_path) == 1:
elif len(split_path) == 1:
query = Query()
query.project = unicode(split_path[0])
return self.iter_issues(query)
raise CollectionNotFound(split_path)
def validate_collection(self, objs, collection):
if len(collection.split_path) == 0:
return

3
weboob/capabilities/bank.py
View file

@ -96,8 +96,7 @@ class Transfer(CapBaseObject):
class ICapBank(ICapCollection):
def iter_resources(self, objs, split_path):
if Account in objs:
if len(split_path) > 0:
raise CollectionNotFound(split_path)
self._restrict_level(split_path)
return self.iter_accounts()

3
weboob/capabilities/bill.py
View file

@ -56,8 +56,7 @@ class Subscription(CapBaseObject):
class ICapBill(ICapCollection):
def iter_resources(self, objs, split_path):
if Subscription in objs:
if len(split_path) > 0:
raise CollectionNotFound(split_path)
self._restrict_level(split_path)
return self.iter_subscription()

4
weboob/capabilities/collection.py
View file

@ -100,3 +100,7 @@ class ICapCollection(IBaseCap):
i.next()
except StopIteration:
raise CollectionNotFound(collection.split_path)
def _restrict_level(self, split_path, lmax=0):
if len(split_path) > lmax:
raise CollectionNotFound(split_path)