init new bot

This commit is contained in:
ada-dmitry
2024-08-26 20:22:16 +03:00
parent 8208bd226e
commit 564398d0dd
1675 changed files with 164420 additions and 115 deletions
+32
View File
@@ -0,0 +1,32 @@
from typing import Any, Dict, Optional
from aiogram.fsm.storage.base import BaseStorage, StateType, StorageKey
class FSMContext:
def __init__(self, storage: BaseStorage, key: StorageKey) -> None:
self.storage = storage
self.key = key
async def set_state(self, state: StateType = None) -> None:
await self.storage.set_state(key=self.key, state=state)
async def get_state(self) -> Optional[str]:
return await self.storage.get_state(key=self.key)
async def set_data(self, data: Dict[str, Any]) -> None:
await self.storage.set_data(key=self.key, data=data)
async def get_data(self) -> Dict[str, Any]:
return await self.storage.get_data(key=self.key)
async def update_data(
self, data: Optional[Dict[str, Any]] = None, **kwargs: Any
) -> Dict[str, Any]:
if data:
kwargs.update(data)
return await self.storage.update_data(key=self.key, data=kwargs)
async def clear(self) -> None:
await self.set_state(state=None)
await self.set_data({})