NexusDashboard/app/moderation.py
Aaron Kimbre d63dd807e3 Handle deleted chars in pet name association
Fix pre-built obj generation
2022-02-12 19:09:45 -06:00

140 lines
4.6 KiB
Python

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('/<status>', methods=['GET'])
@login_required
@gm_level(3)
def index(status):
return render_template('moderation/index.html.j2', status=status)
@moderation_blueprint.route('/approve_pet/<id>', 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/<id>', 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/<status>', 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"""
<div class="row">
<div class="col">
<a role="button" class="btn btn-success btn btn-block"
href='{url_for('moderation.approve_pet', id=id)}'>
Approve
</a>
</div>
<div class="col">
<a role="button" class="btn btn-danger btn btn-block"
href='{url_for('moderation.reject_pet', id=id)}'>
Reject
</a>
</div>
</div>
"""
pet_data["2"] = "Awaiting Moderation"
elif status == 2:
pet_data["0"] = f"""
<a role="button" class="btn btn-danger btn btn-block"
href='{url_for('moderation.reject_pet', id=id)}'>
Reject
</a>
"""
pet_data["2"] = "<span class='text-success'>Approved</span>"
elif status == 0:
pet_data["0"] = f"""
<a role="button" class="btn btn-success btn btn-block"
href='{url_for('moderation.approve_pet', id=id)}'>
Approve
</a>
"""
pet_data["2"] = "<span class='text-danger'>Rejected</span>"
try:
pet_data["3"] = f"""
<a role="button" class="btn btn-primary btn btn-block"
href='{url_for('characters.view', id=pet_data["3"])}'>
{CharacterInfo.query.filter(CharacterInfo.id==pet_data['3']).first().name}
</a>
"""
except Exception as e:
PetNames.query.filter(PetNames.id==id).first().delete()
pet_data["3"] = "Deleted Character"
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()
else:
pet.delete()