From 08ca1dc35641978fa8e5ffabd89a2cf58a8e9668 Mon Sep 17 00:00:00 2001
From: littlewolf <nickptic@code-artel.com>
Date: Thu, 31 Oct 2019 07:47:45 +0300
Subject: [PATCH] Move addresses to new key

---
 apps/establishment/transfer_data.py           | 2 ++
 apps/transfer/management/commands/transfer.py | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/apps/establishment/transfer_data.py b/apps/establishment/transfer_data.py
index 000df561..6e345e27 100644
--- a/apps/establishment/transfer_data.py
+++ b/apps/establishment/transfer_data.py
@@ -127,6 +127,8 @@ def transfer_establishment_addresses():
 data_types = {
     "establishment": [
         transfer_establishment,
+    ],
+    "location_establishment": [
         transfer_establishment_addresses
     ],
 
diff --git a/apps/transfer/management/commands/transfer.py b/apps/transfer/management/commands/transfer.py
index 8af8208d..f3abbfab 100644
--- a/apps/transfer/management/commands/transfer.py
+++ b/apps/transfer/management/commands/transfer.py
@@ -20,7 +20,8 @@ class Command(BaseCommand):
         'commercial',
         'overlook',
         'tmp',
-        'menu'
+        'menu',
+        'location_establishment'
     ]
 
     def handle(self, *args, **options):