Django проверка поля модели переопределения администратора - PullRequest
Новая
       5

Django проверка поля модели переопределения администратора

0 голосов
/ 13 марта 2020

У меня есть fee DecimalField в моей модели следующим образом: 

class CardTypes(models.Model):
    class Meta:
        db_table = "card_types"
        verbose_name = 'Card Type'
        verbose_name_plural = 'Card Types'

    name = models.CharField("Name", max_length=255, null=True, blank=True)
    fee = models.DecimalField("Fee", max_digits=65, decimal_places=0, default=0)
    image_url = models.CharField("Image Url", max_length=255, null=True, blank=True)

    deleted_at = models.DateTimeField(null=True, blank=True)

    def card_type_image(self):
        if self.image_url is not None:
            return mark_safe('<img src="%s" width="130" height="90" />' % (self.image_url))
        else:
            return "No image yet"
    card_type_image.short_description = 'image'

    def fee_convert(self):
        if self.fee is not None:
            currency = Currency.objects.get(pk=1)
            return Decimal(self.fee / pow(10, currency.decimals))
        else:
            return None
    fee_convert.short_description = 'Fee'

Я установил DecimalField("Fee", max_digits=65, decimal_places=0, default=0), потому что я хочу сохранить его без десятичного знака в моей БД, но на сайте администратора я хотите взаимодействовать с ним преобразованным методом.

Например, если плата в базе данных равна 0,00005, то в админке она будет отображаться и сохраняться только со значением 5 (потому что currency.decimals = 5)

Мне удалось показать преобразованное значение и сохранить со значением 5 на стороне администратора следующим образом: 

class CardTypeAdminForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(CardTypeAdminForm, self).__init__(*args, **kwargs)
        if self.instance.fee is not None:
            try:
                self.initial['fee'] = self.instance.fee_convert()
            except:
                pass

    def clean_fee(self):
        if self.cleaned_data['fee'] is not None:
            currency = Currency.objects.get(pk=1)
            data = Decimal(self.cleaned_data['fee'] * pow(10, currency.decimals))
        else:
            data = None

        return data

class CardTypesAdmin(admin.ModelAdmin):
    list_display = ('name', 'card_type_image',)
    list_display_links = ('name', 'card_type_image', )

    form = CardTypeAdminForm
    fields = (
        'id',
        'name',
        'fee',
        'image_url',
        'card_type_image',
        'deleted_at',
    )

    readonly_fields = (
        'id',
        'card_type_image',
        'deleted_at'
    )

admin.site.register(CardTypes, CardTypesAdmin)

Но в случае, если я хочу ввести поле вознаграждения в админ со значением 0,005 или с десятичным знаком (должен работать, потому что в БД он будет сохранен как 500), но проверка администратором возвращает следующую ошибку проверки:

Убедитесь, что не более 0 десятичных знаков.

Поскольку поле формы администратора наследует проверку из поля модели, могу ли я ее переопределить?

1 Ответ

0 голосов
/ 13 марта 2020

Я нашел способ, создав новое поле в форме замены поля формы оплаты: 

class CardTypeAdminForm(forms.ModelForm):
    currency = Currency.objects.get(pk=1)
    fee = forms.DecimalField(required=False, decimal_places=currency.decimals)
    def __init__(self, *args, **kwargs):
        super(CardTypeAdminForm, self).__init__(*args, **kwargs)
        if self.instance.fee is not None:
            try:
                self.initial['fee'] = self.instance.fee_convert()
            except:
                pass

    def clean_fee(self):
        if self.cleaned_data['fee'] is not None:
            currency = Currency.objects.get(pk=1)
            data = Decimal(self.cleaned_data['fee'] * pow(10, currency.decimals))
        else:
            data = None

        return data
    class Meta:
        model = CardTypes
        exclude = ['fee']

class CardTypesAdmin(admin.ModelAdmin):
    list_display = ('name', 'card_type_image',)
    list_display_links = ('name', 'card_type_image', )

    form = CardTypeAdminForm
    fields = (
        'id',
        'name',
        'fee',
        'image_url',
        'card_type_image',
        'deleted_at',
    )

    readonly_fields = (
        'id',
        'card_type_image',
        'deleted_at'
    )

    def save_model(self, request, obj, form, change):
        obj.fee = form.cleaned_data['fee']
        return super(CardTypesAdmin, self).save_model(request, obj, form, change)
...