diff --git a/sstreasury/models.py b/sstreasury/models.py index 0424706..b6b5dbe 100644 --- a/sstreasury/models.py +++ b/sstreasury/models.py @@ -160,8 +160,11 @@ class BudgetRevision(models.Model): return False def can_withdraw(self, user): - if not self.can_edit(user): + if not self.can_view(user): return False + if user != self.author and user not in self.contributors.all() and not user.groups.filter(name='Treasury').exists(): + return False + if self.state == BudgetState.AWAIT_REVIEW.value or self.state == BudgetState.ENDORSED.value: return True return False