acks-tools/acks/views.py

from flask import Blueprint, render_template, url_for, redirect
from flask_basicauth import BasicAuth



basic_auth = BasicAuth()
default_views = Blueprint('default_views', __name__, url_prefix='/')

@default_views.route('/')
def index():
    return render_template('index.html')

@default_views.route('/handbook')
def handbook():
    return render_template('handbook.html')

@default_views.route('/wiki')
@basic_auth.required
def wiki():
    return render_template('wiki.html')

@default_views.route('/worldmap')
def worldmap():
    return render_template('worldmap.html')

#@default_views.route('/treasure')
#@default_views.route('/treasure/<string:treasure_type>')
def treasure(treasure_type=None):
    # import requests
    # from lxml import html
    # Unused, we can't tap into Autarch's generator, so just link to a new tab
    # in navigation bar now
    if treasure_type is not None:
        with requests.Session() as s:
            # First grab the page to pull a fresh form_build_id
            treasure_page = s.get('http://autarch.co/treasure')
            dom_tree = html.fromstring(treasure_page.content)
            form_build_id = dom_tree.find('.//input[@name="form_build_id"]').attrib['value']

            # Use that form_build_id to make a request for generated treasure
            headers = {'content-type': 'application/x-www-form-urlencoded; charset=UTF-8'}
            payload = {
                "form_id": "acks_treasure_form",
                "form_build_id": form_build_id,
                "treasure_type": treasure_type
            }
            response = s.post("http://autarch.co/system/ajax", data=payload, headers=headers)
            print("TG Response: {}, Payload: {}".format(response.text, payload))
            generated_treasure = response.json()[1]["data"].replace('class="form-textarea"', 'class="uk-textarea"')
            return render_template('treasure.html', generated_treasure=generated_treasure, treasure_type=treasure_type)
    return render_template('treasure.html')