Переопределение ModelForm.save не обновляет все затронутые поля
Я хочу хранить денежные суммы в виде целых чисел в базе данных. Для удобства я добавил поля с плавающей точкой:
# File: models.py
class Transaction(models.Model):
user = models.ForeignKey(User, related_name='transactions')
date = models.DateTimeField(default=datetime.now)
# transacted btc-amount in satoshi; positive when I bought btc, negative else
amount_btc_satoshi = models.IntegerField()
# for convenience: transacted btc-amout in units of 1 btc
amount_btc = models.FloatField(null=True)
# transacted fiat-amount in 1e-5 euros; positive when I sold btc, negative else
amount_eur_milicent = models.IntegerField()
# for convenience: transacted fiat-amout in units of 1 eur
amount_eur = models.FloatField(null=True)
# True if I bought bitcoins, False if I sold bitcoins
is_bid = models.BooleanField()
# effective fiat price per 1 BTC in EUR
price_per_btc = models.FloatField()
Для удобства я переопределил метод save в моем производном ModelForm. Предполагается автоматически обновлять некоторые зависимые поля на основе amount_btc
а также amount_eur
:
# File: forms.py
class TransactionForm(forms.ModelForm):
def clean(self):
cleaned_data = super(TransactionForm, self).clean()
if cleaned_data['amount_btc'] > 0. and cleaned_data['amount_eur'] > 0.:
raise forms.ValidationError('Either BTC amount or fiat amount must be negative.')
return cleaned_data
def save(self, commit=True, *args, **kwargs):
instance = super(TransactionForm, self).save(commit=False, *args, **kwargs)
# store input data in integer format
instance.amount_btc_satoshi = int(round(self.cleaned_data['amount_btc'] * 1e8))
instance.amount_eur_milicent = int(round(self.cleaned_data['amount_eur'] * 1e5))
# provide convenient amounts
instance.amount_btc = instance.amount_btc_satoshi / 1e8
instance.amount_eur = instance.amount_eur_milicent / 1e5
instance.is_bid = instance.amount_btc_satoshi > 0
instance.price_per_btc = -1. * instance.amount_eur / instance.amount_btc
if commit:
instance.save()
return instance
class Meta:
model = Transaction
fields = ['date', 'amount_btc', 'amount_eur']
Теперь добавление нового transaction
работает как положено с is_bid
и другие зависимые поля правильно установлены. Однако редактирование существующей записи приводит к обновлению только одного поля. Например, ни price_per_btc
ни amount_eur_milicent
меняется, даже если amount_eur
есть (см. ниже):
# File: views.py
@login_required
def transaction_add(request):
form = TransactionForm(request.POST)
if form.is_valid():
transaction = form.save(commit=False)
transaction.user = request.user
transaction.save()
else:
messages.error(request, ';'.join('{}: {}'.format(key, value) for key, value in form.errors.items()))
return redirect(request.POST['next'])
@login_required
def transaction_edit(request, id):
transaction = Transaction.objects.get(id=id)
form = TransactionForm(request.POST, instance=transaction)
if form.is_valid():
transaction.save()
else:
messages.error(request, ';'.join('{}: {}'.format(key, value) for key, value in form.errors.items()))
return redirect(request.POST['next'])
Я старался instance.save(update_fields=None)
, но это никак не повлияло. Идеи?
1 ответ
Экземпляр транзакции не будет изменен таким образом. Вы должны явно вызывать form.save() (потому что форма на самом деле содержит измененные данные)
transaction = Transaction.objects.get(id=id)
form = TransactionForm(request.POST, instance=transaction)
if form.is_valid():
form.save()
Вы не будете создавать новый объект, просто отредактируйте существующий, так как первичный ключ объекта не изменился