为什么在我的MultiValueField子类上没有调用`compress`方法?

时间:2014-03-05 23:01:52

标签: python django django-forms django-models django-multiwidget

我在Django中实现了一个自定义DurationField,它存储一个表示持续时间的整数。我还定义了一个自定义多功能小组件,它接受五个不同的数字输入(表示周,天,小时,分钟和秒的数量)。

我设法让它在管理员中工作,decompress似乎按预期工作;当我更改解释器中的值并保存它们时,它们在五个字段中正确显示。但是,compress似乎没有被调用;当我尝试更改值(或只是保存表单而不更改它们)时,我从Enter a whole number.收到IntegerField错误。似乎整个值列表(例如[u'0', u'1', u'2', u'0', u'0'])都传递给to_python

我觉得我失踪/忽略了一些小事,但我想我已经盯着我的代码太长时间才弄明白它可能是什么。

这是我的model_fields.py

from datetime import timedelta                                                   

from django.db.models import PositiveIntegerField                                
from django.forms import IntegerField, MultiWidget                               
from django.forms.fields import MultiValueField                                  
from django.forms.widgets import TextInput  # NumberInput in Django 1.6          


DURATION_FORM_FIELDS = ('weeks', 'days', 'hours', 'minutes', 'seconds')


class DurationMultiWidget(MultiWidget):                                          

    def __init__(self, attrs=None):                                              
        _widgets = tuple(                                                        
            [TextInput(attrs=attrs) for field in DURATION_FORM_FIELDS]           
        )                                                                        
        super(DurationMultiWidget, self).__init__(_widgets, attrs)               

    def decompress(self, value):                                                 

        if value:                                              
            td = timedelta(seconds=value)                                        
            return [                                                             
                getattr(td, label, 0) for label in DURATION_FORM_FIELDS          
            ]                                                                    
        return [None for field in DURATION_FORM_FIELDS]

class DurationMultiValueField(MultiValueField):                                  

    def __init__(self, *args, **kwargs):                                         
        fields = tuple(                                                          
            [IntegerField(label=field_label) for field_label in DURATION_FORM_FIELDS]
        )                                                                        
        super(DurationMultiValueField, self).__init__(fields=fields, *args, **kwargs)

    def compress(self, data_list):                                                                                                                                                          
        duration_dict = dict(zip(DURATION_FORM_FIELDS, data_list))               
        timedelta_object = timedelta(**duration_dict)                            
        return int(timedelta_object.total_seconds())                            


class DurationField(PositiveIntegerField):                                       

    def get_internal_type(self):                                                 
        return 'DurationField'                                                   

    def formfield(self, **kwargs):                                               
        defaults = {'form_class': DurationMultiValueField}                       
        defaults.update(kwargs)                                                  
        return super(DurationField, self).formfield(**kwargs)                   


from south.modelsinspector import add_introspection_rules                        
add_introspection_rules([], ["^polls\.model_fields\.DurationField"])

这是admin.py

from django import forms
from django.contrib import admin

from .models import Poll
from .model_fields import DurationMultiWidget


class PollAdminForm(forms.ModelForm):                                                                                                                                                                                                                                     

    class Meta:                                                                  
        model = Poll                                                             
        widgets = {                                                             
            'limit': DurationMultiWidget(),                                      
        }                                                                        

    ...                                                                       


class PollAdmin(admin.ModelAdmin):                                               
    form = PollAdminForm


admin.site.register(Poll, PollAdmin) 

这里是models.py,以防万一:

from django.db import models
from django.utils.translation import ugettext_lazy as _ 

class Poll(models.Model):                                                                                                         
    ...                                                                             
    limit = DurationField(_('limit'), default=0) 

1 个答案:

答案 0 :(得分:0)

呃,这是一个错字:在formfield()方法下,我不小心通过了**kwargs而不是**defaults

顺便说一句,在极少数情况下,有人试图做同样的事情,你必须在kwargs.pop('min_value')中进行DurationMultiValueField.__init__(因为它是一个PositiveIntegerField)。