AFPy/PonyConf
AFPy
/
PonyConf
7
2
Fork 3
Code Issues 10 Pull Requests 1 Projects Releases Wiki Activity

fix tests

This commit is contained in:
Élie Bouttier 2016-07-12 00:50:23 +02:00
parent a1c0dce757
commit 731bda30ef
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
proposals/views.py
View File

@ -85,25 +85,27 @@ class TopicMixin(object):
return Topic.objects.filter(site=get_current_site(self.request)).all()
class TopicFormMixin(object):
def get_form_kwargs(self):
kwargs = super(TopicFormMixin, self).get_form_kwargs()
kwargs.update({'site_id': get_current_site(self.request).id})
return kwargs
class TopicList(LoginRequiredMixin, TopicMixin, ListView):
pass
class TopicCreate(OrgaRequiredMixin, TopicMixin, CreateView):
class TopicCreate(OrgaRequiredMixin, TopicMixin, TopicFormMixin, CreateView):
model = Topic
form_class = TopicCreateForm
def get_form_kwargs(self):
kwargs = super(TopicCreate, self).get_form_kwargs()
kwargs.update({'site_id': get_current_site(self.request).id})
return kwargs
def form_valid(self, form):
form.instance.site = get_current_site(self.request)
return super().form_valid(form)
class TopicUpdate(OrgaRequiredMixin, TopicMixin, UpdateView):
class TopicUpdate(OrgaRequiredMixin, TopicMixin, TopicFormMixin, UpdateView):
def get_form_class(self):
return TopicCreateForm if self.request.user.is_superuser else TopicUpdateForm