dwm-app/src/api/get_breeding_pairs.py

47 lines
1.5 KiB
Python
Raw Normal View History

2024-02-22 03:23:13 +00:00
from flask import Flask, jsonify, request, Blueprint
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
#def get_usage_list(monster):
# used_in = []
# for entry in breeding_info:
# if entry['offspring'].lower() == monster.lower():
# used_in.append({'base': entry['base'], 'mate': entry['mate']})
# return used_in
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})
2024-02-22 14:20:06 +00:00
#@app.route('api/breeding/usage/<monster>', methods=['GET'])
2024-02-22 03:23:13 +00:00
#def usage_list(monster):
# used_in = get_usage_list(monster)
# return jsonify({'used_in': used_in})
#