friwall/web/nat.py

import flask
import flask_login

from . import db
from . import system

blueprint = flask.Blueprint('nat', __name__, url_prefix='/nat')

@blueprint.route('/', methods=('GET', 'POST'))
@flask_login.login_required
def index():
    try:
        if not flask_login.current_user.is_admin:
            return flask.Response('forbidden', status=403, mimetype='text/plain')

        with db.locked():
            nat = { office: "" for office in db.read('networks') }
            nat |= db.read('nat')

            if flask.request.method == 'POST':
                form = flask.request.form
                for office, address in form.items():
                    if office in nat:
                        nat[office] = address
                db.write('nat', nat)
                system.run(system.save_config)
                return flask.redirect(flask.url_for('nat.index'))

            return flask.render_template('nat/index.html', nat=nat)
    except TimeoutError:
        return flask.render_template('busy.html')
    except Exception as e:
        return flask.Response(f'something went catastrophically wrong: {e}',
                status=400, mimetype='text/plain')