mirror of https://github.com/logos-co/open-law.git
99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
from flask_login import current_user
|
|
from flask import Response, flash, redirect, url_for, request
|
|
|
|
from app import models as m, db
|
|
from app.logger import log
|
|
|
|
|
|
class BookRouteVerifier:
|
|
_routes = []
|
|
|
|
@classmethod
|
|
def add_route(cls, route_name: str):
|
|
cls._routes.append(route_name)
|
|
|
|
@classmethod
|
|
def remove_route(cls, route_name: str):
|
|
cls._routes.remove(route_name)
|
|
|
|
@classmethod
|
|
def is_present(cls, route_name: str) -> bool:
|
|
return route_name in cls._routes
|
|
|
|
|
|
def register_book_verify_route(blueprint_name: str):
|
|
def decorator(func: callable):
|
|
BookRouteVerifier.add_route(f"{blueprint_name}.{func.__name__}")
|
|
return func
|
|
|
|
return decorator
|
|
|
|
|
|
def book_validator() -> Response | None:
|
|
if not BookRouteVerifier.is_present(request.endpoint):
|
|
return None
|
|
|
|
request_args = (
|
|
{**request.view_args, **request.args} if request.view_args else {**request.args}
|
|
)
|
|
|
|
book_id = request_args.get("book_id")
|
|
if book_id:
|
|
book: m.Book = db.session.get(m.Book, book_id)
|
|
if not book or book.is_deleted:
|
|
log(log.INFO, "User: [%s] is not owner of book: [%s]", current_user, book)
|
|
flash("Book not found!", "danger")
|
|
return redirect(url_for("book.my_library"))
|
|
|
|
collection_id = request_args.get("collection_id")
|
|
if collection_id is not None:
|
|
collection: m.Collection = db.session.get(m.Collection, collection_id)
|
|
if not collection or collection.is_deleted:
|
|
log(log.WARNING, "Collection with id [%s] not found", collection_id)
|
|
flash("Collection not found", "danger")
|
|
return redirect(url_for("book.collection_view", book_id=book_id))
|
|
|
|
sub_collection_id = request_args.get("sub_collection_id")
|
|
if sub_collection_id is not None:
|
|
sub_collection: m.Collection = db.session.get(m.Collection, sub_collection_id)
|
|
if not sub_collection or sub_collection.is_deleted:
|
|
log(log.WARNING, "Sub_collection with id [%s] not found", sub_collection_id)
|
|
flash("Subcollection not found", "danger")
|
|
return redirect(
|
|
url_for(
|
|
"book.collection_view", book_id=book_id, collection_id=collection_id
|
|
)
|
|
)
|
|
|
|
section_id = request_args.get("section_id")
|
|
if section_id is not None:
|
|
section: m.Section = db.session.get(m.Section, section_id)
|
|
if not section:
|
|
log(log.WARNING, "Section with id [%s] not found", section)
|
|
flash("Section not found", "danger")
|
|
return redirect(url_for("book.collection_view", book_id=book_id))
|
|
|
|
interpretation_id = request_args.get("interpretation_id")
|
|
if interpretation_id is not None:
|
|
interpretation: m.Interpretation = db.session.get(
|
|
m.Interpretation, interpretation_id
|
|
)
|
|
if not interpretation or interpretation.is_deleted:
|
|
log(log.WARNING, "Interpretation with id [%s] not found", interpretation_id)
|
|
flash("Interpretation not found", "danger")
|
|
return redirect(
|
|
url_for(
|
|
"book.qa_view", book_id=book_id, interpretation_id=interpretation_id
|
|
)
|
|
)
|
|
|
|
comment_id = request_args.get("comment_id")
|
|
if comment_id is not None:
|
|
comment: m.Comment = db.session.get(m.Comment, comment_id)
|
|
if not comment or comment.is_deleted:
|
|
log(log.WARNING, "Comment with id [%s] not found", comment_id)
|
|
flash("Comment not found", "danger")
|
|
return redirect(
|
|
url_for("book.qa_view", interpretation_id=interpretation_id)
|
|
)
|