FIL-190: Added Generic Foreign Key to Placeholder model

django-cms · czpython · Jul 25, 2018 · Jul 20, 2018 · Jul 20, 2018 · Jul 20, 2018
commit 67cd0fed1debaba9839c6fe1a0ec429611219914
diff --git a/cms/migrations/0022_auto_20180719_1638.py b/cms/migrations/0022_auto_20180719_1638.py
@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.14 on 2018-07-19 16:38
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('contenttypes', '0002_remove_content_type_name'),
+        ('cms', '0021_auto_20180507_1432'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='placeholder',
+            name='content_type',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='contenttypes.ContentType'),
+        ),
+        migrations.AddField(
+            model_name='placeholder',
+            name='object_id',
+            field=models.PositiveIntegerField(blank=True, null=True),
+        ),
+    ]
diff --git a/cms/models/placeholdermodel.py b/cms/models/placeholdermodel.py
@@ -10,6 +10,8 @@
 from django.utils import six
 from django.utils.encoding import python_2_unicode_compatible
 from django.utils.translation import ugettext_lazy as _, force_text
+from django.contrib.contenttypes.fields import GenericForeignKey
+from django.contrib.contenttypes.models import ContentType
 
 from cms.cache.placeholder import clear_placeholder_cache
 from cms.exceptions import LanguageError
@@ -38,6 +40,10 @@ class Placeholder(models.Model):
     cache_placeholder = True
     is_static = False
     is_editable = True
+    content_type = models.ForeignKey(ContentType,  blank=True, null=True, on_delete=models.SET_NULL)
+    object_id = models.PositiveIntegerField(blank=True, null=True)
+    content_object = GenericForeignKey('content_type', 'object_id')
+
 
     class Meta:
         app_label = 'cms'