From 56b3d0dcc25f0662f390600b8660911f57c33869 Mon Sep 17 00:00:00 2001
From: GladkihViktor <vgladkih7@gmail.com>
Date: Sat, 2 Nov 2019 17:54:24 +0300
Subject: [PATCH 1/2] Update transportation

---
 .../management/commands/upd_transportation.py | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 apps/establishment/management/commands/upd_transportation.py

diff --git a/apps/establishment/management/commands/upd_transportation.py b/apps/establishment/management/commands/upd_transportation.py
new file mode 100644
index 00000000..1360d3b4
--- /dev/null
+++ b/apps/establishment/management/commands/upd_transportation.py
@@ -0,0 +1,29 @@
+from django.core.management.base import BaseCommand
+
+from establishment.models import Establishment
+from transfer.models import Establishments as OldEst
+
+class Command(BaseCommand):
+    help = 'Update transportation column in establishment'
+
+    def handle(self, *args, **kwargs):
+        count = 0
+        raw_qs = OldEst.objects.raw(
+            '''
+                select
+                    e.id,
+                    l.transportation 
+                from establishments e 
+                join locations l on l.id = e.location_id 
+                where trim(l.transportation) is not null 
+                    and length(trim(l.transportation)) >1;
+            '''
+        )
+        queryset = [vars(query) for query in raw_qs]
+        for obj in queryset:
+            establishment = Establishment.objects.filter(old_id=obj['id']).first()
+            if establishment:
+                establishment.transportation = obj['transportation']
+                count += 1
+                establishment.save()
+        self.stdout.write(self.style.WARNING(f'Updated {count} objects.'))
\ No newline at end of file

From 9acf2f5c5fc8ab4a46518137a109e7005449670c Mon Sep 17 00:00:00 2001
From: GladkihViktor <vgladkih7@gmail.com>
Date: Sat, 2 Nov 2019 17:58:00 +0300
Subject: [PATCH 2/2] Code style

---
 apps/establishment/management/commands/upd_transportation.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/apps/establishment/management/commands/upd_transportation.py b/apps/establishment/management/commands/upd_transportation.py
index 1360d3b4..4f400ee8 100644
--- a/apps/establishment/management/commands/upd_transportation.py
+++ b/apps/establishment/management/commands/upd_transportation.py
@@ -3,6 +3,7 @@ from django.core.management.base import BaseCommand
 from establishment.models import Establishment
 from transfer.models import Establishments as OldEst
 
+
 class Command(BaseCommand):
     help = 'Update transportation column in establishment'