dwm-app/src/api/get_breeding_pairs.py

33 lines
1.1 KiB
Python
Raw Normal View History

2024-02-22 15:58:22 +00:00
from flask import jsonify, Blueprint
2024-02-22 03:23:13 +00:00
get_breeding_pairs_bp = Blueprint('breeding_pairs',__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_breeding_pairs(monster):
pairs = []
for entry in breeding_info:
2024-02-22 14:20:06 +00:00
if entry['offspring'].lower() == monster.lower():# or entry['mate'].lower() == monster.lower():
2024-02-22 03:23:13 +00:00
pairs.append({'base': entry['base'], 'mate': entry['mate'], 'offspring': entry['offspring']})
return pairs
2024-02-22 14:20:06 +00:00
@get_breeding_pairs_bp.route('/api/breeding/pairs/<monster>', methods=['GET'])
2024-02-22 03:23:13 +00:00
def breeding_pairs(monster):
pairs = get_breeding_pairs(monster)
return jsonify({'breeding_pairs': pairs})