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

fix tests

Browse Source
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() 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): class TopicList(LoginRequiredMixin, TopicMixin, ListView):
pass pass
class TopicCreate(OrgaRequiredMixin, TopicMixin, CreateView): class TopicCreate(OrgaRequiredMixin, TopicMixin, TopicFormMixin, CreateView):
model = Topic model = Topic
form_class = TopicCreateForm 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): def form_valid(self, form):
form.instance.site = get_current_site(self.request) form.instance.site = get_current_site(self.request)
return super().form_valid(form) return super().form_valid(form)
class TopicUpdate(OrgaRequiredMixin, TopicMixin, UpdateView): class TopicUpdate(OrgaRequiredMixin, TopicMixin, TopicFormMixin, UpdateView):
def get_form_class(self): def get_form_class(self):
return TopicCreateForm if self.request.user.is_superuser else TopicUpdateForm return TopicCreateForm if self.request.user.is_superuser else TopicUpdateForm