mirror of https://github.com/logos-co/open-law.git
97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
from flask import flash, redirect, url_for
|
|
|
|
from app import forms as f, models as m, db
|
|
from app.logger import log
|
|
|
|
|
|
def add_contributor_to_book(
|
|
form: f.AddContributorForm,
|
|
book_id: int,
|
|
selected_tab: str = "",
|
|
user_id: int = None,
|
|
):
|
|
if not user_id:
|
|
user_id = form.user_id.data
|
|
book_contributor = m.BookContributor.query.filter_by(
|
|
user_id=user_id, book_id=book_id
|
|
).first()
|
|
if book_contributor:
|
|
log(log.INFO, "Contributor: [%s] already exists", book_contributor)
|
|
flash("Already exists!", "danger")
|
|
return redirect(
|
|
url_for("book.settings", selected_tab=selected_tab, book_id=book_id)
|
|
)
|
|
|
|
role = m.BookContributor.Roles(int(form.role.data))
|
|
contributor = m.BookContributor(user_id=user_id, book_id=book_id, role=role)
|
|
log(log.INFO, "New contributor [%s]", contributor)
|
|
contributor.save()
|
|
|
|
groups = (
|
|
db.session.query(m.AccessGroup)
|
|
.filter(
|
|
m.BookAccessGroups.book_id == book_id,
|
|
m.AccessGroup.id == m.BookAccessGroups.access_group_id,
|
|
m.AccessGroup.name == role.name.lower(),
|
|
)
|
|
.all()
|
|
)
|
|
for group in groups:
|
|
m.UserAccessGroups(user_id=user_id, access_group_id=group.id).save()
|
|
|
|
flash("Contributor was added!", "success")
|
|
return redirect(
|
|
url_for("book.settings", selected_tab=selected_tab, book_id=book_id)
|
|
)
|
|
|
|
|
|
def delete_contributor_from_book(
|
|
form: f.DeleteContributorForm,
|
|
book_id: int,
|
|
selected_tab: str = "",
|
|
user_id: int = None,
|
|
):
|
|
if not user_id:
|
|
user_id = form.user_id.data
|
|
book_contributor = m.BookContributor.query.filter_by(
|
|
user_id=user_id, book_id=book_id
|
|
).first()
|
|
if not book_contributor:
|
|
log(
|
|
log.INFO,
|
|
"BookContributor does not exists user: [%s], book: [%s]",
|
|
user_id,
|
|
book_id,
|
|
)
|
|
flash("Does not exists!", "success")
|
|
return redirect(
|
|
url_for("book.settings", selected_tab=selected_tab, book_id=book_id)
|
|
)
|
|
|
|
book: m.Book = db.session.get(m.Book, book_id)
|
|
user: m.User = db.session.get(m.User, user_id)
|
|
if book:
|
|
for access_group in book.access_groups:
|
|
access_group: m.AccessGroup
|
|
if user in access_group.users:
|
|
log(
|
|
log.INFO,
|
|
"Delete user [%s] from AccessGroup [%s]",
|
|
user,
|
|
access_group,
|
|
)
|
|
relationships_to_delete = m.UserAccessGroups.query.filter_by(
|
|
user_id=user_id, access_group_id=access_group.id
|
|
).all()
|
|
for relationship in relationships_to_delete:
|
|
db.session.delete(relationship)
|
|
|
|
log(log.INFO, "Delete BookContributor [%s]", book_contributor)
|
|
db.session.delete(book_contributor)
|
|
db.session.commit()
|
|
|
|
flash("Success!", "success")
|
|
return redirect(
|
|
url_for("book.settings", selected_tab=selected_tab, book_id=book_id)
|
|
)
|