partner model

apps/partner/admin.py

@@ -6,3 +6,4 @@ from partner import models
 @admin.register(models.Partner)
 class PartnerModelAdmin(admin.ModelAdmin):
     """Model admin for Partner model."""
+    raw_id_fields = ('establishment',)

apps/partner/migrations/0003_auto_20191121_0929.py

@@ -0,0 +1,46 @@
+# Generated by Django 2.2.7 on 2019-11-21 09:29
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('establishment', '0065_establishment_purchased_products'),
+        ('partner', '0002_auto_20191101_0939'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='partner',
+            name='establishment',
+            field=models.ForeignKey(default=1, on_delete=django.db.models.deletion.CASCADE, related_name='partners', to='establishment.Establishment', verbose_name='Establishment'),
+            preserve_default=False,
+        ),
+        migrations.AddField(
+            model_name='partner',
+            name='expiry_date',
+            field=models.DateField(blank=True, null=True, verbose_name='expiry date'),
+        ),
+        migrations.AddField(
+            model_name='partner',
+            name='old_id',
+            field=models.PositiveIntegerField(blank=True, default=None, null=True, verbose_name='old id'),
+        ),
+        migrations.AddField(
+            model_name='partner',
+            name='price_per_month',
+            field=models.DecimalField(blank=True, decimal_places=2, max_digits=10, null=True, verbose_name='price per month'),
+        ),
+        migrations.AddField(
+            model_name='partner',
+            name='starting_date',
+            field=models.DateField(blank=True, null=True, verbose_name='starting date'),
+        ),
+        migrations.AddField(
+            model_name='partner',
+            name='type',
+            field=models.PositiveSmallIntegerField(choices=[(0, 'Partner'), (1, 'Sponsor')], default=0),
+        ),
+    ]

apps/partner/models.py

@@ -1,13 +1,33 @@
 from django.db import models
 from django.utils.translation import gettext_lazy as _
 
+from establishment.models import Establishment
 from utils.models import ImageMixin, ProjectBaseMixin
 
 
 class Partner(ProjectBaseMixin):
     """Partner model."""
+
+    PARTNER = 0
+    SPONSOR = 1
+    MODEL_TYPES = (
+        (PARTNER, _('Partner')),
+        (SPONSOR, _('Sponsor')),
+    )
+
+    old_id = models.PositiveIntegerField(_('old id'), blank=True, null=True, default=None)
     url = models.URLField(verbose_name=_('Partner URL'))
     image = models.URLField(verbose_name=_('Partner image URL'), null=True)
+    establishment = models.ForeignKey(
+        Establishment,
+        verbose_name=_('Establishment'),
+        related_name='partners',
+        on_delete=models.CASCADE,
+    )
+    type = models.PositiveSmallIntegerField(choices=MODEL_TYPES, default=PARTNER)
+    starting_date = models.DateField(_('starting date'), blank=True, null=True)
+    expiry_date = models.DateField(_('expiry date'), blank=True, null=True)
+    price_per_month = models.DecimalField(_('price per month'), max_digits=10, decimal_places=2, blank=True, null=True)
 
     class Meta:
         verbose_name = _('partner')