forked from mirrors/bookwyrm
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
""" using django model forms """
|
|
from django import forms
|
|
from django.forms import widgets
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from bookwyrm import models
|
|
from bookwyrm.models.user import FeedFilterChoices
|
|
from .custom_form import CustomForm
|
|
|
|
|
|
# pylint: disable=missing-class-docstring
|
|
class FeedStatusTypesForm(CustomForm):
|
|
class Meta:
|
|
model = models.User
|
|
fields = ["feed_status_types"]
|
|
help_texts = {f: None for f in fields}
|
|
widgets = {
|
|
"feed_status_types": widgets.CheckboxSelectMultiple(
|
|
choices=FeedFilterChoices,
|
|
),
|
|
}
|
|
|
|
|
|
class ImportForm(forms.Form):
|
|
csv_file = forms.FileField()
|
|
|
|
|
|
class ShelfForm(CustomForm):
|
|
class Meta:
|
|
model = models.Shelf
|
|
fields = ["user", "name", "privacy", "description"]
|
|
|
|
|
|
class GoalForm(CustomForm):
|
|
class Meta:
|
|
model = models.AnnualGoal
|
|
fields = ["user", "year", "goal", "privacy"]
|
|
|
|
|
|
class ReportForm(CustomForm):
|
|
class Meta:
|
|
model = models.Report
|
|
fields = ["user", "reporter", "status", "links", "note"]
|
|
|
|
|
|
class ReadThroughForm(CustomForm):
|
|
def clean(self):
|
|
"""don't let readthroughs end before they start"""
|
|
cleaned_data = super().clean()
|
|
start_date = cleaned_data.get("start_date")
|
|
finish_date = cleaned_data.get("finish_date")
|
|
if start_date and finish_date and start_date > finish_date:
|
|
self.add_error(
|
|
"finish_date", _("Reading finish date cannot be before start date.")
|
|
)
|
|
|
|
class Meta:
|
|
model = models.ReadThrough
|
|
fields = ["user", "book", "start_date", "finish_date"]
|