class InvestigationCreateView(CreateView):
form_class = InvestigationForm
template_name = 'hse/investigation/investigation_create.html'
success_url = reverse_lazy('hse-incidents')
def get_context_data(self, **kwargs):
context = super(InvestigationCreateView, self).get_context_data(**kwargs)
id = self.kwargs['pk']
incident = Incident.objects.get(id=id)
context['incident'] = incident
return context
def form_valid(self, form):
self.object = form.save(commit=False)
id = self.kwargs['pk']
incident = Incident.objects.get(id=id)
self.object.incident = incident
self.object.save()
return HttpResponseRedirect(self.get_success_url())
在上面的代码中,语句id = self.kwargs['pk']和event = Incident.objects.get(id=id)在get_context_data()和form_valid()方法中重复
我怎样才能避免在 python 中编写这样的重复。
我尝试在同一个类中创建一个单独的方法,该方法将在两种方法中调用,但这对我不起作用。