Skip to content

Commit b003384

Browse files
committed
before_perform() & after_perform() should be called on class, not pass it in
Fixes a problem where you would have: @classmethod def before_perform(cls, metadata): pass and you would get an exception: ... before_perform(payload_class, metadata) TypeError: before_perform() takes exactly 2 arguments (3 given)
1 parent 3a50d9a commit b003384

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pyres/job.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def perform(self):
7878
check_after = True
7979
try:
8080
if before_perform:
81-
before_perform(payload_class, metadata)
81+
payload_class.before_perform(metadata)
8282
return payload_class.perform(*args)
8383
except:
8484
check_after = False
@@ -91,7 +91,7 @@ def perform(self):
9191
finally:
9292
after_perform = getattr(payload_class, "after_perform", None)
9393
if after_perform and check_after:
94-
after_perform(payload_class, metadata)
94+
payload_class.after_perform(metadata)
9595
delattr(payload_class,'resq')
9696

9797
def fail(self, exception):

0 commit comments

Comments
 (0)