|
|
@@ -85,13 +85,13 @@ class DataStore(Generic[DataType]):
|
|
|
await self.connection.commit()
|
|
|
|
|
|
|
|
|
-class DataFor(Injector[Optional[DataType]]):
|
|
|
+class DataFor(Injector[DataType]):
|
|
|
def __init__(self, datatype: Type[DataType], key: Injector[str], kwargs: dict[str, Any]):
|
|
|
self.datatype = datatype
|
|
|
self.key = key
|
|
|
self.kwargs = kwargs
|
|
|
|
|
|
- async def inject(self, message: Message, context: Context) -> DataStore[DataType]:
|
|
|
+ async def inject(self, message: Message, context: Context) -> DataType:
|
|
|
key = await self.key.inject(message, context)
|
|
|
async with context.database() as db:
|
|
|
store = DataStore(self.datatype, db)
|
