from course_service.courses.models import Course


class InApiTeacher:
    @staticmethod
    def add_teacher(slug: str, token: str) -> Course:
        course = Course.objects.get(slug=slug)
        course.teachers.append(token)
        course.save()
        return course

    @staticmethod
    def delete_teacher(slug: str, token: str) -> None:
        course = Course.objects.get(slug=slug)
        course.teachers.remove(token)
        course.save()
        return None

    @staticmethod
    def get_token_list(slug: str) -> list:
        return Course.objects.get(slug=slug).teachers