Commit 5ad748c4bde3455d88aac7b60721da5a0c6fee1d

Authored by Felipe Henrique de Almeida Bormann
1 parent 98ab71bc

fixed replicate subject on category

Showing 1 changed file with 7 additions and 3 deletions   Show diff stats
subjects/views.py
@@ -158,16 +158,20 @@ class SubjectCreateView(CreateView): @@ -158,16 +158,20 @@ class SubjectCreateView(CreateView):
158 def form_valid(self, form): 158 def form_valid(self, form):
159 159
160 self.object = form.save() 160 self.object = form.save()
161 - self.object.category = Category.objects.get(slug=self.kwargs['slug']) 161 + if self.kwargs.get('slug'):
  162 + self.object.category = Category.objects.get(slug=self.kwargs['slug'])
  163 + if self.kwargs.get('subject_slug'):
  164 + subject = get_object_or_404(Subject, slug = self.kwargs['subject_slug'])
  165 + self.object.category = subject.category
162 self.object.save() 166 self.object.save()
163 167
164 168
165 return super(SubjectCreateView, self).form_valid(form) 169 return super(SubjectCreateView, self).form_valid(form)
166 170
167 def get_success_url(self): 171 def get_success_url(self):
168 - objeto = self.object.name  
169 172
170 - messages.success(self.request, _('Subject "%s" was registered on "%s" successfully!')%(objeto, self.kwargs['slug'])) 173 +
  174 + messages.success(self.request, _('Subject "%s" was registered on "%s" successfully!')%(self.object.name, self.object.category.name ))
171 return reverse_lazy('subjects:index') 175 return reverse_lazy('subjects:index')
172 176
173 177