from flask import render_template, Blueprint, redirect, url_for, request, abort, flash from flask_user import login_required from app.models import PetNames, db, CharacterXML, CharacterInfo from datatables import ColumnDT, DataTables from app.forms import CreatePlayKeyForm, EditPlayKeyForm from app import gm_level, log_audit 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 log_audit(f"Approved pet name {pet_data.pet_name} from {pet_data.owner.name}") flash(f"Approved pet name {pet_data.pet_name} from {pet_data.owner.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 log_audit(f"Rejected pet name {pet_data.pet_name} from {pet_data.owner.name}") flash(f"Rejected pet name {pet_data.pet_name} from {pet_data.owner.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"): # call this to make things nicer accociate_pets_and_owners() columns = [ ColumnDT(PetNames.id), ColumnDT(PetNames.pet_name), ColumnDT(PetNames.approved), ColumnDT(PetNames.owner_id), ] 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" pet_data["3"] = f""" {CharacterInfo.query.filter(CharacterInfo.id==pet_data['3']).first().name} """ return data def accociate_pets_and_owners(): pets = PetNames.query.filter(PetNames.owner_id == None).all() if pets: for pet in pets: owner = CharacterXML.query.filter(CharacterXML.xml_data.like(f"%{pet.id}%")).first() if owner: pet.owner_id = owner.id pet.save()