sonny/newsreader
1
0
Fork 0
Code Issues 13 Pull requests 2 Projects Releases 58 Packages Wiki Activity Actions

Move widget to forms.py

Browse source
This commit is contained in:
sonny 2020-05-21 17:49:53 +02:00
parent af8403f7ce
commit ad74254593
2 changed files with 14 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/newsreader/news/core/forms.py
View file

@ -1,8 +1,21 @@
from django import forms from django import forms
from django.forms.widgets import CheckboxSelectMultiple
from newsreader.news.collection.models import CollectionRule from newsreader.news.collection.models import CollectionRule
from newsreader.news.core.models import Category from newsreader.news.core.models import Category
from newsreader.news.core.widgets import RulesWidget
class RulesWidget(CheckboxSelectMultiple):
template_name = "news/core/widgets/rules.html"
option_template_name = "news/core/widgets/rule.html"
def create_option(self, *args, **kwargs):
option = super().create_option(*args, **kwargs)
instance = self.choices.queryset.get(pk=option["value"])
if self.category and instance.category:
option["selected"] = self.category.pk == instance.category.pk
return {**option, "instance": instance}
class CategoryForm(forms.ModelForm): class CategoryForm(forms.ModelForm):
@ -16,7 +29,6 @@ class CategoryForm(forms.ModelForm):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.fields["rules"].queryset = CollectionRule.objects.filter(user=self.user) self.fields["rules"].queryset = CollectionRule.objects.filter(user=self.user)
self.fields["rules"].widget.category = self.instance self.fields["rules"].widget.category = self.instance
def save(self, commit=True): def save(self, commit=True):

14
src/newsreader/news/core/widgets.py
View file

@ -1,14 +0,0 @@
from django.forms.widgets import CheckboxSelectMultiple
class RulesWidget(CheckboxSelectMultiple):
template_name = "news/core/widgets/rules.html"
option_template_name = "news/core/widgets/rule.html"
def create_option(self, *args, **kwargs):
option = super().create_option(*args, **kwargs)
instance = self.choices.queryset.get(pk=option["value"])
if self.category and instance.category:
option["selected"] = self.category.pk == instance.category.pk
return {**option, "instance": instance}