DandelionNStuff/pymilter
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Set milter_protocol attribute of noreply wrapper

Browse Source
This commit is contained in:
Stuart Gathman
2009-06-04 22:02:09 +00:00
parent 84bd61aac1
commit 7311f65150
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Milter/__init__.py
+2 -1
View File
@@ -46,7 +46,7 @@ def nocallback(func):
def noreply(func): def noreply(func):
try: try:
nr_mask = func.milter_protocol = OPTIONAL_CALLBACKS[func.__name__][0] nr_mask = OPTIONAL_CALLBACKS[func.__name__][0]
except KeyErro: except KeyErro:
raise ValueError( raise ValueError(
'@noreply applied to non-optional method: '+func.__name__) '@noreply applied to non-optional method: '+func.__name__)
@@ -54,6 +54,7 @@ def noreply(func):
rc = func(self,*args) rc = func(self,*args)
if self._protocol & nr_mask: return NOREPLY if self._protocol & nr_mask: return NOREPLY
return rc return rc
wrapper.milter_protocol = nr_mask
return wrapper return wrapper
class DisabledAction(RuntimeError): class DisabledAction(RuntimeError):