117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
from typing import Optional
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from frog_api.models import Entry, Project, Version
|
|
from frog_api.serializers import (
|
|
ProjectSerializer,
|
|
TerseEntrySerializer,
|
|
)
|
|
|
|
|
|
class ProjectView(APIView):
|
|
"""
|
|
API endpoint that allows projects to be viewed.
|
|
"""
|
|
|
|
def get(self, request, format=None):
|
|
"""
|
|
Return a list of all projects.
|
|
"""
|
|
projects = Project.objects.all()
|
|
serializer = ProjectSerializer(projects, many=True)
|
|
return Response(serializer.data)
|
|
|
|
|
|
def get_latest_entry(project, version, category) -> Optional[Entry]:
|
|
return Entry.objects.filter(
|
|
category__slug=category,
|
|
category__version__slug=version,
|
|
category__version__project__slug=project,
|
|
).first()
|
|
|
|
|
|
def get_entry_json(entry) -> dict:
|
|
entry_data = TerseEntrySerializer(entry).data
|
|
entry_data["measures"] = {m["type"]: m["value"] for m in entry_data["measures"]}
|
|
return entry_data
|
|
|
|
|
|
def get_versions_digest_for_project(project) -> dict:
|
|
versions = {}
|
|
for version in Version.objects.filter(project__slug=project):
|
|
entry = get_latest_entry(project, version.slug, "total")
|
|
if entry is not None:
|
|
versions[version.slug] = get_entry_json(entry)
|
|
return versions
|
|
|
|
|
|
class RootDigestView(APIView):
|
|
"""
|
|
API endpoint that returns the most recent entry for each version of each project.
|
|
"""
|
|
|
|
def get(self, request):
|
|
"""
|
|
Return the most recent entry for ovreall progress for each version of each project.
|
|
"""
|
|
|
|
projects = {}
|
|
for project in Project.objects.all():
|
|
versions = get_versions_digest_for_project(project.slug)
|
|
if len(versions) > 0:
|
|
projects[project.slug] = versions
|
|
|
|
return Response({"progress": projects})
|
|
|
|
|
|
class ProjectDigestView(APIView):
|
|
"""
|
|
API endpoint that returns the most recent entry for each version of a project.
|
|
"""
|
|
|
|
def get(self, request, project):
|
|
"""
|
|
Return the most recent entry for overall progress for each version of a project.
|
|
"""
|
|
|
|
if not Project.objects.filter(slug=project).exists():
|
|
return Response(
|
|
f"Project {project} not found", status=status.HTTP_404_NOT_FOUND
|
|
)
|
|
|
|
projects = {}
|
|
|
|
versions = get_versions_digest_for_project(project)
|
|
if len(versions) > 0:
|
|
projects[project] = versions
|
|
|
|
return Response({"progress": projects})
|
|
|
|
|
|
class VersionDigestView(APIView):
|
|
"""
|
|
API endpoint that returns the most recent entry for a version of a project.
|
|
"""
|
|
|
|
def get(self, request, project, version):
|
|
"""
|
|
Return the most recent entry for overall progress for a version of a project.
|
|
"""
|
|
|
|
if not Project.objects.filter(slug=project).exists():
|
|
return Response(
|
|
f"Project {project} not found", status=status.HTTP_404_NOT_FOUND
|
|
)
|
|
|
|
if not Version.objects.filter(slug=version, project__slug=project).exists():
|
|
return Response(
|
|
f"Version {version} not found", status=status.HTTP_404_NOT_FOUND
|
|
)
|
|
|
|
entry = get_latest_entry(project, version, "total")
|
|
entry_json = get_entry_json(entry)
|
|
|
|
return Response(entry_json)
|