From 09647fa4d361b061a361668a36b3c849eab8c8c5 Mon Sep 17 00:00:00 2001
From: alex <birukov.system@gmail.com>
Date: Fri, 25 Oct 2019 12:03:43 +0300
Subject: [PATCH] start transfer_data

---
 apps/establishment/transfer_data.py | 39 +++++++++++++++++++++++++----
 1 file changed, 34 insertions(+), 5 deletions(-)

diff --git a/apps/establishment/transfer_data.py b/apps/establishment/transfer_data.py
index 5f96fcaa..ebb13c4a 100644
--- a/apps/establishment/transfer_data.py
+++ b/apps/establishment/transfer_data.py
@@ -1,16 +1,45 @@
 from pprint import pprint
 
+from transfer.models import Establishments
 from transfer.serializers.establishment import EstablishmentSerializer
 
 
 def transfer_establishment():
     result = []
 
-    serialized_data = EstablishmentSerializer(data=result, many=True)
-    if serialized_data.is_valid():
-        serialized_data.save()
-    else:
-        pprint(f"Establishment serializer errors: {serialized_data.errors}")
+    old_establishments = Establishments.objects.all().prefetch_related('establishmentinfos_set', 'schedules_set')
+    for item in old_establishments:
+        data = {
+            'name': item.name,
+            'slug': item.slug,
+            'type': item.type,
+            'location': item.location.id,
+            'schedules': [],
+        }
+        info = item.establishmentinfos_set.first()
+        if info:
+            data.update({
+                'website': info.website,
+                'facebook': info.facebook,
+                'twitter': info.twitter,
+                'lafourchette': info.lafourchette,
+                'booking': info.booking_url,
+            })
+        for schedule in item.schedules_set.all():
+            data['schedules'].append({
+                'raw_timetable': schedule.timetable
+            })
+        result.append(data)
+
+    print('-' * 30)
+    print(len(result))
+    pprint(result[0])
+
+    # serialized_data = EstablishmentSerializer(data=result, many=True)
+    # if serialized_data.is_valid():
+    #     serialized_data.save()
+    # else:
+    #     pprint(f"Establishment serializer errors: {serialized_data.errors}")
 
 
 data_types = {