1
from django.conf import settings
2
from django.db import models
3
from django.db.models.fields import FieldDoesNotExist
5
class CurrentSiteManager(models.Manager):
6
"Use this to limit objects to those associated with the current site."
7
def __init__(self, field_name='site'):
8
super(CurrentSiteManager, self).__init__()
9
self.__field_name = field_name
10
self.__is_validated = False
12
def get_query_set(self):
13
if not self.__is_validated:
15
self.model._meta.get_field(self.__field_name)
16
except FieldDoesNotExist:
17
raise ValueError, "%s couldn't find a field named %s in %s." % \
18
(self.__class__.__name__, self.__field_name, self.model._meta.object_name)
19
self.__is_validated = True
20
return super(CurrentSiteManager, self).get_query_set().filter(**{self.__field_name + '__id__exact': settings.SITE_ID})