From 2c5e1e6602dc698fc9b378b01803890d55a1454c Mon Sep 17 00:00:00 2001 From: Zach Taylor Date: Tue, 20 Oct 2020 11:28:14 -0500 Subject: [PATCH] Add entity_ct to Attribute model --- eav/migrations/0001_initial.py | 1 + eav/models.py | 9 +++++++++ 2 files changed, 10 insertions(+) diff --git a/eav/migrations/0001_initial.py b/eav/migrations/0001_initial.py index 7417deb..5d9f027 100644 --- a/eav/migrations/0001_initial.py +++ b/eav/migrations/0001_initial.py @@ -27,6 +27,7 @@ class Migration(migrations.Migration): ('modified', models.DateTimeField(auto_now=True, verbose_name='Modified')), ('required', models.BooleanField(default=False, verbose_name='Required')), ('display_order', models.PositiveIntegerField(default=1, verbose_name='Display order')), + ('entity_ct', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.PROTECT, related_name='attribute_entities', to='contenttypes.ContentType', verbose_name='Belongs to')), ], options={ 'ordering': ['name'], diff --git a/eav/models.py b/eav/models.py index 3321616..555617b 100644 --- a/eav/models.py +++ b/eav/models.py @@ -184,6 +184,15 @@ class Attribute(models.Model): """ required = models.BooleanField(verbose_name = _('Required'), default = False) + entity_ct = models.ForeignKey( + ContentType, + null = True, + blank = True, + on_delete = models.PROTECT, + related_name = 'attribute_entities', + verbose_name = _('Belongs to') + ) + enum_group = models.ForeignKey( EnumGroup, verbose_name = _('Choice Group'),