django-orchestra/orchestra/contrib/websites/apps.py

18 lines
641 B
Python
Raw Normal View History

2015-03-23 15:36:51 +00:00
from django.apps import AppConfig
from django.contrib.contenttypes.fields import GenericRelation
from orchestra.utils import database_ready
class WebsiteConfig(AppConfig):
2015-04-05 10:46:24 +00:00
name = 'orchestra.contrib.websites'
2015-03-23 15:36:51 +00:00
def ready(self):
if database_ready():
2015-04-08 14:41:09 +00:00
from django.contrib.contenttypes.models import ContentType
2015-03-23 15:36:51 +00:00
from .models import Content
qset = Content.content_type.field.get_limit_choices_to()
for ct in ContentType.objects.filter(qset):
relation = GenericRelation('websites.Content')
ct.model_class().add_to_class('content_set', relation)