Remove unnecessary calls to mk_vumi_api()

praekeltfoundation · Jun 22, 2015 · 1dc4494 · 1dc4494
1 parent 16d4d69
commit 1dc4494
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/go/base/management/commands/go_manage_message_cache.py b/go/base/management/commands/go_manage_message_cache.py
@@ -110,6 +110,6 @@ def _apply_command(self, func, dry_run=None):
 
     def handle_command_rebuild(self, *args, **options):
         def rebuild(batch_id):
-            self.mk_vumi_api().mdb.reconcile_cache(batch_id)
+            self.vumi_api.mdb.reconcile_cache(batch_id)
 
         self._apply_command(rebuild)
diff --git a/go/base/management/commands/go_send_app_worker_command.py b/go/base/management/commands/go_send_app_worker_command.py
@@ -9,7 +9,7 @@ class Command(BaseGoCommand):
     help = "Send a VumiApi command to an application worker"
     args = "<worker-name> <command> key1=value1 key2=value2"
 
-    def handle(self, worker_name, command, *parameters, **config):
+    def handle_no_command(self, worker_name, command, *parameters, **config):
         handler = getattr(self, 'handle_%s' % (command,), None)
         if handler is None:
             raise CommandError('Unknown command %s' % (command,))
@@ -24,7 +24,7 @@ def handle(self, worker_name, command, *parameters, **config):
                 args.append(parameter)
 
         cmd = handler(worker_name, command, *args, **kwargs)
-        self.mk_vumi_api().mapi.send_command(cmd)
+        self.vumi_api.mapi.send_command(cmd)
 
     def handle_reconcile_cache(self, worker_name, command, account_key,
                                conversation_key):