from flask import render_template

from common.utils.rights import permission_required, view_logging_required
from models import RightModel
from applications.view import index_bp


@index_bp.get('/rights/')
@view_logging_required
@permission_required("admin:power:main")
def rights_index():
    return render_template('admin/rights/rights.html')


@index_bp.get('/rights/power/<int:power_id>')
@view_logging_required
@permission_required("admin:power:edit")
def rights_edit(power_id):
    power = RightModel.query.filter_by(id=power_id).first()
    icon = str(power.icon).split()
    if len(icon) == 2:
        icon = icon[1]
    else:
        icon = None
    return render_template('admin/rights/rights_edit.html', power=power, icon=icon)


@index_bp.get('/rights/add')
@view_logging_required
@permission_required("admin:power:main")
def rights_add():
    return render_template('admin/rights/rights_add.html')