Provide a nicer, type-safe way of adding calls to the batch

Author: chriso

src/dispatch/function.py

@@ -359,8 +359,12 @@ def __init__(self, client: Client):
         self.client = client
         self.calls: list[Call] = []
 
-    def add(self, call: Call) -> Batch:
-        """Add a call to the batch."""
+    def add(self, func: Function[P, T], *args: P.args, **kwargs: P.kwargs) -> Batch:
+        """Add a call to the specified function to the batch."""
+        return self.add_call(func.build_call(*args, correlation_id=None, **kwargs))
+
+    def add_call(self, call: Call) -> Batch:
+        """Add a Call to the batch."""
         self.calls.append(call)
         return self
 

tests/test_client.py

@@ -75,9 +75,9 @@ def test_call_pickle(self):
     def test_batch(self):
         batch = self.dispatch_client.batch()
 
-        batch.add(Call(function="my-function", input=42)).add(
+        batch.add_call(Call(function="my-function", input=42)).add_call(
             Call(function="my-function", input=23)
-        ).add(Call(function="my-function2", input=11))
+        ).add_call(Call(function="my-function2", input=11))
 
         dispatch_ids = batch.dispatch()
         self.assertEqual(len(dispatch_ids), 3)