diff --git a/engine/core/models.py b/engine/core/models.py index 668e741d..46065a5a 100644 --- a/engine/core/models.py +++ b/engine/core/models.py @@ -1937,8 +1937,12 @@ class OrderProduct(ExportModelOperationsMixin("order_product"), NiceModel): ] def return_balance_back(self): + if self.status == "RETURNED": + return self.status = "RETURNED" self.save() + if not self.order.user: + raise NotImplementedError self.order.user.payments_balance.amount += self.buy_price self.order.user.payments_balance.save()