rename Batch to BaseBatch

piccolo/engine/base.py

@@ -34,7 +34,7 @@ def validate_savepoint_name(savepoint_name: str) -> None:
         )
 
 
-class Batch(metaclass=ABCMeta):
+class BaseBatch(metaclass=ABCMeta):
     @abstractmethod
     async def __aenter__(self, *args, **kwargs): ...
 
@@ -116,7 +116,7 @@ async def batch(
         query: Query,
         batch_size: int = 100,
         node: t.Optional[str] = None,
-    ) -> Batch:
+    ) -> BaseBatch:
         pass
 
     @abstractmethod

piccolo/engine/postgres.py

@@ -8,8 +8,8 @@
 
 from piccolo.engine.base import (
     BaseAtomic,
+    BaseBatch,
     BaseTransaction,
-    Batch,
     Engine,
     validate_savepoint_name,
 )
@@ -30,7 +30,7 @@
 
 
 @dataclass
-class AsyncBatch(Batch):
+class AsyncBatch(BaseBatch):
     connection: Connection
     query: Query
     batch_size: int

piccolo/engine/sqlite.py

@@ -15,8 +15,8 @@
 
 from piccolo.engine.base import (
     BaseAtomic,
+    BaseBatch,
     BaseTransaction,
-    Batch,
     Engine,
     validate_savepoint_name,
 )
@@ -317,7 +317,7 @@ def convert_M2M_out(value: str) -> t.List:
 
 
 @dataclass
-class AsyncBatch(Batch):
+class AsyncBatch(BaseBatch):
     connection: Connection
     query: Query
     batch_size: int

piccolo/query/methods/objects.py

@@ -5,7 +5,7 @@
 from piccolo.columns.column_types import ForeignKey
 from piccolo.columns.combination import And, Where
 from piccolo.custom_types import Combinable, TableInstance
-from piccolo.engine.base import Batch
+from piccolo.engine.base import BaseBatch
 from piccolo.query.base import Query
 from piccolo.query.methods.select import Select
 from piccolo.query.mixins import (
@@ -298,7 +298,7 @@ async def batch(
         batch_size: t.Optional[int] = None,
         node: t.Optional[str] = None,
         **kwargs,
-    ) -> Batch:
+    ) -> BaseBatch:
         if batch_size:
             kwargs.update(batch_size=batch_size)
         if node:

piccolo/query/methods/select.py

@@ -9,7 +9,7 @@
 from piccolo.columns.m2m import M2MSelect
 from piccolo.columns.readable import Readable
 from piccolo.custom_types import TableInstance
-from piccolo.engine.base import Batch
+from piccolo.engine.base import BaseBatch
 from piccolo.query.base import Query
 from piccolo.query.mixins import (
     AsOfDelegate,
@@ -442,7 +442,7 @@ async def batch(
         batch_size: t.Optional[int] = None,
         node: t.Optional[str] = None,
         **kwargs,
-    ) -> Batch:
+    ) -> BaseBatch:
         if batch_size:
             kwargs.update(batch_size=batch_size)
         if node: