mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2025-04-19 08:55:15 +00:00
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
from flask import Flask, jsonify, request
|
|
import numpy as np
|
|
from datetime import datetime
|
|
import modules.class_load as cl
|
|
from pprint import pprint
|
|
|
|
app = Flask(__name__)
|
|
|
|
# Constants
|
|
DATE_FORMAT = '%Y-%m-%d'
|
|
EXPECTED_ARRAY_SHAPE = (2, 24)
|
|
FILEPATH = r'.\load_profiles.npz'
|
|
|
|
def get_load_forecast(year_energy):
|
|
"""Initialize LoadForecast with the given year_energy."""
|
|
return cl.LoadForecast(filepath=FILEPATH, year_energy=float(year_energy))
|
|
|
|
def validate_date(date_str):
|
|
"""Validate the date string and return a datetime object."""
|
|
try:
|
|
return datetime.strptime(date_str, DATE_FORMAT)
|
|
except ValueError:
|
|
raise ValueError("Date is not in the correct format. Expected format: YYYY-MM-DD.")
|
|
|
|
@app.route('/getdata', methods=['GET'])
|
|
def get_data():
|
|
# Retrieve the date and year_energy from query parameters
|
|
date_str = request.args.get('date')
|
|
year_energy = request.args.get('year_energy')
|
|
|
|
if not date_str or not year_energy:
|
|
return jsonify({"error": "Missing 'date' or 'year_energy' query parameter."}), 400
|
|
|
|
try:
|
|
# Validate and convert the date
|
|
date_obj = validate_date(date_str)
|
|
lf = get_load_forecast(year_energy)
|
|
|
|
# Get daily statistics for the requested date
|
|
array_list = lf.get_daily_stats(date_str)
|
|
pprint(array_list)
|
|
pprint(array_list.shape)
|
|
|
|
# Check if the shape of the array is valid
|
|
if array_list.shape == EXPECTED_ARRAY_SHAPE:
|
|
return jsonify({date_str: array_list.tolist()})
|
|
else:
|
|
return jsonify({"error": "Data not found for the given date."}), 404
|
|
|
|
except ValueError as e:
|
|
# Return a descriptive error message for date validation issues
|
|
return jsonify({"error": str(e)}), 400
|
|
except Exception:
|
|
# Return a generic error message for unexpected errors
|
|
return jsonify({"error": "An unexpected error occurred."}), 500
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True)
|