from flask import render_template, Blueprint, redirect, url_for, request, abort, flash from flask_user import login_required from app.models import PetNames, db from datatables import ColumnDT, DataTables from app.forms import CreatePlayKeyForm, EditPlayKeyForm from app import gm_level moderation_blueprint = Blueprint('moderation', __name__) @moderation_blueprint.route('/', methods=['GET']) @login_required @gm_level(3) def index(status): return render_template('moderation/index.html.j2', status=status) @moderation_blueprint.route('/approve_pet/', methods=['GET']) @login_required @gm_level(3) def approve_pet(id): pet_data = PetNames.query.filter(PetNames.id == id).first() pet_data.approved = 2 flash(f"Approved pet name {pet_data.pet_name}", "success") pet_data.save() return redirect(request.referrer if request.referrer else url_for("main.index")) @moderation_blueprint.route('/reject_pet/', methods=['GET']) @login_required @gm_level(3) def reject_pet(id): pet_data = PetNames.query.filter(PetNames.id == id).first() pet_data.approved = 0 flash(f"Rejected pet name {pet_data.pet_name}", "danger") pet_data.save() return redirect(request.referrer if request.referrer else url_for("main.index")) @moderation_blueprint.route('/get_pets/', methods=['GET']) @login_required @gm_level(3) def get_pets(status="all"): columns = [ ColumnDT(PetNames.id), ColumnDT(PetNames.pet_name), ColumnDT(PetNames.approved), ] query = None if status=="all": query = db.session.query().select_from(PetNames) elif status=="approved": query = db.session.query().select_from(PetNames).filter(PetNames.approved==2) elif status=="unapproved": query = db.session.query().select_from(PetNames).filter(PetNames.approved==1) else: raise Exception("Not a valid filter") params = request.args.to_dict() rowTable = DataTables(params, query, columns) data = rowTable.output_result() for pet_data in data["data"]: id = pet_data["0"] status = pet_data["2"] if status == 1: pet_data["0"] = f""" """ pet_data["2"] = "Awaiting Moderation" elif status == 2: pet_data["0"] = f""" Reject """ pet_data["2"] = "Approved" elif status == 0: pet_data["0"] = f""" Approve """ pet_data["2"] = "Rejected" return data