o
HEDiI; @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ej
r'ddl Z dZ dZ
d ZG d
d d eZG d d
d
eZdedefddZdS )zt
This module has redis storage for finite-state machine based on `redis `_ driver.
N ) BaseStorage)json)
deprecatedstatedatabucketc
@ s e Zd ZdZedddd9dd Zd
d Zd d
Zd:ddZdddde j
e e df de j
e e df de j
fddZddddddde j
e e df de j
e e df fddZddddde j
e e df de j
e e df de je de je fd d Zddddde j
e e df de j
e e df de je de j
fd d Zdddd de j
e e df de j
e e df d!e je j fd"d#Zdddd$de j
e e df de j
e e df d%e j
fd&d'Zdddd$de j
e e df de j
e e df d%e j
fd(d)Zde je je e f fd*d+Zd;d-d.Zd/d0 Zddddde j
e e df de j
e e df de je de j
fd1d2Z dddd3de j
e e df de j
e e df d4e j
fd5d6Z dddd3de j
e e df de j
e e df d4e j
fd7d8Z dS )< RedisStorageaY
Simple Redis-base storage for FSM.
Usage:
.. code-block:: python3
storage = RedisStorage('localhost', 6379, db=5)
dp = Dispatcher(bot, storage=storage)
And need to close Redis connection when shutdown
.. code-block:: python3
await dp.storage.close()
await dp.storage.wait_closed()
zL`RedisStorage` will be removed in aiogram v3.0. Use `RedisStorage2` instead.r
stacklevel localhost Nc K s8 || _ || _|| _|| _|| _|| _d | _t | _ d S N)
_host_port_db _password_ssl_kwargs_redisasyncioLock_connection_lock)selfhostportdbpasswordsslloopkwargs r! p/var/www/www-root/data/www/ovozai.pdev.uz/venv/lib/python3.10/site-packages/aiogram/contrib/fsm_storage/redis.py__init__) s zRedisStorage.__init__c s | j 4 I d H . | jr | jjs+| j W d I d H d S W d I d H d S W d I d H d S 1 I d H s r! r! r" get_state} s zRedisStorage.get_statec s | j ||dI d H }|d S )Nr. r )r9 rD r! r! r" get_data s zRedisStorage.get_datar/ r0 r r c s@ | j ||dI d H }| |}| j||||d dI d H d S )Nr. r r/ r0 r r )r9 rC r? )r r/ r0 r r> r! r! r" set_state s
zRedisStorage.set_stater/ r0 r r c s6 | j ||dI d H }| j|||d |dI d H d S )Nr. r rH r9 r? )r r/ r0 r r> r! r! r" set_data s zRedisStorage.set_datac s` |d u ri }| j ||dI d H }|di }|j|fi | | j|||d |dI d H d S )Nr. r r rH r9 rB updater? )r r/ r0 r r r> record_datar! r! r" update_data s zRedisStorage.update_datac sZ | I dH }g }|ddI dH }|D ]}|dd^ }}}|||f q|S )
Get list of all stored chat's and user's
:return: list of tuples where first element is chat id and second is user id
NKEYSfsm:*zutf-8r2 )r- r5 decodesplitappend)r r8 resultkeysitem_r/ r0 r! r! r" get_states_list s z RedisStorage.get_states_listTc sV | I dH }|r|dI dH dS |ddI dH }|jdg|R I dH dS )i
Reset states in DB
:param full: clean DB or clean only states
:return:
NFLUSHDBrR rS r; )r- r5 )r fullr8 rX r! r! r" reset_all s zRedisStorage.reset_allc C dS r' r! r& r! r! r"
has_bucket zRedisStorage.has_bucketc s" | j ||dI d H }|di S )Nr. r )r9 rB rD r! r! r"
get_bucket s zRedisStorage.get_bucketr/ r0 r r c s< | j ||dI d H }| j|||d |d |dI d H d S )Nr. r r r: rK )r r/ r0 r r> r! r! r"
set_bucket s &zRedisStorage.set_bucketc sb | j ||dI d H }|di }|d u ri }|j|fi | | j|||d ||dI d H d S )Nr. r r r: rM )r r/ r0 r r r>
record_bucketr! r! r"
update_bucket s "zRedisStorage.update_bucket)r r
NNNN)r) r* T) __name__
__module__ __qualname____doc__r r# r% r( r- typingUnionstrintDictr9 r? OptionalrE rF AnyStrrI rL rP ListTupler[ r_ ra rc re rg r! r! r! r" r s
.&
&
&
&
&
&
r c @ s e Zd ZdZ dDdeded eje d
eje d eje d ed
eje j
dedeje deje deje fddZ e ddddEddZ
dd Zd d Zd d Zdddd!d"ejeedf d#ejeedf d$eje deje fd%d&Zdddd!d"ejeedf d#ejeedf d$eje dejfd'd(Zdddd)d"ejeedf d#ejeedf d*ejej fd+d,Zdddd-d"ejeedf d#ejeedf d.ejfd/d0Zdddd-d"ejeedf d#ejeedf d.ejfd1d2Zd3d4 Zdddd!d"ejeedf d#ejeedf d$eje dejfd5d6Zdddd7d"ejeedf d#ejeedf d8ejfd9d:Z dddd7d"ejeedf d#ejeedf d8ejfd;d<Z dFd>d?Z dej ej eef fd@dAZ!dBdC Z"dS )G
RedisStorage2a
Busted Redis-base storage for FSM.
Works with Redis connection pool and customizable keys prefix.
Usage:
.. code-block:: python3
storage = RedisStorage2('localhost', 6379, db=5, pool_size=10, prefix='my_fsm_key')
dp = Dispatcher(bot, storage=storage)
And need to close Redis connection when shutdown
.. code-block:: python3
await dp.storage.close()
r r
N
fsmr r r r r pool_sizer prefix state_ttldata_ttl
bucket_ttlc
K sJ ddl m}
|
d||||||dd| | _|f| _| | _|
| _| | _d S )Nr )RedisT)r r r r r max_connectionsdecode_responsesr! )
redis.asyncior~ r _prefix
_state_ttl _data_ttl _bucket_ttl)r r r r r r ry r rz r{ r| r} r r~ r! r! r" r# s
zRedisStorage2.__init__zWThis method will be removed in aiogram v3.0. You should use your own instance of Redis.r r
r) aioredis.Redisc s | j S r )r r& r! r! r" r- s zRedisStorage2.redisc G s d | jttt| S )Nr2 )joinr tuplemapro )r partsr! r! r" generate_key s zRedisStorage2.generate_keyc s | j I d H d S r )r r% r& r! r! r" r% zRedisStorage2.closec s d S r r! r& r! r! r" r( s zRedisStorage2.wait_closedr@ r/ r0 rA c s> | j ||d\}}| ||t}| j|I d H p | |S Nr. )r4 r STATE_KEYr rB rC )r r/ r0 rA keyr! r! r" rE s zRedisStorage2.get_statec J | j ||d\}}| ||t}| j|I d H }|r!t|S |p$i S r )r4 r STATE_DATA_KEYr rB r r6 r r/ r0 rA r
raw_resultr! r! r" rF
zRedisStorage2.get_datarG r c sd | j ||d\}}| ||t}|d u r | j|I d H d S | jj|| || jdI d H d S Nr. )ex)r4 r r r deletesetrC r )r r/ r0 r r r! r! r" rI & s $zRedisStorage2.set_staterJ r c ` | j ||d\}}| ||t}|r%| jj|t|| jdI d H d S | j|I d H d S r ) r4 r r r r r r= r r )r r/ r0 r r r! r! r" rL / $zRedisStorage2.set_datac sP |d u ri }| j ||i dI d H }|j|fi | | j|||dI d H d S )Nr@ rJ )rF rN rL )r r/ r0 r r temp_datar! r! r" rP 8 s zRedisStorage2.update_datac C r` r' r! r& r! r! r" ra @ rb zRedisStorage2.has_bucketc r r )r4 r STATE_BUCKET_KEYr rB r r6 r r! r! r" rc C r zRedisStorage2.get_bucketrd r c r r ) r4 r r r r r r= r r )r r/ r0 r r r! r! r" re L r zRedisStorage2.set_bucketc sN |d u ri }| j ||dI d H }|j|fi | | j|||dI d H d S )Nr. rd )rc rN re )r r/ r0 r r temp_bucketr! r! r" rg U s zRedisStorage2.update_bucketTc sH |r
| j I dH dS | j | dI dH }| j j| I dH dS )r\ N*)r flushdbrX r r )r r^ rX r! r! r" r_ ^ s
zRedisStorage2.reset_allc sR g }| j | ddtI dH }|D ]}|d^ }}}}|||f q|S )rQ r Nr2 )r rX r r rU rV )r rW rX rY rZ r/ r0 r! r! r" r[ k s z RedisStorage2.get_states_listc s t || I d H d S r )migrate_redis1_to_redis2)r redis1r! r! r"
import_redis1z r zRedisStorage2.import_redis1) r r
NNNrw Nrx NNN)r) r rh )#ri rj rk rl ro rp rm rr boolr AbstractEventLoopr# r r- r r% r( rn rE dictrq rF rs rI rL rP ra rc re rg r_ rt ru r[ r r! r! r! r" rv s
! &
&
&
&
&
&
&
rv storage1storage2c s t | tstt| dt |ts tt| dtd}| I dH D ]N\}}| j||dI dH }|j|||dI dH | j ||dI dH }|j
|||dI dH | j ||dI dH }|j |||dI dH |
d| d | q)dS )
z
Helper for migrating from RedisStorage to RedisStorage2
:param storage1: instance of RedisStorage
:param storage2: instance of RedisStorage2
:return:
z is not RedisStorage instance.zaiogram.RedisStorageNr. rG rJ rd zMigrated user z in chat )
isinstancer TypeErrortypelogging getLoggerr[ rE rI rF rL rc re info)r r logr/ r0 r r r r! r! r" r ~ s
r )rl r r rm dispatcher.storager utilsr utils.deprecatedr
TYPE_CHECKINGr+ r r r r rv r r! r! r! r" s" @ +