Django Views фильтрует объекты по пользователю из пользовательской модели

Я создал пользовательскую модель пользователя, и я пытаюсь фильтровать данные в views.py по этому пользователю.

Ошибка, которую я получаю:

У объекта 'SomeClassView' нет атрибута 'user'

Моя цель - "инкапсулировать" данные для каждого пользователя.

модель пользователя:

class CustomUserManger(BaseUserManager):

    use_in_migrations = True

    def create_user(self, email, username, password, **other_fields):
        email = self.normalize_email(email)
        user = self.model(email=email, username=username, **other_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_superuser(self, email, username, password, **other_fields):
        other_fields.setdefault('is_staff', True)
        other_fields.setdefault('is_superuser', True)
        other_fields.setdefault('is_active', True)
        if other_fields.get('is_staff') is not True:
            raise ValueError('Superuser must be assigned to staff')
        if other_fields.get('is_superuser') is not True:
            raise ValueError('Superuser must be assigned to superusers')
        return self.create_user(email, username, password, **other_fields)


class User(AbstractUser, PermissionsMixin):
    username = models.CharField(_('username'), max_length=20, unique=True)
    email = models.EmailField(unique=True)
    password = models.CharField(max_length=128)
    is_staff = models.BooleanField(default=True)
    is_active = models.BooleanField(default=True)

    objects = CustomUserManger()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['username']

    def __str__(self):
        return self.username

Просмотров:

class SomeClassView(viewsets.ModelViewSet):
    user = SomeClass.user
    serializer_class = WagonSerializer
    authentication_classes = (SessionAuthentication, )

    @login_required
    def get_queryset(self):
        user = self.request.user
        return SomeClass.objects.filter(user=user)

Вернуться на верх