dwm-app/api/get_breeding_pairs.py

26 lines
950 B
Python

from flask import jsonify, Blueprint
from utils.utils import breeding_pair_data
get_breeding_pairs_bp = Blueprint('breeding_pairs',__name__)
# Read the CSV file and store breeding information
breeding_info = []
with open(breeding_pair_data, '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_breeding_pairs(monster):
pairs = []
for entry in breeding_info:
if entry['offspring'].lower() == monster.lower():# or entry['mate'].lower() == monster.lower():
pairs.append({'base': entry['base'], 'mate': entry['mate'], 'offspring': entry['offspring']})
return pairs
@get_breeding_pairs_bp.route('/api/breeding/pairs/<monster>', methods=['GET'])
def breeding_pairs(monster):
pairs = get_breeding_pairs(monster)
return jsonify({'breeding_pairs': pairs})