118 lines
2.9 KiB
Python
118 lines
2.9 KiB
Python
from flask import Flask, render_template, g, abort, request, jsonify
|
|
import sqlite3
|
|
|
|
app = Flask(__name__)
|
|
|
|
DATABASE = "static/data/monsters.db"
|
|
|
|
|
|
def connect_db():
|
|
return sqlite3.connect(DATABASE)
|
|
|
|
|
|
@app.before_request
|
|
def before_request():
|
|
g.db = connect_db()
|
|
|
|
|
|
@app.teardown_request
|
|
def teardown_request(exception):
|
|
if hasattr(g, "db"):
|
|
g.db.close()
|
|
|
|
|
|
@app.route("/")
|
|
def show_index():
|
|
return render_template("index.html")
|
|
|
|
|
|
@app.route("/get_families")
|
|
def get_families():
|
|
cursor = g.db.cursor()
|
|
cursor.execute("SELECT DISTINCT name FROM families")
|
|
families = [row[0] for row in cursor.fetchall()]
|
|
return jsonify(families)
|
|
|
|
|
|
@app.route("/get_monsters")
|
|
def get_monsters():
|
|
selected_family = request.args.get("family")
|
|
cursor = g.db.cursor()
|
|
|
|
if selected_family:
|
|
cursor.execute(
|
|
"""
|
|
SELECT name FROM monsters
|
|
WHERE family_id = (SELECT id FROM families WHERE name = ?)
|
|
""",
|
|
(selected_family,),
|
|
)
|
|
else:
|
|
cursor.execute("SELECT DISTINCT name FROM monsters")
|
|
|
|
monsters = [row[0] for row in cursor.fetchall()]
|
|
return jsonify(monsters)
|
|
|
|
|
|
@app.route("/monster/<monster_name>")
|
|
def monster_info(monster_name):
|
|
cursor = g.db.cursor()
|
|
|
|
# Retrieve monster information from the database based on name
|
|
cursor.execute(
|
|
"""
|
|
SELECT
|
|
monsters.id, monsters.name, families.name AS family, monsters.in_story,
|
|
monsters.agl, monsters.int, monsters.maxlvl, monsters.atk, monsters.mp,
|
|
monsters.exp, monsters.hp, monsters.def
|
|
FROM
|
|
monsters
|
|
JOIN families ON monsters.family_id = families.id
|
|
WHERE
|
|
monsters.name = ?
|
|
""",
|
|
(monster_name,),
|
|
)
|
|
|
|
monster_info = cursor.fetchone()
|
|
|
|
if monster_info is None:
|
|
abort(404)
|
|
|
|
# Retrieve skills for the monster
|
|
cursor.execute("SELECT skill FROM skills WHERE monster_id = ?", (monster_info[0],))
|
|
skills = [row[0] for row in cursor.fetchall()]
|
|
|
|
# Retrieve spawn locations for the monster
|
|
cursor.execute(
|
|
"SELECT map, description FROM spawn_locations WHERE monster_id = ?",
|
|
(monster_info[0],),
|
|
)
|
|
spawn_locations = [
|
|
{"map": row[0], "description": row[1]} for row in cursor.fetchall()
|
|
]
|
|
|
|
return render_template(
|
|
"monsters.html",
|
|
monster={
|
|
"id": monster_info[0],
|
|
"name": monster_info[1],
|
|
"family": monster_info[2],
|
|
"in_story": "Yes" if monster_info[3] else "No",
|
|
"agl": monster_info[4],
|
|
"int": monster_info[5],
|
|
"maxlvl": monster_info[6],
|
|
"atk": monster_info[7],
|
|
"mp": monster_info[8],
|
|
"exp": monster_info[9],
|
|
"hp": monster_info[10],
|
|
"def": monster_info[11],
|
|
"skills": skills,
|
|
"spawn_locations": spawn_locations,
|
|
},
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(debug=True)
|