dwm-app/app.py

126 lines
3.7 KiB
Python
Raw Normal View History

2024-01-30 16:06:00 +00:00
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
})
2024-01-30 22:02:58 +00:00
# Add a route for the breeding page
@app.route('/breeding')
def breeding():
# Pass any necessary data to the breeding template (e.g., breeding_data)
return render_template('breeding.html')
# Add this function to handle breeding logic
def breed_monsters(parent1, parent2):
# Implement your breeding logic here
# For now, let's assume the result is a string
result = f"{parent1} x {parent2} => Offspring Monster"
return result
# Add this route for breeding
@app.route('/breed', methods=['GET'])
def breed():
# Get parent monsters from query parameters
parent1 = request.args.get('parent1')
parent2 = request.args.get('parent2')
# Validate input (you may want to add more validation)
if not parent1 or not parent2:
return jsonify({'error': 'Invalid input'})
# Call the breeding function
result = breed_monsters(parent1, parent2)
# Return the breeding result as JSON
return jsonify({'result': result})
2024-01-30 16:06:00 +00:00
if __name__ == '__main__':
app.run(debug=True)