stricter id2url decorator

This commit is contained in:
Christophe Benz 2010-05-20 16:28:19 +02:00
commit 15d253133b

View file

@ -41,12 +41,16 @@ def id2url(id2url):
else: else:
if '@' in arg: if '@' in arg:
_id, provider = arg.split('@') _id, provider = arg.split('@')
if provider in self.DOMAIN: if provider == self.DOMAIN:
url = id2url(_id) url = id2url(_id)
if url is None:
return None
else: else:
return None return None
else: else:
url = id2url(arg) url = id2url(arg)
if url is None:
return None
new_args = [url] new_args = [url]
new_args.extend(args[1:]) new_args.extend(args[1:])
return func(self, *new_args, **kwargs) return func(self, *new_args, **kwargs)