from django.contrib.auth import get_user_model

from ..ammo.my_course_permissions import MyCoursePermissionsListAmmo
from .available_courses_mixin import AvailableCoursesPayloadMixin
from .base_ammo import UsersWithUidBaseAmmoFactory

User = get_user_model()


class MyCoursePermissionsAmmoFactory(AvailableCoursesPayloadMixin, UsersWithUidBaseAmmoFactory):
    ammo_class = MyCoursePermissionsListAmmo
