from flask import Flask, render_template, request
import numpy as np, json, os
from training.query import query
app = Flask(__name__)
#main functionality
@app.route('/', methods=['GET', 'POST'])
def home():
if request.method == "POST":
#do back end things here and then return what we want, take it back in a success function in JS and update the page.
heroes = [name.encode('UTF8') for name in request.json['heroes']]
radiant = [get_hero_id(hero) for hero in heroes[:5] if get_hero_id(hero)]
dire = [get_hero_id(hero) for hero in heroes[5:] if get_hero_id(hero)]
mmr = int(request.json['mmr'])
text = query(mmr, radiant, dire)
if isinstance(text, list):
text = ''.join([str(hero[0]) + ': ' + str(round(hero[1][0] * 100, 2))+'% win rate. <br>' for hero in text[:10]])
return text
hero_names = get_full_hero_list()
radiant_heroes, dire_heroes = get_hero_factions()
edited_names = [name.replace(" ", "_").replace("\'", "").lower() for name in hero_names]
return render_template('main2.html', hero_names=sorted(hero_names), edited_names=sorted(edited_names), radiant_heroes=radiant_heroes, dire_heroes=dire_heroes)
def get_full_hero_list():
hero_list = []
heroes_json_data = json_data["heroes"]
for hero in heroes_json_data:
hero_list.append(hero["name"])
return hero_list
#gets hero factions and primary attribute from new json file
def get_hero_factions():
attributes = ["str", "agi", "int"]
radiant_heroes, dire_heroes = {}, {}
for attr in attributes:
radiant_heroes[attr] = []
dire_heroes[attr] = []
for hero in json_data["heroes"]:
if hero['faction'] == "Radiant":
radiant_heroes[hero['primary_attribute'][:3].lower()].append(hero["name"].encode('UTF8'))
else:
dire_heroes[hero['primary_attribute'][:3].lower()].append(hero["name"].encode('UTF8'))
return radiant_heroes, dire_heroes
def get_hero_id(name):
for hero in json_data["heroes"]:
if hero["name"] == name:
return hero["id"]
if __name__ == '__main__':
global json_data
json_data = json.load(open(os.path.join('metadata.json'), "rt"))
#app.run(debug= True,host="127.0.0.1",port=5000, threaded=True)
app.run(debug=False,host="0.0.0.0",port=5000, threaded=True)
