def create_root_comment(self, referendum): """ Creates a root Comment """ root_comment = Comment.add_root(content='referendum', user_id=referendum.user.id) referendum.comment = root_comment referendum.save() return root_comment
def form_valid(self, form): """ If the form is valid, save the associated model. """ self.object = form.save(commit=False) self.object.user = self.request.user referendum_text = "%s \n %s " % (self.object.text_remove_rules, self.object.text_add_rules) root_comment = Comment.add_root(content=referendum_text, user_id=self.request.user.id) root_comment.save() self.object.comment = root_comment self.object = form.save(commit=True) self.object.title = "Referendum %i " % self.object.id self.object.save() return super(ReferendumCreateView, self).form_valid(form)