96 lines
2.8 KiB
Python
96 lines
2.8 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)
|