improve the export decorator slightly to allow for it to be used without

being called.
This commit is contained in:
Damien Churchill 2009-03-16 13:30:55 +00:00
parent c0f11fbae4
commit db853c7c56
1 changed files with 7 additions and 1 deletions

View File

@ -34,6 +34,7 @@ from twisted.internet.protocol import Factory, Protocol
from twisted.internet import ssl, reactor
from OpenSSL import crypto, SSL
from types import FunctionType
import deluge.rencode as rencode
from deluge.log import LOG as log
@ -60,7 +61,12 @@ def export(auth_level=AUTH_LEVEL_DEFAULT):
func._rpcserver_auth_level = auth_level
return func
return wrap
if type(auth_level) is FunctionType:
func = auth_level
auth_level = AUTH_LEVEL_DEFAULT
return wrap(func)
else:
return wrap
class DelugeError(Exception):
pass