from courses.models import Course


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

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

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