from django.shortcuts import get_object_or_404, render

from control_plane.trading_studio.models import TradingProject


def trading_studio(request):
    projects = TradingProject.objects.select_related("project", "current_champion").order_by("name")
    return render(request, "control_plane/trading_studio.html", {"trading_projects": projects})


def trading_studio_project(request, project_id):
    trading_project = get_object_or_404(TradingProject.objects.select_related("project", "current_champion"), id=project_id)
    return render(request, "control_plane/trading_studio_project.html", {
        "trading_project": trading_project,
        "datasets": trading_project.market_datasets.prefetch_related("versions").all(),
        "features": trading_project.features.order_by("name"),
        "strategies": trading_project.strategies.prefetch_related("versions").all(),
        "cohorts": trading_project.cohorts.prefetch_related("experiments").all(),
        "reports": trading_project.research_reports.order_by("-created_at")[:10],
    })