Skip to content

Commit f475dcb

Browse files
author
Matt George
committed
tweaking before and after perform methods
1 parent 7103f8a commit f475dcb

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

pyres/job.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,14 +73,16 @@ def perform(self):
7373
metadata["enqueue_timestamp"] = self.enqueue_timestamp
7474

7575
before_perform = getattr(payload_class, "before_perform", None)
76-
if before_perform:
77-
before_perform(metadata)
7876

7977
metadata["failed"] = False
8078
metadata["perform_timestamp"] = time.time()
79+
check_after = True
8180
try:
81+
if before_perform:
82+
before_perform(metadata)
8283
return payload_class.perform(*args)
8384
except:
85+
check_after = False
8486
metadata["failed"] = True
8587
if not self.retry(payload_class, args):
8688
metadata["retried"] = False
@@ -89,7 +91,7 @@ def perform(self):
8991
metadata["retried"] = True
9092
finally:
9193
after_perform = getattr(payload_class, "after_perform", None)
92-
if after_perform:
94+
if after_perform and check_after:
9395
after_perform(metadata)
9496

9597
def fail(self, exception):

0 commit comments

Comments
 (0)