You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
928 B
20 lines
928 B
from django.conf import settings |
|
from django.db import models |
|
from django.db.models.fields import FieldDoesNotExist |
|
|
|
class CurrentSiteManager(models.Manager): |
|
"Use this to limit objects to those associated with the current site." |
|
def __init__(self, field_name='site'): |
|
super(CurrentSiteManager, self).__init__() |
|
self.__field_name = field_name |
|
self.__is_validated = False |
|
|
|
def get_query_set(self): |
|
if not self.__is_validated: |
|
try: |
|
self.model._meta.get_field(self.__field_name) |
|
except FieldDoesNotExist: |
|
raise ValueError, "%s couldn't find a field named %s in %s." % \ |
|
(self.__class__.__name__, self.__field_name, self.model._meta.object_name) |
|
self.__is_validated = True |
|
return super(CurrentSiteManager, self).get_query_set().filter(**{self.__field_name + '__id__exact': settings.SITE_ID})
|
|
|