2021-03-08 16:49:10 +00:00
|
|
|
""" using django model forms """
|
2020-06-03 16:38:30 +00:00
|
|
|
import datetime
|
2020-09-29 17:21:10 +00:00
|
|
|
from collections import defaultdict
|
2020-06-03 16:38:30 +00:00
|
|
|
|
2020-03-23 16:40:09 +00:00
|
|
|
from django import forms
|
2020-09-29 17:21:10 +00:00
|
|
|
from django.forms import ModelForm, PasswordInput, widgets
|
|
|
|
from django.forms.widgets import Textarea
|
2020-11-28 00:24:53 +00:00
|
|
|
from django.utils import timezone
|
2021-04-04 20:22:36 +00:00
|
|
|
from django.utils.translation import gettext_lazy as _
|
2020-01-29 09:05:27 +00:00
|
|
|
|
2020-09-21 15:10:37 +00:00
|
|
|
from bookwyrm import models
|
2020-01-29 09:05:27 +00:00
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class CustomForm(ModelForm):
|
2021-03-08 16:49:10 +00:00
|
|
|
""" add css classes to the forms """
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
def __init__(self, *args, **kwargs):
|
2021-03-08 16:49:10 +00:00
|
|
|
css_classes = defaultdict(lambda: "")
|
|
|
|
css_classes["text"] = "input"
|
|
|
|
css_classes["password"] = "input"
|
|
|
|
css_classes["email"] = "input"
|
|
|
|
css_classes["number"] = "input"
|
|
|
|
css_classes["checkbox"] = "checkbox"
|
|
|
|
css_classes["textarea"] = "textarea"
|
2020-09-29 17:21:10 +00:00
|
|
|
super(CustomForm, self).__init__(*args, **kwargs)
|
|
|
|
for visible in self.visible_fields():
|
2021-03-08 16:49:10 +00:00
|
|
|
if hasattr(visible.field.widget, "input_type"):
|
2020-09-29 17:21:10 +00:00
|
|
|
input_type = visible.field.widget.input_type
|
|
|
|
if isinstance(visible.field.widget, Textarea):
|
2021-03-08 16:49:10 +00:00
|
|
|
input_type = "textarea"
|
|
|
|
visible.field.widget.attrs["cols"] = None
|
|
|
|
visible.field.widget.attrs["rows"] = None
|
|
|
|
visible.field.widget.attrs["class"] = css_classes[input_type]
|
2020-09-29 17:21:10 +00:00
|
|
|
|
2020-12-22 17:26:40 +00:00
|
|
|
|
2020-12-13 02:13:00 +00:00
|
|
|
# pylint: disable=missing-class-docstring
|
2020-09-29 17:21:10 +00:00
|
|
|
class LoginForm(CustomForm):
|
2020-01-29 09:05:27 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.User
|
2021-03-08 16:49:10 +00:00
|
|
|
fields = ["localname", "password"]
|
2020-01-29 09:05:27 +00:00
|
|
|
help_texts = {f: None for f in fields}
|
|
|
|
widgets = {
|
2021-03-08 16:49:10 +00:00
|
|
|
"password": PasswordInput(),
|
2020-01-29 09:05:27 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class RegisterForm(CustomForm):
|
2020-01-29 09:05:27 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.User
|
2021-03-08 16:49:10 +00:00
|
|
|
fields = ["localname", "email", "password"]
|
2020-01-29 09:05:27 +00:00
|
|
|
help_texts = {f: None for f in fields}
|
2021-03-08 16:49:10 +00:00
|
|
|
widgets = {"password": PasswordInput()}
|
2020-01-29 09:05:27 +00:00
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class RatingForm(CustomForm):
|
2020-04-03 19:43:49 +00:00
|
|
|
class Meta:
|
2021-01-01 19:05:49 +00:00
|
|
|
model = models.ReviewRating
|
2021-03-08 17:48:25 +00:00
|
|
|
fields = ["user", "book", "rating", "privacy"]
|
2020-04-03 19:43:49 +00:00
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class ReviewForm(CustomForm):
|
2020-01-29 09:05:27 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.Review
|
2020-12-17 04:10:50 +00:00
|
|
|
fields = [
|
2021-03-08 16:49:10 +00:00
|
|
|
"user",
|
|
|
|
"book",
|
|
|
|
"name",
|
|
|
|
"content",
|
|
|
|
"rating",
|
|
|
|
"content_warning",
|
|
|
|
"sensitive",
|
|
|
|
"privacy",
|
|
|
|
]
|
2020-01-29 09:05:27 +00:00
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class CommentForm(CustomForm):
|
2020-03-21 23:50:49 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.Comment
|
2021-03-21 00:39:05 +00:00
|
|
|
fields = [
|
|
|
|
"user",
|
|
|
|
"book",
|
|
|
|
"content",
|
|
|
|
"content_warning",
|
|
|
|
"sensitive",
|
|
|
|
"privacy",
|
|
|
|
"progress",
|
2021-03-21 01:03:20 +00:00
|
|
|
"progress_mode",
|
2021-03-21 00:39:05 +00:00
|
|
|
]
|
2020-03-21 23:50:49 +00:00
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class QuotationForm(CustomForm):
|
2020-04-08 16:40:47 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.Quotation
|
2020-12-17 04:10:50 +00:00
|
|
|
fields = [
|
2021-03-08 16:49:10 +00:00
|
|
|
"user",
|
|
|
|
"book",
|
|
|
|
"quote",
|
|
|
|
"content",
|
|
|
|
"content_warning",
|
|
|
|
"sensitive",
|
|
|
|
"privacy",
|
|
|
|
]
|
2020-04-08 16:40:47 +00:00
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class ReplyForm(CustomForm):
|
2020-02-18 05:39:08 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.Status
|
2020-12-17 04:10:50 +00:00
|
|
|
fields = [
|
2021-03-08 16:49:10 +00:00
|
|
|
"user",
|
|
|
|
"content",
|
|
|
|
"content_warning",
|
|
|
|
"sensitive",
|
|
|
|
"reply_parent",
|
|
|
|
"privacy",
|
|
|
|
]
|
|
|
|
|
2020-02-18 05:39:08 +00:00
|
|
|
|
2021-01-29 19:14:18 +00:00
|
|
|
class StatusForm(CustomForm):
|
|
|
|
class Meta:
|
|
|
|
model = models.Status
|
2021-03-08 16:49:10 +00:00
|
|
|
fields = ["user", "content", "content_warning", "sensitive", "privacy"]
|
2021-01-29 19:14:18 +00:00
|
|
|
|
2020-02-18 05:39:08 +00:00
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class EditUserForm(CustomForm):
|
2020-01-29 09:05:27 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.User
|
2021-03-18 16:06:00 +00:00
|
|
|
fields = [
|
|
|
|
"avatar",
|
|
|
|
"name",
|
|
|
|
"email",
|
|
|
|
"summary",
|
|
|
|
"show_goal",
|
2021-03-21 23:37:52 +00:00
|
|
|
"manually_approves_followers",
|
|
|
|
"discoverable",
|
2021-03-28 22:51:40 +00:00
|
|
|
"preferred_timezone",
|
2021-03-18 16:06:00 +00:00
|
|
|
]
|
2020-01-29 09:05:27 +00:00
|
|
|
help_texts = {f: None for f in fields}
|
2020-02-18 05:39:08 +00:00
|
|
|
|
2020-02-21 06:19:19 +00:00
|
|
|
|
2021-04-01 15:32:06 +00:00
|
|
|
class LimitedEditUserForm(CustomForm):
|
|
|
|
class Meta:
|
|
|
|
model = models.User
|
|
|
|
fields = [
|
|
|
|
"avatar",
|
|
|
|
"name",
|
|
|
|
"summary",
|
|
|
|
"manually_approves_followers",
|
|
|
|
"discoverable",
|
|
|
|
]
|
|
|
|
help_texts = {f: None for f in fields}
|
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class TagForm(CustomForm):
|
2020-02-21 02:01:50 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.Tag
|
2021-03-08 16:49:10 +00:00
|
|
|
fields = ["name"]
|
2020-02-21 02:01:50 +00:00
|
|
|
help_texts = {f: None for f in fields}
|
2021-03-08 16:49:10 +00:00
|
|
|
labels = {"name": "Add a tag"}
|
2020-02-21 02:01:50 +00:00
|
|
|
|
2020-03-23 16:40:09 +00:00
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class CoverForm(CustomForm):
|
2020-03-28 22:06:16 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.Book
|
2021-03-08 16:49:10 +00:00
|
|
|
fields = ["cover"]
|
2020-03-28 22:06:16 +00:00
|
|
|
help_texts = {f: None for f in fields}
|
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class EditionForm(CustomForm):
|
2020-03-28 22:06:16 +00:00
|
|
|
class Meta:
|
2020-04-02 15:44:53 +00:00
|
|
|
model = models.Edition
|
2020-03-28 22:06:16 +00:00
|
|
|
exclude = [
|
2021-03-08 16:49:10 +00:00
|
|
|
"remote_id",
|
|
|
|
"origin_id",
|
|
|
|
"created_date",
|
|
|
|
"updated_date",
|
|
|
|
"edition_rank",
|
2021-03-08 18:07:02 +00:00
|
|
|
"authors",
|
2021-03-08 16:49:10 +00:00
|
|
|
"parent_work",
|
|
|
|
"shelves",
|
|
|
|
"subjects", # TODO
|
|
|
|
"subject_places", # TODO
|
|
|
|
"connector",
|
2020-03-28 22:06:16 +00:00
|
|
|
]
|
|
|
|
|
2021-03-08 16:49:10 +00:00
|
|
|
|
2020-12-22 17:26:40 +00:00
|
|
|
class AuthorForm(CustomForm):
|
|
|
|
class Meta:
|
|
|
|
model = models.Author
|
|
|
|
exclude = [
|
2021-03-08 16:49:10 +00:00
|
|
|
"remote_id",
|
|
|
|
"origin_id",
|
|
|
|
"created_date",
|
|
|
|
"updated_date",
|
2020-12-22 17:26:40 +00:00
|
|
|
]
|
|
|
|
|
2020-03-28 22:06:16 +00:00
|
|
|
|
2020-03-23 16:40:09 +00:00
|
|
|
class ImportForm(forms.Form):
|
|
|
|
csv_file = forms.FileField()
|
2020-06-03 16:38:30 +00:00
|
|
|
|
2021-03-08 16:49:10 +00:00
|
|
|
|
2020-06-03 16:38:30 +00:00
|
|
|
class ExpiryWidget(widgets.Select):
|
|
|
|
def value_from_datadict(self, data, files, name):
|
2021-03-08 16:49:10 +00:00
|
|
|
""" human-readable exiration time buckets """
|
2020-06-03 16:38:30 +00:00
|
|
|
selected_string = super().value_from_datadict(data, files, name)
|
|
|
|
|
2021-03-08 16:49:10 +00:00
|
|
|
if selected_string == "day":
|
2020-06-03 16:38:30 +00:00
|
|
|
interval = datetime.timedelta(days=1)
|
2021-03-08 16:49:10 +00:00
|
|
|
elif selected_string == "week":
|
2020-06-03 16:38:30 +00:00
|
|
|
interval = datetime.timedelta(days=7)
|
2021-03-08 16:49:10 +00:00
|
|
|
elif selected_string == "month":
|
|
|
|
interval = datetime.timedelta(days=31) # Close enough?
|
|
|
|
elif selected_string == "forever":
|
2020-06-03 16:38:30 +00:00
|
|
|
return None
|
|
|
|
else:
|
2021-03-08 16:49:10 +00:00
|
|
|
return selected_string # "This will raise
|
2020-06-03 16:38:30 +00:00
|
|
|
|
2020-11-28 00:24:53 +00:00
|
|
|
return timezone.now() + interval
|
2020-06-03 16:38:30 +00:00
|
|
|
|
2021-03-08 16:49:10 +00:00
|
|
|
|
2021-03-21 01:23:59 +00:00
|
|
|
class InviteRequestForm(CustomForm):
|
2021-03-21 02:14:41 +00:00
|
|
|
def clean(self):
|
|
|
|
""" make sure the email isn't in use by a registered user """
|
|
|
|
cleaned_data = super().clean()
|
|
|
|
email = cleaned_data.get("email")
|
|
|
|
if email and models.User.objects.filter(email=email).exists():
|
|
|
|
self.add_error("email", _("A user with this email already exists."))
|
|
|
|
|
2021-03-21 01:23:59 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.InviteRequest
|
|
|
|
fields = ["email"]
|
|
|
|
|
|
|
|
|
2020-09-29 17:21:10 +00:00
|
|
|
class CreateInviteForm(CustomForm):
|
2020-06-03 16:38:30 +00:00
|
|
|
class Meta:
|
|
|
|
model = models.SiteInvite
|
2021-04-05 17:17:01 +00:00
|
|
|
exclude = ["code", "user", "times_used", "invitees"]
|
2020-06-03 16:38:30 +00:00
|
|
|
widgets = {
|
2021-03-08 16:49:10 +00:00
|
|
|
"expiry": ExpiryWidget(
|
|
|
|
choices=[
|
|
|
|
("day", _("One Day")),
|
|
|
|
("week", _("One Week")),
|
|
|
|
("month", _("One Month")),
|
|
|
|
("forever", _("Does Not Expire")),
|
|
|
|
]
|
|
|
|
),
|
|
|
|
"use_limit": widgets.Select(
|
|
|
|
choices=[
|
|
|
|
(i, _("%(count)d uses" % {"count": i}))
|
|
|
|
for i in [1, 5, 10, 25, 50, 100]
|
|
|
|
]
|
|
|
|
+ [(None, _("Unlimited"))]
|
|
|
|
),
|
2020-06-03 16:38:30 +00:00
|
|
|
}
|
2020-11-10 22:52:04 +00:00
|
|
|
|
2021-03-08 16:49:10 +00:00
|
|
|
|
2020-11-10 22:52:04 +00:00
|
|
|
class ShelfForm(CustomForm):
|
|
|
|
class Meta:
|
|
|
|
model = models.Shelf
|
2021-03-08 16:49:10 +00:00
|
|
|
fields = ["user", "name", "privacy"]
|
2021-01-16 16:18:54 +00:00
|
|
|
|
2021-01-29 23:38:42 +00:00
|
|
|
|
2021-01-16 16:18:54 +00:00
|
|
|
class GoalForm(CustomForm):
|
|
|
|
class Meta:
|
|
|
|
model = models.AnnualGoal
|
2021-03-08 16:49:10 +00:00
|
|
|
fields = ["user", "year", "goal", "privacy"]
|
2021-01-29 23:38:42 +00:00
|
|
|
|
|
|
|
|
|
|
|
class SiteForm(CustomForm):
|
|
|
|
class Meta:
|
|
|
|
model = models.SiteSettings
|
|
|
|
exclude = []
|
2021-01-31 16:08:52 +00:00
|
|
|
|
|
|
|
|
|
|
|
class ListForm(CustomForm):
|
|
|
|
class Meta:
|
|
|
|
model = models.List
|
2021-03-08 16:49:10 +00:00
|
|
|
fields = ["user", "name", "description", "curation", "privacy"]
|
2021-03-09 02:36:34 +00:00
|
|
|
|
|
|
|
|
|
|
|
class ReportForm(CustomForm):
|
|
|
|
class Meta:
|
|
|
|
model = models.Report
|
|
|
|
fields = ["user", "reporter", "statuses", "note"]
|