import graphene

from vbv_lernwelt.course.models import Course
from vbv_lernwelt.dashboard.graphql.types.dashboard import CourseStatisticsType


class DashboardQuery(graphene.ObjectType):
    course_statistics = graphene.List(
        CourseStatisticsType, course_id=graphene.ID(required=True)
    )

    def resolve_course_statistics(root, info, course_id: str):
        query = Course.objects.filter(
            id=course_id
        )

        courses = query.distinct()

        return [
            CourseStatisticsType(course_id=course.id, course_title=course.title)  # noqa
            for course in courses
        ]