Question

我正在尝试制作日历应用，目前正在试图阻止人们在该时段已经很忙的情况下保存事件。我相信我的问题是modelForm没有user_id作为表单对象的一部分。如何传递user_id，以便可以使用它使用clean（）进行过滤。

我尝试使用get_form_kwargs（），但这似乎并不是正确的选择。

请您指出正确的方向

view.py

class CalendarEventAdd(LoginRequiredMixin, CreateView):
    model = Event
    form_class = EventForm
    template_name = 'calendar_create_event_form.html' 
    success_url = reverse_lazy('calendar_list')

def form_valid(self, form):
    form.instance.manage = self.request.user
    return super(CalendarEventAdd, self).form_valid(form)

def get_form_kwargs(self, *args, **kwargs):
    kwargs = super(CalendarEventAdd, self).get_form_kwargs()
    kwargs['user_id'] = self.request.user
    return kwargs

form.py

class EventForm(forms.ModelForm):

class Meta:
    model = Event
    widgets = {
        'start_time': DateInput(attrs={'type': 'datetime-local'}, format='%Y-%m-%dT%H:%M'),
        'end_time': DateInput(attrs={'type': 'datetime-local'}, format='%Y-%m-%dT%H:%M'),
        }
    fields = ['avilibility','start_time','end_time']


def __init__(self, *args, **kwargs):   
    super(EventForm, self).__init__(*args, **kwargs)
    # input_formats parses HTML5 datetime-local input to datetime field
    self.fields['start_time'].input_formats = ('%Y-%m-%dT%H:%M',)
    self.fields['end_time'].input_formats = ('%Y-%m-%dT%H:%M',)
    
    
def clean(self, *args, **kwargs):
    form_start_time = self.cleaned_data.get('start_time')
    form_end_time = self.cleaned_data.get('end_time')
    form_manage_id = self.cleaned_data.get('manage_id')
    between = Event.objects.filter(manage_id=1, end_time__gte=form_start_time, start_time__lte=form_end_time)
    if between:
        raise forms.ValidationError('Already Calendar entry for this time')
    super().clean()

model.py

class Event(models.Model):

    avilibility_choices = [
    ('Avilible', 'Avilible'),
    ('Busy', 'Busy'),
]
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    manage = models.ForeignKey(get_user_model(), on_delete=models.CASCADE)
    avilibility = models.CharField(max_length=10, choices=avilibility_choices, editable=True)
    start_time = models.DateTimeField()
    end_time = models.DateTimeField()

    def __str__(self):
        return self.avilibility + " - " + str(self.start_time) + " - " + str(self.end_time) + " - " +str(self.id)

    def get_absolute_url(self):
        return reverse('calendar_event_detail', args=[str(self.id)])

    def get_absolute_url_edit(self):
        return reverse('calendar_event_detail_update', args=[str(self.id)])

    def get_html_url(self):
        url = reverse('event_edit', args=(self.id,))
        return f'<a href="{url}"> {self.avilibility} </a>'

Answer 1

在您看来，您应该传递用户的主键，因此：

    def get_form_kwargs(self, *args, **kwargs):
        kwargs = super(CalendarEventAdd, self).get_form_kwargs()
        kwargs['user_id'] = self.request.user.pk
        return kwargs

您可以将其保存在EventForm对象中：

class EventForm(forms.ModelForm):
    
    def __init__(self, *args, user_id=None, **kwargs):   
        super(EventForm, self).__init__(*args, **kwargs)
        self.user_id = user_id
        # input_formats parses HTML5 datetime-local input to datetime field
        self.fields['start_time'].input_formats = ('%Y-%m-%dT%H:%M',)
        self.fields['end_time'].input_formats = ('%Y-%m-%dT%H:%M',)
    
    
    def clean(self, *args, **kwargs):
        form_start_time = self.cleaned_data.get('start_time')
        form_end_time = self.cleaned_data.get('end_time')
        form_manage_id = self.cleaned_data.get('manage_id')
        between = Event.objects.exclude(pk=self.instance.pk).filter(
            manage_id=self.user_id,
            end_time__gte=form_start_time,
            start_time__lte=form_end_time
        )
        if between.exists():
            raise forms.ValidationError('Already Calendar entry for this time')
        return super().clean()

.exclude(pk=self.instance.pk)将排除您正在编辑的对象，如果您以后决定不仅使用EventForm来进行 create 还要进行 update Event对象。

注意：documentation建议使用AUTH_USER_MODEL setting [Django-doc] get_user_model() [Django-doc]。这样比较安全，因为如果尚未加载身份验证应用，则设置仍可以指定模型名称。因此最好写：
from django.conf import settings

class Event(models.Model):
    # …
    manage = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.CASCADE
    )

Django createView将当前的user_id传递给Modelform

1 个答案: