import base64 from flask import Blueprint, render_template, request, flash, redirect, url_for, jsonify from flask_login import login_required, current_user from app.controllers import create_pagination from sqlalchemy import not_ from app import models as m, db from app import forms as f from app.logger import log from config import config configuration = config() bp = Blueprint("user", __name__, url_prefix="/user") @bp.route("/", methods=["GET"]) @login_required def get_all(): q = request.args.get("q", type=str, default=None) users = m.User.query.order_by(m.User.id) if q: users = users.filter(m.User.username.like(f"{q}%")) pagination = create_pagination(total=users.count()) return render_template( "user/users.html", users=users.paginate(page=pagination.page, per_page=pagination.per_page), page=pagination, search_query=q, ) @bp.route("/profile", methods=["GET", "POST"]) @login_required def profile(): form = f.EditUserForm() if form.validate_on_submit(): user: m.User = current_user user.username = form.name.data if form.avatar_img.data: img_data = form.avatar_img.data.read() img_data = base64.b64encode(img_data) current_user.avatar_img = img_data.decode("utf-8") user.is_activated = True user.save() return redirect(url_for("main.index")) elif form.is_submitted: log(log.ERROR, "Update user errors: [%s]", form.errors) for field, errors in form.errors.items(): field_label = form._fields[field].label.text for error in errors: flash(error.replace("Field", field_label), "danger") if current_user.is_activated: form.name.data = current_user.username return render_template("user/profile.html", form=form) @bp.route("/save", methods=["POST"]) @login_required def save(): form = f.UserForm() if form.validate_on_submit(): u: m.User = m.User.query.get(int(form.user_id.data)) if not u: log(log.ERROR, "Not found user by id : [%s]", form.user_id.data) flash("Cannot save user data", "danger") u.username = form.username.data u.activated = form.activated.data if form.password.data.strip("*\n "): u.password = form.password.data u.save() if form.next_url.data: return redirect(form.next_url.data) return redirect(url_for("user.get_all")) else: log(log.ERROR, "User save errors: [%s]", form.errors) flash(f"{form.errors}", "danger") return redirect(url_for("user.get_all")) @bp.route("/create", methods=["POST"]) @login_required def create(): form = f.NewUserForm() if form.validate_on_submit(): user = m.User( username=form.username.data, password=form.password.data, activated=form.activated.data, ) log(log.INFO, "Form submitted. User: [%s]", user) flash("User added!", "success") user.save() return redirect(url_for("user.get_all")) @bp.route("/delete/", methods=["DELETE"]) @login_required def delete(id): u = m.User.query.filter_by(id=id).first() if not u: log(log.INFO, "There is no user with id: [%s]", id) flash("There is no such user", "danger") return "no user", 404 db.session.delete(u) db.session.commit() log(log.INFO, "User deleted. User: [%s]", u) flash("User deleted!", "success") return "ok", 200 @bp.route("/search", methods=["GET"]) @login_required def search(): q = request.args.get("q", type=str, default=None) if not q: return jsonify({"message": "q parameter is required"}), 422 book_id = request.args.get("book_id", type=str, default=None) query_user = m.User.query query_user = query_user.order_by(m.User.username) query_user = query_user.filter(m.User.username.ilike(f"{q}%")) if book_id: book_contributors = m.BookContributor.query.filter_by(book_id=book_id).all() user_ids = [contributor.user_id for contributor in book_contributors] user_ids.append(current_user.id) query_user = query_user.filter(not_(m.User.id.in_(user_ids))) query_user = query_user.limit(configuration.MAX_SEARCH_RESULTS) users = [{"username": user.username, "id": user.id} for user in query_user.all()] return jsonify({"users": users})