Download as
Browse Code »

Commit 5ad748c4bde3455d88aac7b60721da5a0c6fee1d

Authored by Felipe Henrique de Almeida Bormann
1 parent 98ab71bc
Exists in master and in 3 other branches amadeus_univasf, image-crop, refactoring

fixed replicate subject on category

Showing 1 changed file with 7 additions and 3 deletions   Show diff stats
subjects/views.py
  Diff comments   View file @5ad748c
... ... @@ -158,16 +158,20 @@ class SubjectCreateView(CreateView):
158 158 def form_valid(self, form):
159 159  
160 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 166 self.object.save()
163 167  
164 168  
165 169 return super(SubjectCreateView, self).form_valid(form)
166 170  
167 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 175 return reverse_lazy('subjects:index')
172 176  
173 177  
... ...