From ea427d6451521a46081c0e65685a48380db790eb Mon Sep 17 00:00:00 2001
From: littlewolf <wolfverwolf@mail.ru>
Date: Thu, 3 Oct 2019 16:35:54 +0300
Subject: [PATCH] Update module

---
 apps/transfer/management/commands/transfer.py | 26 ++++++++++++-------
 apps/transfer/utils.py                        |  1 +
 2 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/apps/transfer/management/commands/transfer.py b/apps/transfer/management/commands/transfer.py
index 13cfbf90..79a0185e 100644
--- a/apps/transfer/management/commands/transfer.py
+++ b/apps/transfer/management/commands/transfer.py
@@ -9,20 +9,28 @@ from transfer.utils import transfer_objects
 class Command(BaseCommand):
     help = 'Transfer data between databases'
 
+    types = [
+        'dictionaries',
+        'keys'
+    ]
+
     def handle(self, *args, **options):
+        option = set(options.keys()) & set(self.types)
+        print(option)
         if options['dictionaries']:
-            self._transfer_dictionaries()
+            self._transfer_objects()
 
     def add_arguments(self, parser):
-        parser.add_argument(
-            '-d',
-            '--dictionaries',
-            action='store_true',
-            default=False,
-            help='Вывод короткого сообщения'
-        )
+        for type in self.types:
+            parser.add_argument(
+                f'-{type[:1]}',
+                f'--{type}',
+                action='store_true',
+                default=False,
+                help=f'Transfer {type} objects'
+            )
 
-    def _transfer_dictionaries(self):
+    def _transfer_objects(self):
         for app in apps.get_app_configs():
             if exists(f"{app.path}/transfer.py"):
                 card_module = SourceFileLoader("transfer", f"{app.path}/transfer.py").load_module()
diff --git a/apps/transfer/utils.py b/apps/transfer/utils.py
index 639a2ccd..d31075bf 100644
--- a/apps/transfer/utils.py
+++ b/apps/transfer/utils.py
@@ -1,6 +1,7 @@
 from collections import OrderedDict
 
 def transfer_objects(app, transfer_card):
+    print(app)
     for model, card in transfer_card.items():
         print(model)
         print(f"\t{card}")