Django Orm Query: How To Swap Value Of A Attribute?

Here is the case. I want to swap value of user. user_with_apple = User.objects.filter(fruit=apple) user_with_pear = User.objects.filter(fruit=pear) user_with_apple.update(fruit=pe

Solution 1:

Variant 1

I suppose you have CharField for fruit. Then add unknown option to choice.

FRUIT_CHOICES = (
    ...
    (unknown, 'Unknown'),
    (apple, 'Apple'),
    (pear, 'Pear'),
    ...
)

class User(models.Model):
    ...
    fruit = models.CharField(..., choices=FRUIT_CHOICES)
    ...

Then try this.

User.objects.filter(fruit=apple).update(fruit=unknown)
User.objects.filter(fruit=pear).update(fruit=apple)
User.objects.filter(fruit=unknown).update(fruit=pear)

Variant 2

Add access null values for fruit field.

FRUIT_CHOICES = (
    ...
    (apple, 'Apple'),
    (pear, 'Pear'),
    ...
)

class User(models.Model):
    ...
    fruit = models.CharField(..., blank=True, null=True, choices=FRUIT_CHOICES)
    ...

User.objects.filter(fruit=apple).update(fruit=None)
User.objects.filter(fruit=pear).update(fruit=apple)
User.objects.filter(fruit=None).update(fruit=pear)

Solution 2:

Here is what that comes to mind:

If you call list on your QuerySets, you force Django to give you the objects. The problem with this is that you can't call update on a list, so you need to iterate over the lists, and update each object.

This may help you on your way:

users_with_apple = list(User.objects.filter(fruit=apple))
users_with_pear = list(User.objects.filter(fruit=pear))

for user in users_with_apple:
    user.fruit = pear
    user.save()
for user in users_with_pear:
    user.fruit = apple
    user.save()