From 23f8c0bbcdc380c78a78db5c49c5852aec02c8b4 Mon Sep 17 00:00:00 2001
From: Anatoly <a.feteleu@spider.ru>
Date: Fri, 10 Jan 2020 14:40:45 +0300
Subject: [PATCH] added field address_id to request data in
 /api/back/establishments/ in Establishment make as mandatory field address

---
 .../migrations/0071_auto_20200110_1055.py     | 19 +++++++++++++++++++
 apps/establishment/models.py                  |  2 +-
 apps/establishment/serializers/back.py        |  5 +++++
 3 files changed, 25 insertions(+), 1 deletion(-)
 create mode 100644 apps/establishment/migrations/0071_auto_20200110_1055.py

diff --git a/apps/establishment/migrations/0071_auto_20200110_1055.py b/apps/establishment/migrations/0071_auto_20200110_1055.py
new file mode 100644
index 00000000..a35fc17e
--- /dev/null
+++ b/apps/establishment/migrations/0071_auto_20200110_1055.py
@@ -0,0 +1,19 @@
+# Generated by Django 2.2.7 on 2020-01-10 10:55
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('establishment', '0070_auto_20191227_1443'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='establishment',
+            name='address',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.PROTECT, to='location.Address', verbose_name='address'),
+        ),
+    ]
diff --git a/apps/establishment/models.py b/apps/establishment/models.py
index 551f475c..5331ac6e 100644
--- a/apps/establishment/models.py
+++ b/apps/establishment/models.py
@@ -493,7 +493,7 @@ class Establishment(GalleryMixin, ProjectBaseMixin, URLImageMixin,
                                                     blank=True,
                                                     related_name='subtype_establishment',
                                                     verbose_name=_('subtype'))
-    address = models.ForeignKey(Address, blank=True, null=True, default=None,
+    address = models.ForeignKey(Address, null=True,
                                 on_delete=models.PROTECT,
                                 verbose_name=_('address'))
     price_level = models.PositiveIntegerField(blank=True, null=True,
diff --git a/apps/establishment/serializers/back.py b/apps/establishment/serializers/back.py
index aeb947c0..23283981 100644
--- a/apps/establishment/serializers/back.py
+++ b/apps/establishment/serializers/back.py
@@ -21,6 +21,11 @@ class EstablishmentListCreateSerializer(model_serializers.EstablishmentBaseSeria
         queryset=models.EstablishmentType.objects.all(),
         write_only=True
     )
+    address_id = serializers.PrimaryKeyRelatedField(
+        source='address',
+        queryset=models.Address.objects.all(),
+        write_only=True
+    )
     phones = model_serializers.ContactPhonesSerializer(read_only=True,
                                                        many=True, )
     emails = model_serializers.ContactEmailsSerializer(read_only=True,