import graphene

from vbv_lernwelt.competence.graphql.types import CompetenceCertificateObjectType
from vbv_lernwelt.competence.models import CompetenceCertificate


class CompetenceCertificateQuery(object):
    competence_certificate = graphene.Field(
        CompetenceCertificateObjectType,
        id=graphene.ID(),
    )

    def resolve_competence_certificate(root, info, id):
        return CompetenceCertificate.objects.get(id=id)