python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add set_state

Browse source
This commit is contained in:
Oleg Matviichuk 2025-07-30 00:04:23 +02:00
parent 5412c1a1b2
commit 35b615158f
1 changed files with 16 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

20
aiogram/fsm/storage/sqlite.py
View file

@ -2,6 +2,7 @@ from typing import Any, Dict, Mapping, Optional, cast
from aiosqlite import Connection, connect
from aiogram.fsm.state import State
from aiogram.fsm.storage.base import (
BaseEventIsolation,
BaseStorage,
@ -48,9 +49,9 @@ class SqliteStorage(BaseStorage):
connection = await connect(db_filename)
await connection.execute(
f"""CREATE TABLE IF NOT EXISTS aiogram_fsm (
id TEXT PRIMARY KEY,
state BLOB,
data BLOB)"""
id TEXT PRIMARY KEY,
state TEXT,
data BLOB)"""
)
await connection.commit()
return cls(connection=connection)
@ -59,7 +60,18 @@ class SqliteStorage(BaseStorage):
await self._connection.close()
async def set_state(self, key: StorageKey, state: StateType = None) -> None:
pass
id = self._key_builder.build(key)
state = cast(str, state.state if isinstance(state, State) else state)
await self._connection.execute(
f"""INSERT INTO aiogram_fsm (id, state)
VALUES (?, ?)
ON CONFLICT (id)
DO UPDATE SET state = ?""",
(id, state, state),
)
await self._connection.commit()
async def get_state(self, key: StorageKey) -> Optional[str]:
pass