dwm-app/src/api/get_breeding_usage.py

37 lines
1.1 KiB
Python

from flask import jsonify, Blueprint
get_breeding_usage_bp = Blueprint('breeding_usage',__name__)
import os
# Get the current script directory
script_dir = os.path.dirname(os.path.abspath(__file__))
# Construct the path to the CSV file in the data directory
csv_file_path = os.path.join(script_dir, '..', 'data', 'updated_breeding_pairs.csv')
# Read the CSV file and store breeding information
breeding_info = []
with open(csv_file_path, 'r') as file:
lines = file.readlines()
header = lines[0].strip().split(',')
for line in lines[1:]:
data = line.strip().split(',')
breeding_info.append(dict(zip(header, data)))
def get_usage_list(monster):
used_in = []
for entry in breeding_info:
if entry['base'].lower() == monster.lower() or entry['mate'].lower() == monster.lower():
used_in.append({'base': entry['base'], 'mate': entry['mate'], 'offspring': entry['offspring']})
return used_in
@get_breeding_usage_bp.route('/api/breeding/usage/<monster>', methods=['GET'])
def usage_list(monster):
used_in = get_usage_list(monster)
return jsonify({'used_in': used_in})