Блог

Я создал набор форм, позволяющий пользователям еженедельно регистрировать часы работы. Проблема, с которой я сталкиваюсь, заключается в том, что я не могу сохранить «0» в полях ввода — любая десятичная дробь работает, кроме 0 (см. gif в конце для иллюстрации).

TLDR: Набор форм сохраняет любые входные данные, кроме 0, без понятия, почему. Смотрите gif здесь для иллюстрации: https://imgur.com/a/iCMexQk

Мои Timesheet и TimesheetEntry модели выглядят следующим образом:

 
class Timesheet(Model):

    year = PositiveIntegerField(validators=[MaxValueValidator(2500), MinValueValidator(1900)])
    week = PositiveIntegerField()
    project = ForeignKey("projects.Project", on_delete=CASCADE)
    user = ForeignKey(User, on_delete=CASCADE)
    status = CharField(
        max_length=15, choices=STATUS_CHOICES, blank=True, default=STATUS_OPEN
    )

    objects = TimesheetManager()

    class Meta:
        db_table = 'timesheet'
        ordering = ('id',)
        unique_together = (('year', 'week', 'user', 'project',),)

    def __getattr__(self, attr):
        allowed_days = (
            'day_1', 'day_2', 'day_3', 'day_4', 'day_5', 'day_6', 'day_7'
        )
        if attr in allowed_days:
            day = int(attr[-1]) - 1
            entry = self.timesheetentry_set.filter(
                date=Week(self.year, self.week).day(day)
            ).first()
            if entry:
                return entry.hours
            return None
        return super().__getattr__(attr)

    def total_duration(self):
        return self.timesheetentry_set.aggregate(
            total_duration=Coalesce(Sum('hours'), 0)
        ).get('total_duration')


class TimesheetEntry(Model):
    timesheet = ForeignKey(Timesheet, on_delete=CASCADE)
    hours = DecimalField(max_digits=4, decimal_places=1, null=True, blank=True)
    date = DateField()

    class Meta:
        db_table = 'timesheet_entry'
        ordering = ('date',)

 

И forms.py выглядит следующим образом:

 
DAYS = (
    'day_1', 'day_2', 'day_3', 'day_4', 'day_5', 'day_6', 'day_7'
)

class TimesheetModelForm(ModelForm):

    class Meta:
        model = Timesheet
        exclude = ("user", "status")

class BaseFormSetValidation(BaseModelFormSet):

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        for form in self.forms:
            for day in DAYS:
                form.fields[day].initial = getattr(form.instance, day)
       


    def add_fields(self, form, index):
        super().add_fields(form, index)
        for day in DAYS:
            form.fields[day] = DecimalField(required=False)

    def save(self, commit=True):
        super(BaseFormSetValidation, self).save(commit)
        from isoweek import Week
        for form in self.forms:
            week = Week(form.instance.year, form.instance.week)
            for day in DAYS:
                if form.cleaned_data.get(day):
                    date = week.day(int(day[-1]) - 1)
    
                    TimesheetEntry.objects.update_or_create(
                        timesheet=form.instance,
                        date=date,
                        defaults={'hours': form.cleaned_data.get(day)}
                    )



TimesheetModelFormSet = modelformset_factory(
    Timesheet,
    formset=BaseFormSetValidation,
    exclude=("year", "week", "project", "user"),
    extra=0,
)
 

views.py

 class TimesheetEditorView(BaseTimesheet, TemplateView):
    form_class = TimesheetModelFormSet
    template_name = "timesheets/timesheet.html"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        # only show timesheet rows that belongs to logged in user
        timesheet = Timesheet.objects.filter(
            year=context["year"],
            week=context["week"],
            user=self.request.user
        ).order_by("project_id")

        timesheet_formset = self.form_class(queryset=timesheet)
        create_timesheet_form = TimesheetModelForm(self.request.user)

        context.update(
            timesheet=Timesheet.objects.none(),
            timesheet_formset=timesheet_formset,
            create_timesheet_form=create_timesheet_form
        )
        return context


    def post(self, request, *args, **kwargs):
        context = self.get_context_data(**kwargs)
        timesheet_formset = self.form_class(request.POST, error_class=DivErrorList)
        context.update(timesheet_formset=timesheet_formset)

        if timesheet_formset.is_valid():
            timesheet_formset.save()
            message = "Weekly timesheet (is_valid())"

            messages.info(request, message, extra_tags='timesheet')
            success_url = reverse("timesheets:current-week", args=(context["year"], context["week"]))
            return HttpResponseRedirect(success_url)
        else:
            print(timesheet_formset.errors)
            print(timesheet_formset.non_form_errors)

        return render(request, "timesheets/timesheet.html", context)
 

timesheets.html

 
<tbody>

                        {{ timesheet_formset.management_form }}
                        {% for row in timesheet_formset %}
                        
                        <tr class="tracker">
                            <th scope="row" class="align-middle">{{ row.instance.project }} {{ row.id }}</th>
                            <td>{{ row.day_1 }}</td>
                            <td>{{ row.day_2 }}</td>
                            <td>{{ row.day_3 }}</td>
                            <td>{{ row.day_4 }}</td>
                            <td>{{ row.day_5 }}</td>
                            <td>{{ row.day_6 }}</td>
                            <td>{{ row.day_7 }}</td>
                            <td class="align-middle">{{ row.instance.total_duration }}</td>
 
                        </tr>
                        {% endfor %}
                    </tbody>
 

Любой десятичный ввод работает, кроме 0 — форма сохраняется и is_valid() запускается, но по какой-то причине форма обрабатывает 0 как None и неправильно сохраняет.

Я прикрепил gif, чтобы проиллюстрировать проблему — обратите внимание, как работает любой десятичный ввод, но когда я ввожу «0», он просто возвращается к последнему значению:

https://imgur.com/a/iCMexQk

Есть какие-нибудь идеи о том, почему это происходит?

В вашем методе save() BaseFormSetValidation вы проверяете ценность:

 if form.cleaned_data.get(day):
 

Значение 0 является ложным, поэтому условие if будет False выполнено, а условие не будет выполнено.

None , 0 , False являются ложными значениями.

Вместо этого вы можете явно проверить наличие None :

 class BaseFormSetValidation(BaseModelFormSet):

    ...

    def save(self, commit=True):
        super(BaseFormSetValidation, self).save(commit)
        from isoweek import Week
        for form in self.forms:
            week = Week(form.instance.year, form.instance.week)
            for day in DAYS:
                if form.cleaned_data.get(day) is not None:
                    date = week.day(int(day[-1]) - 1)
    
                    TimesheetEntry.objects.update_or_create(
                        timesheet=form.instance,
                        date=date,
                        defaults={'hours': form.cleaned_data.get(day)}
                    )