partners changed relation

apps/partner/migrations/0004_auto_20200128_1746.py

@@ -0,0 +1,44 @@
+# Generated by Django 2.2.7 on 2020-01-28 17:46
+
+from django.db import migrations, models
+import django.db.models.deletion
+import django.utils.timezone
+
+
+def make_relations(apps, schemaeditor):
+    PartnerToEstablishment = apps.get_model('partner', 'PartnerToEstablishment')
+    Partner = apps.get_model('partner', 'Partner')
+    Establishment = apps.get_model('establishment', 'Establishment')
+    touched_partners_ids = []
+    for establishment in Establishment.objects.filter(partners__isnull=False):
+        for related_partner in establishment.partners.all():
+            real_partner = Partner.objects.filter(name=related_partner.name, type=related_partner.type,
+                                                  url=related_partner.url).first()
+            touched_partners_ids.append(related_partner.pk)
+            PartnerToEstablishment(establishment=establishment, partner=real_partner, partner_bind_date=real_partner.created).save()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('establishment', '0083_establishment_instagram'),
+        ('partner', '0003_auto_20191121_1059'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='PartnerToEstablishment',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('partner_bind_date', models.DateTimeField(default=django.utils.timezone.now, editable=False, verbose_name='Date partner binded')),
+                ('establishment', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='establishment.Establishment')),
+                ('partner', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='partner.Partner')),
+            ],
+        ),
+        migrations.AddField(
+            model_name='partner',
+            name='establishments',
+            field=models.ManyToManyField(null=True, related_name='new_partners', through='partner.PartnerToEstablishment', to='establishment.Establishment', verbose_name='Establishments'),
+        ),
+        migrations.RunPython(make_relations, migrations.RunPython.noop),
+    ]

apps/partner/migrations/0005_auto_20200128_1754.py

@@ -0,0 +1,45 @@
+# Generated by Django 2.2.7 on 2020-01-28 17:54
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+def delete_unused_partners(apps, schema_editor):
+    PartnerToEstablishment = apps.get_model('partner', 'PartnerToEstablishment')
+    Partner = apps.get_model('partner', 'Partner')
+    ids_to_preserve = []
+    for p_t_e in PartnerToEstablishment.objects.all():
+        ids_to_preserve.append(PartnerToEstablishment.partner.pk)
+    PartnerToEstablishment.objects.exclude(id__in=ids_to_preserve).delete()
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ('location', '0037_address_district_name'),
+        ('establishment', '0083_establishment_instagram'),
+        ('partner', '0004_auto_20200128_1746'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='partner',
+            name='establishments',
+        ),
+        migrations.AddField(
+            model_name='partner',
+            name='country',
+            field=models.ForeignKey(default=None, null=True, on_delete=django.db.models.deletion.SET_NULL,
+                                    to='location.Country'),
+        ),
+        migrations.RemoveField(
+            model_name='partner',
+            name='establishment',
+        ),
+        migrations.AddField(
+            model_name='partner',
+            name='establishment',
+            field=models.ManyToManyField(related_name='partners', through='partner.PartnerToEstablishment',
+                                         to='establishment.Establishment', verbose_name='Establishments'),
+        ),
+        migrations.RunPython(delete_unused_partners, migrations.RunPython.noop, atomic=True)
+    ]

apps/partner/models.py

@@ -1,5 +1,6 @@
 from django.db import models
 from django.utils.translation import gettext_lazy as _
+from django.utils import timezone
 
 from establishment.models import Establishment
 from utils.models import ImageMixin, ProjectBaseMixin
@@ -19,18 +20,14 @@ class Partner(ProjectBaseMixin):
     name = models.CharField(_('name'), max_length=255, blank=True, null=True)
     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,
-        blank=True,
-        null=True,
-    )
+    establishment = models.ManyToManyField('establishment.Establishment', related_name='partners',
+                                           through='PartnerToEstablishment',
+                                           verbose_name=_('Establishments'))
     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)
+    country = models.ForeignKey('location.Country', null=True, default=None, on_delete=models.SET_NULL)
 
     class Meta:
         verbose_name = _('partner')
@@ -38,3 +35,10 @@ class Partner(ProjectBaseMixin):
 
     def __str__(self):
         return f'{self.url}'
+
+
+class PartnerToEstablishment(models.Model):
+    partner_bind_date = models.DateTimeField(default=timezone.now, editable=False,
+                                             verbose_name=_('Date partner binded'))
+    partner = models.ForeignKey(Partner, on_delete=models.CASCADE, null=True)
+    establishment = models.ForeignKey('establishment.Establishment', on_delete=models.CASCADE, null=True)