From 0695cdce6ba4cc1b3c7764b756168b9fdb229f9e Mon Sep 17 00:00:00 2001
From: alex <birukov.system@gmail.com>
Date: Fri, 17 Jan 2020 11:36:13 +0300
Subject: [PATCH] available_for_events field for employee

---
 .../0074_employee_available_for_events.py      | 18 ++++++++++++++++++
 apps/establishment/models.py                   |  2 ++
 apps/establishment/serializers/back.py         |  3 ++-
 apps/establishment/views/back.py               |  5 ++++-
 4 files changed, 26 insertions(+), 2 deletions(-)
 create mode 100644 apps/establishment/migrations/0074_employee_available_for_events.py

diff --git a/apps/establishment/migrations/0074_employee_available_for_events.py b/apps/establishment/migrations/0074_employee_available_for_events.py
new file mode 100644
index 00000000..cd5622e8
--- /dev/null
+++ b/apps/establishment/migrations/0074_employee_available_for_events.py
@@ -0,0 +1,18 @@
+# Generated by Django 2.2.7 on 2020-01-17 08:33
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('establishment', '0073_auto_20200115_1710'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='employee',
+            name='available_for_events',
+            field=models.BooleanField(default=False, verbose_name='Available for events'),
+        ),
+    ]
diff --git a/apps/establishment/models.py b/apps/establishment/models.py
index 80ac27c2..01615271 100644
--- a/apps/establishment/models.py
+++ b/apps/establishment/models.py
@@ -1070,6 +1070,8 @@ class Employee(BaseAttributes):
                                   verbose_name=_('Tags'))
     # old_id = profile_id
     old_id = models.IntegerField(verbose_name=_('Old id'), null=True, blank=True)
+    available_for_events = models.BooleanField(_('Available for events'), default=False)
+    photo = models
 
     objects = EmployeeQuerySet.as_manager()
 
diff --git a/apps/establishment/serializers/back.py b/apps/establishment/serializers/back.py
index 07117d70..e1818cfc 100644
--- a/apps/establishment/serializers/back.py
+++ b/apps/establishment/serializers/back.py
@@ -286,7 +286,8 @@ class EmployeeBackSerializers(serializers.ModelSerializer):
             'birth_date',
             'email',
             'phone',
-            'toque_number'
+            'toque_number',
+            'available_for_events',
         ]
 
 
diff --git a/apps/establishment/views/back.py b/apps/establishment/views/back.py
index c365caa5..b34cc845 100644
--- a/apps/establishment/views/back.py
+++ b/apps/establishment/views/back.py
@@ -184,7 +184,10 @@ class EmployeesListSearchViews(generics.ListAPIView):
     """Employee search view"""
     pagination_class = None
     permission_classes = (permissions.AllowAny,)
-    queryset = models.Employee.objects.all()
+    queryset = models.Employee.objects.all().prefetch_related(
+        'establishmentemployee_set',
+        'establishmentemployee_set__establishment',
+    )
     filter_class = filters.EmployeeBackSearchFilter
     serializer_class = serializers.EmployeeBackSerializers