0
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 中编写这样的重复。

我尝试在同一个类中创建一个单独的方法,该方法将在两种方法中调用,但这对我不起作用。

4

0 回答 0