使用 django 进行条件 update_or_create

Question

测试模型

class Room(models.Model):
    """
    This stores details of rooms available
    """
    name = models.CharField(
        null=False,
        blank=False,
        max_length=100,
        help_text='Enter the name of the room'
    )
    capacity = models.PositiveIntegerField(
        null=False,
        blank=False,
        help_text='Enter the number of person\'s the room can accommodate'
    )
    created_at = models.DateTimeField(auto_now_add=True)
    modified_at = models.DateTimeField(auto_now=True)

如果模型存在并且它的 modified_at 时间小于 x 时间，我想更新模型，否则只需创建模型。 作为参考，这里是我希望 django 执行的原始 sql。

INSERT INTO room VALUES (1,'2018-04-30 18:15:32.96468+04:30','2018-04-30 18:15:32.96468+04:30','Room-A',30) ON CONFLICT(id) DO UPDATE SET capacity=10 WHERE room.modified_at < '2017-04-30 18:15:32.96468+04:30';

我也想知道我写的 SQL 查询是否是原子的

Answer 1

以下示例可能会正常工作

第一个选项

try:
    obj = Room.objects.get(
        id=id, # test with other fields if you want
    )
    if obj.modified_at < DATETIME:
        obj.capacity = 10
        obj.save()
    else:
        obj = Room.objects.create(
            # fields attributes
        )
except Room.DoesNotExist:
    obj = Room.objects.create(
        # fields attributes
    )

第二个选项

或者你可以使用 django 的 Conditional Expression 来这样做

from django.db.models import F, Case, When
import datetime

your_date = datetime.datetime.now()
condition = Case(When(modified_at__lt=your_date,then=10),default=F('capacity'))

• 我们检查modified_at是否小于your_date
• 那么这个条件的值为10，
• 否则，我们保持与F('capacity')相同的字段值

其余代码

Room.objects.update_or_create(name='new_name',
           defaults={'name':'new_name','capacity':conditition})