mirror of https://github.com/logos-co/open-law.git
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
from flask_wtf import FlaskForm
|
|
from wtforms import StringField, SubmitField, ValidationError
|
|
from wtforms.validators import DataRequired, Length
|
|
|
|
from app import models as m, db
|
|
from app.logger import log
|
|
|
|
|
|
class BaseBookForm(FlaskForm):
|
|
label = StringField("Label", [DataRequired(), Length(6, 256)])
|
|
|
|
|
|
class CreateBookForm(BaseBookForm):
|
|
submit = SubmitField("Add new book")
|
|
|
|
|
|
class EditBookForm(BaseBookForm):
|
|
book_id = StringField("User ID", [DataRequired()])
|
|
submit = SubmitField("Edit book")
|
|
|
|
def validate_book_id(self, field):
|
|
book_id = field.data
|
|
book: m.Book = db.session.get(m.Book, book_id)
|
|
if not book or book.is_deleted:
|
|
log(log.WARNING, "Book with id [%s] not found", book_id)
|
|
raise ValidationError("Book not found")
|
|
|
|
def validate_label(self, field):
|
|
label = field.data
|
|
book_id = self.book_id.data
|
|
|
|
existing_book: m.Book = m.Book.query.filter_by(
|
|
is_deleted=False,
|
|
label=label,
|
|
).first()
|
|
if existing_book:
|
|
log(
|
|
log.WARNING, "Book with label [%s] already exists: [%s]", label, book_id
|
|
)
|
|
raise ValidationError("Book label must be unique!")
|