Search code examples
pythondjangodjango-modelsdjango-viewsdjango-custom-manager

Facing issue in custom manager of django


I am trying to create a custom manager to retrieve all posts with the published status. New to managers!! Thank you in advance <3.

models.py


class PublishedManager(models.Model):
    def get_query_set(self):
        return super(PublishedManager, self).get_query_set().filter(status='published')


class Post(models.Model):
    STATUS_CHOICES = (
        ('draft', 'Draft'),
        ('published', 'Published'),
    )
    title = models.CharField(max_length=255)
    slug = models.SlugField(max_length=255, unique_for_date='publish')
    author = models.ForeignKey(
        User, on_delete=models.CASCADE, related_name='blog_posts')
    body = models.TextField()
    publish = models.DateTimeField(default=timezone.now)
    created = models.DateTimeField(auto_now_add=True)
    updated = models.DateTimeField(auto_now=True)
    status = models.CharField(
        max_length=10, choices=STATUS_CHOICES, default='draft')
    objects = models.Manager()
    published = PublishedManager()

    class Meta:
        ordering = ('-publish',)

    def __str__(self):
        return self.title

    def get_absolute_url(self):
        return reverse('blog:post_detail', args=[self.publish.year, self.publish.month, self.publish.day, self.slug])

views.py

def post_list(request):
    posts = Post.published.all()
    print(posts)
    return render(request, 'post/list.html', {'posts': posts})


def post_detail(request):
    post = get_object_or_404(Post, slug=post, status='published',
                             publish__year=year, publish__month=month, publish__day=day)

    return render(request, 'post/detail.html', {'post': post})

Error

'PublishedManager' object has no attribute 'all' (views.py, line 6, in post_list)


Solution

  • You should use Manager as a base class for manager, not Model. And method name should be get_queryset instead of get_query_set:

    class PublishedManager(models.Manager):
        def get_queryset(self):
            return super(PublishedManager, self).get_queryset().filter(status='published')
    

    You can find more details in the docs.