from django.db.models import Manager


class ScreenshotManager(Manager):
    def published(self, user=None, is_staff=False):
        query = self.get_query_set()
        query = query.order_by('uploaded_at')
        if is_staff:
            return query
        elif user:
            return query.filter(Q(published=True) | Q(user=user))
        else:
            return query.filter(published=True)

from django.db.models import Manager
from django.db.models import Q


class ScreenshotManager(Manager):
    def published(self, user=None, is_staff=False):
        query = self.get_query_set()
        query = query.order_by('uploaded_at')
        if is_staff:
            return query
        elif user:
            return query.filter(Q(published=True) | Q(uploaded_by=user))
        else:
            return query.filter(published=True)

Fix missing import and bad query for screenshots
Fix missing import and bad query for screenshots

  from django.db.models import Manager
+ from django.db.models import Q
  
  
  class ScreenshotManager(Manager):
      def published(self, user=None, is_staff=False):
          query = self.get_query_set()
          query = query.order_by('uploaded_at')
          if is_staff:
              return query
          elif user:
-             return query.filter(Q(published=True) | Q(user=user))
+             return query.filter(Q(published=True) | Q(uploaded_by=user))
          else:
              return query.filter(published=True)
  
Fix missing import and bad query for screenshots
## Code Before:
from django.db.models import Manager


class ScreenshotManager(Manager):
    def published(self, user=None, is_staff=False):
        query = self.get_query_set()
        query = query.order_by('uploaded_at')
        if is_staff:
            return query
        elif user:
            return query.filter(Q(published=True) | Q(user=user))
        else:
            return query.filter(published=True)

## Instruction:
Fix missing import and bad query for screenshots
## Code After:
from django.db.models import Manager
from django.db.models import Q


class ScreenshotManager(Manager):
    def published(self, user=None, is_staff=False):
        query = self.get_query_set()
        query = query.order_by('uploaded_at')
        if is_staff:
            return query
        elif user:
            return query.filter(Q(published=True) | Q(uploaded_by=user))
        else:
            return query.filter(published=True)
