update exception

layercake/layercake/dynamodb.py

@@ -512,7 +512,7 @@ class DynamoDBPersistenceLayer:
             return True
 
 
-class MissingError(ValueError):
+class MissingError(Exception):
     pass
 
 
@@ -556,14 +556,9 @@ class DynamoDBCollection:
         self,
         persistence_layer: DynamoDBPersistenceLayer,
         /,
-        exception_cls: Type[ValueError] = MissingError,
+        exception_cls: Type[Exception] = MissingError,
         tz: str = TZ,
     ) -> None:
-        if not issubclass(exception_cls, ValueError):
-            raise TypeError(
-                f'exception_cls must be a subclass of ValueError, got {exception_cls}'
-            )
-
         self.persistence_layer = persistence_layer
         self.exception_cls = exception_cls
         self.tz = tz
@@ -572,11 +567,13 @@ class DynamoDBCollection:
         self,
         key: Key,
         path_spec: str | None = None,
+        /,
         raise_on_error: bool = True,
+        exception_cls: Type[Exception] | None = None,
         default: Any = None,
         delimiter: str = '#',
     ) -> Any:
-        exc_cls = self.exception_cls
+        exc_cls = exception_cls or self.exception_cls
         data = self.persistence_layer.get_item(key)
 
         if raise_on_error and not data:
@@ -592,6 +589,7 @@ class DynamoDBCollection:
     def put_item(
         self,
         key: Key,
+        *,
         ttl: int | datetime | None = None,
         **kwargs: Any,
     ) -> bool: