"""Blueprint for DAG API."""

from flask import Blueprint, request
from owsresponse.adaptors.flask import flaskify

from abacus_contract.logic import dag_run_times

dag_api = Blueprint('dag_api', __name__)


@dag_api.route('/dag_run_time/<string:dag_id>', methods=['GET'])
def get_dag_run_time(dag_id: str):
    """Handle getting the dag run times for a particular dag_id."""
    days_back = request.args.get('days_back', default=30, type=int)
    return flaskify(dag_run_times.get_average_dag_run_time(dag_id, days_back=days_back))