def withdraw(user_id, amount): user = User.objects.get(id=user_id) if user.balance < amount: raise InsufficientFundsError() user.balance -= amount user.save()