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.forms.widgets import CheckboxSelectMultiple
from newsreader.news.collection.models import CollectionRule
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):
@ -16,7 +29,6 @@ class CategoryForm(forms.ModelForm):
super().__init__(*args, **kwargs)
self.fields["rules"].queryset = CollectionRule.objects.filter(user=self.user)
self.fields["rules"].widget.category = self.instance
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}