import graphene
from graphene_django.types import DjangoObjectType

from university.models import Course, Upload, Category


class CourseType(DjangoObjectType):
    class Meta:
        model = Course
        convert_choices_to_enum = False


class CategoryType(DjangoObjectType):
    class Meta:
        model = Category


class UploadType(DjangoObjectType):
    url = graphene.String()

    class Meta:
        model = Upload


class Query:
    all_courses = graphene.List(CourseType)
    all_uploads = graphene.List(UploadType)

    def resolve_all_courses(self, info, **kwargs):
        return Course.objects.all()

    def resolve_all_uploads(self, info, **kwargs):
        return Upload.objects.all()