Fix regular old commands

Author: bitterpanda63

aikido_zen/background_process/commands/__init__.py

@@ -23,7 +23,7 @@ def process_incoming_command(connection_manager, obj, conn, queue):
     inbound_request = obj[1]
     if inbound_identifier in commands_map:
         func = commands_map[inbound_identifier]
-        return conn.send(func(connection_manager, inbound_identifier, queue))
+        return conn.send(func(connection_manager, inbound_request))
 
     for cmd in modern_commands:
         if cmd.identifier() == inbound_identifier:

aikido_zen/background_process/commands/check_firewall_lists.py

@@ -1,13 +1,7 @@
 """Exports process_check_firewall_lists"""
 
-from aikido_zen.background_process.cloud_connection_manager import (
-    CloudConnectionManager,
-)
 
-
-def process_check_firewall_lists(
-    connection_manager: CloudConnectionManager, data, conn, queue=None
-):
+def process_check_firewall_lists(connection_manager, data):
     """
     Checks whether an IP is blocked
     data: {"ip": string, "user-agent": string}

aikido_zen/background_process/commands/ping.py

@@ -1,6 +1,6 @@
 """exports `process_ping`"""
 
 
-def process_ping(connection_manager, data, queue=None):
+def process_ping(connection_manager, data):
     """when main process quits , or during testing etc"""
     return "Received"

aikido_zen/background_process/commands/read_property.py

@@ -3,7 +3,7 @@
 from aikido_zen.helpers.logging import logger
 
 
-def process_read_property(connection_manager, data, queue=None):
+def process_read_property(connection_manager, data):
     """
     Takes in one arg : name of property on connection_manager, tries to read it.
     Meant to get config props

aikido_zen/background_process/commands/should_ratelimit.py

@@ -3,7 +3,7 @@
 import aikido_zen.ratelimiting as ratelimiting
 
 
-def process_should_ratelimit(connection_manager, data, queue=None):
+def process_should_ratelimit(connection_manager, data):
     """
     Called to check if the context passed along as data should be rate limited
     data object should be a dict including route_metadata, remote_address and user

aikido_zen/background_process/commands/sync_data.py

@@ -4,7 +4,7 @@
 from aikido_zen.background_process.packages import PackagesStore
 
 
-def process_sync_data(connection_manager, data, conn, queue=None):
+def process_sync_data(connection_manager, data):
     """
     Synchronizes data between the thread-local cache (with a TTL of usually 1 minute) and the
     background thread. Which data gets synced?