chore: Simplify finish_kwargs to finish_args

open-contracting · Jan 4, 2024 · 7c5ffda · 7c5ffda
1 parent 61e447a
commit 7c5ffda
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/process/management/commands/file_worker.py b/process/management/commands/file_worker.py
@@ -66,7 +66,7 @@ def callback(client_state, channel, method, properties, input_message):
         with delete_step(
             ProcessingStep.Name.LOAD,
             finish=finish,
-            finish_kwargs={"collection_id": collection_id, "collection_file_id": collection_file_id},
+            finish_args=(collection_id, collection_file_id),
             collection_file_id=collection_file_id,
         ):
             with transaction.atomic():

diff --git a/process/util.py b/process/util.py
@@ -127,7 +127,7 @@ def create_step(name, collection_id, **kwargs):
 
 
 @contextmanager
-def delete_step(*args, finish=None, finish_kwargs={}, **kwargs):
+def delete_step(*args, finish=None, finish_args=(), **kwargs):
     try:
         yield
     # See the errback() function in the decorator() function. If a duplicate message is received, delete the step, so
@@ -139,7 +139,7 @@ def delete_step(*args, finish=None, finish_kwargs={}, **kwargs):
         _delete_step(*args, **kwargs)
     finally:
         if finish:
-            finish(**finish_kwargs)
+            finish(*finish_args)
 
 
 def _delete_step(step_type, **kwargs):