1from django.db.models.query import QuerySet
2from django.conf import settings
3
4
5class SharedQueries:
6
7    """Some queries that are identical for Gallery and Photo."""
8
9    def is_public(self):
10        """Trivial filter - will probably become more complex as time goes by!"""
11        return self.filter(is_public=True)
12
13    def on_site(self):
14        """Return objects linked to the current site only."""
15        return self.filter(sites__id=settings.SITE_ID)
16
17
18class GalleryQuerySet(SharedQueries, QuerySet):
19    pass
20
21
22class PhotoQuerySet(SharedQueries, QuerySet):
23    pass
24