Tymczasowo wyłącz auto_now / auto_now_add

Mam taki model:

class FooBar(models.Model):
    createtime = models.DateTimeField(auto_now_add=True)
    lastupdatetime = models.DateTimeField(auto_now=True)

Chcę nadpisać dwa pola daty dla niektórych wystąpień modelu (używane podczas migracji danych). Obecne rozwiązanie wygląda następująco:

for field in new_entry._meta.local_fields:
    if field.name == "lastupdatetime":
        field.auto_now = False
    elif field.name == "createtime":
        field.auto_now_add = False

new_entry.createtime = date
new_entry.lastupdatetime = date
new_entry.save()

for field in new_entry._meta.local_fields:
    if field.name == "lastupdatetime":
        field.auto_now = True
    elif field.name == "createtime":
        field.auto_now_add = True

Czy jest lepsze rozwiązanie?

Niedawno spotkałem się z taką sytuacją podczas testowania mojej aplikacji. Musiałem „wymusić” wygasłą sygnaturę czasową. W moim przypadku załatwiłem sprawę, używając aktualizacji zestawu zapytań. Lubię to:

# my model
class FooBar(models.Model):
    title = models.CharField(max_length=255)
    updated_at = models.DateTimeField(auto_now=True, auto_now_add=True)



# my tests
foo = FooBar.objects.get(pk=1)

# force a timestamp
lastweek = datetime.datetime.now() - datetime.timedelta(days=7)
FooBar.objects.filter(pk=foo.pk).update(updated_at=lastweek)

# do the testing.

Naprawdę nie możesz wyłączyć auto_now / auto_now_add w inny sposób niż już to robisz. Jeśli potrzebujesz elastyczności, aby zmienić te wartości, auto_now/ auto_now_addnie jest najlepszym wyborem. Często bardziej elastyczne jest użycie defaulti / lub nadpisanie save()metody wykonywania manipulacji tuż przed zapisaniem obiektu.

Używając defaulti save()metody zastępowanej , jednym ze sposobów rozwiązania problemu byłoby zdefiniowanie modelu w następujący sposób:

class FooBar(models.Model):
    createtime = models.DateTimeField(default=datetime.datetime.now)
    lastupdatetime = models.DateTimeField()

    def save(self, *args, **kwargs):
        if not kwargs.pop('skip_lastupdatetime', False):
            self.lastupdatetime = datetime.datetime.now()

        super(FooBar, self).save(*args, **kwargs)

W swoim kodzie, w którym chcesz pominąć automatyczną zmianę lastupdatetime, po prostu użyj

new_entry.save(skip_lastupdatetime=True)

Jeśli twój obiekt jest zapisany w interfejsie administratora lub w innym miejscu, metoda save () zostanie wywołana bez argumentu skip_lastupdatetime i będzie zachowywać się tak samo, jak wcześniej auto_now.

Skorzystałem z sugestii pytającego i stworzyłem kilka funkcji. Oto przypadek użycia:

turn_off_auto_now(FooBar, "lastupdatetime")
turn_off_auto_now_add(FooBar, "createtime")

new_entry.createtime = date
new_entry.lastupdatetime = date
new_entry.save()

Oto implementacja:

def turn_off_auto_now(ModelClass, field_name):
    def auto_now_off(field):
        field.auto_now = False
    do_to_model(ModelClass, field_name, auto_now_off)

def turn_off_auto_now_add(ModelClass, field_name):
    def auto_now_add_off(field):
        field.auto_now_add = False
    do_to_model(ModelClass, field_name, auto_now_add_off)

def do_to_model(ModelClass, field_name, func):
    field = ModelClass._meta.get_field_by_name(field_name)[0]
    func(field)

Można utworzyć podobne funkcje, aby je ponownie włączyć.

Możesz również użyć update_fieldsparametru save()i przekazać swoje auto_nowpola. Oto przykład:

# Date you want to force
new_created_date = date(year=2019, month=1, day=1)
# The `created` field is `auto_now` in your model
instance.created = new_created_date
instance.save(update_fields=['created'])

Oto wyjaśnienie z dokumentacji Django: https://docs.djangoproject.com/en/stable/ref/models/instances/#specifying-which-fields-to-save

Poszedłem na ścieżkę menedżera kontekstu w celu ponownego wykorzystania.

@contextlib.contextmanager
def suppress_autotime(model, fields):
    _original_values = {}
    for field in model._meta.local_fields:
        if field.name in fields:
            _original_values[field.name] = {
                'auto_now': field.auto_now,
                'auto_now_add': field.auto_now_add,
            }
            field.auto_now = False
            field.auto_now_add = False
    try:
        yield
    finally:
        for field in model._meta.local_fields:
            if field.name in fields:
                field.auto_now = _original_values[field.name]['auto_now']
                field.auto_now_add = _original_values[field.name]['auto_now_add']

Użyj w ten sposób:

with suppress_autotime(my_object, ['updated']):
    my_object.some_field = some_value
    my_object.save()

Bum.

Dla tych, którzy patrzą na to podczas pisania testów, dostępna jest biblioteka Pythona o nazwie frozenegun, która pozwala sfałszować czas - więc kiedy auto_now_addkod działa, otrzymuje czas, którego faktycznie chcesz. Więc:

from datetime import datetime, timedelta
from freezegun import freeze_time

with freeze_time('2016-10-10'):
    new_entry = FooBar.objects.create(...)
with freeze_time('2016-10-17'):
    # use new_entry as you wish, as though it was created 7 days ago

Może być również używany jako dekorator - zobacz powyższy link do podstawowych dokumentów.

Możesz nadpisać auto_now_addbez specjalnego kodu.

Na to pytanie natrafiłem, gdy próbowałem stworzyć obiekt z określoną datą:

Post.objects.create(publication_date=date, ...)

gdzie publication_date = models.DateField(auto_now_add=True).

Oto co zrobiłem:

post = Post.objects.create(...)
post.publication_date = date
post.save()

To zostało pomyślnie zastąpione auto_now_add.

Bardziej długoterminowym rozwiązaniem savejest metoda zastępowania : https://code.djangoproject.com/ticket/16583

Musiałem wyłączyć auto_now dla pola DateTime podczas migracji i mogłem to zrobić.

events = Events.objects.all()
for event in events:
    for field in event._meta.fields:
        if field.name == 'created_date':
            field.auto_now = False
    event.save()

Spóźniłem się na imprezę, ale podobnie jak kilka innych odpowiedzi, jest to rozwiązanie, z którego korzystałem podczas migracji bazy danych. Różnica w stosunku do innych odpowiedzi polega na tym, że powoduje to wyłączenie wszystkich pól auto_now dla modelu przy założeniu, że nie ma powodu, aby mieć więcej niż jedną taką zmienną.

def disable_auto_now_fields(*models):
    """Turns off the auto_now and auto_now_add attributes on a Model's fields,
    so that an instance of the Model can be saved with a custom value.
    """
    for model in models:
        for field in model._meta.local_fields:
            if hasattr(field, 'auto_now'):
                field.auto_now = False
            if hasattr(field, 'auto_now_add'):
                field.auto_now_add = False

Następnie, aby go użyć, możesz po prostu zrobić:

disable_auto_now_fields(Document, Event, ...)

I przejdzie przez wszystkie twoje pola auto_nowi auto_now_addwszystkie pola dla wszystkich klas modeli, w których zdasz.

Nieco bardziej przejrzysta wersja menedżera kontekstu z https://stackoverflow.com/a/35943149/1731460

from contextlib import contextmanager

@contextmanager
def suppress_auto_now(model, field_names):
    """
    idea taken here https://stackoverflow.com/a/35943149/1731460
    """
    fields_state = {}
    for field_name in field_names:
        field = model._meta.get_field(field_name)
        fields_state[field] = {'auto_now': field.auto_now, 'auto_now_add': field.auto_now_add}

    for field in fields_state:
        field.auto_now = False
        field.auto_now_add = False
    try:
        yield
    finally:
        for field, state in fields_state.items():
            field.auto_now = state['auto_now']
            field.auto_now_add = state['auto_now_add']

Możesz go używać nawet z fabrykami (fabrykant)

        with suppress_autotime(Click, ['created']):
            ClickFactory.bulk_create(post=obj.post, link=obj.link, created__iter=created)

kopia Django - Models.DateTimeField - Dynamiczna zmiana wartości auto_now_add

Cóż, spędziłem dzisiejsze popołudnie, aby dowiedzieć się, a pierwszy problem dotyczy tego, jak pobrać obiekt modelu i gdzie w kodzie. Jestem w restframework w serializer.py, na przykład w __init__serializatorze nie mógł jeszcze mieć Modelu. Teraz w to_internal_value możesz pobrać klasę modelu, po pobraniu Field i zmodyfikowaniu właściwości pola jak w tym przykładzie:

class ProblemSerializer(serializers.ModelSerializer):

    def to_internal_value(self, data): 
        ModelClass = self.Meta.model
        dfil = ModelClass._meta.get_field('date_update')
        dfil.auto_now = False
        dfil.editable = True

Potrzebowałem rozwiązania, które będzie działać update_or_create, doszedłem do tego rozwiązania opartego na kodzie @andreaspelme.

Jedyną zmianą jest to, że możesz ustawić pomijanie, ustawiając zmodyfikowane pole na skipnie tylko przez faktyczne przekazanie kwarg skip_modified_updatedo metody save ().

Tylko yourmodelobject.modified='skip'i aktualizacja zostanie pominięta!

from django.db import models
from django.utils import timezone


class TimeTrackableAbstractModel(models.Model):
    created = models.DateTimeField(default=timezone.now, db_index=True)
    modified = models.DateTimeField(default=timezone.now, db_index=True)

    class Meta:
        abstract = True

    def save(self, *args, **kwargs):
        skip_modified_update = kwargs.pop('skip_modified_update', False)
        if skip_modified_update or self.modified == 'skip':
            self.modified = models.F('modified')
        else:
            self.modified = timezone.now()
        super(TimeTrackableAbstractModel, self).save(*args, **kwargs)