o
|
¶oohH ã @ s6 d dl Z d dlZd dlZd dlmZ G dd dZdS )é N)Ú global_configc @ s¼ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-S ).Ú RedisManagerc C s8 t ¡ | _tj| jj| jj| jj| jjdddd| _d S )NTéd é )ÚhostÚportÚdbÚpasswordZdecode_responsesZmax_connectionsZsocket_timeout) r Ú
|
get_configÚconfigÚredisZ StrictRedisZ
|
REDIS_HOSTZ
|
REDIS_PORTZREDIS_DBZREDIS_PASSWORD)Úself© r ú4E:\project\python\wecom-dify-bridge\redis_manager.pyÚ__init__ s
|
ùzRedisManager.__init__c C sf | j j d| d| }t|tr'| j |¡ |r%| jj|g|¢R dS dS | j |t |¡¡ dS )u ç¼å群ç»ç¸å
³æ°æ®ú:group:ú:N)
|
r ÚREDIS_KEY_PREFIXÚ
|
isinstanceÚlistr ÚdeleteZsaddÚsetÚjsonÚdumps)r Úgroup_idÚ data_typeÚdataÚkeyr r r Úcache_group_data s
|
ÿz RedisManager.cache_group_datac C sV | j j d| d| }| ¡ }|dv r | j |¡pg S | j |¡}|r)t |¡S dS )u è·å群ç»ç¸å
³æ°æ®r r )ÚusersÚ sms_accountsN)r r Úlowerr ZsmembersÚgetr Úloads)r r r r r r r r Úget_group_data s zRedisManager.get_group_datac C sN | j j d| d}||t ¡ d}| j |t |¡¡ | j || j j¡ dS )u æ·»å ç¨æ·æ¶æ¯å°Redisú:user_messages:ú :messages)ÚroleÚcontentÚ timestampN) r r Útimer Zrpushr r ÚexpireÚMESSAGE_EXPIRE)r Úuser_idr' r( r Úmessager r r Úadd_user_message) s ýz RedisManager.add_user_messagec C s8 | j j d| d}| j ||¡ | j || j j¡ dS )u æ·»å ä¼è¯idr% ú:conversation_idN)r r r r r+ r, )r r- Úconversation_idr r r r Úadd_conversation_id4 s z RedisManager.add_conversation_idc C s: | j j d| d}| j |dd¡}|rdd |D S g S )u è·åç¨æ·çæææ¶æ¯r% r& r éÿÿÿÿc S s g | ]}t |¡qS r )r r# )Ú.0Úmsgr r r Ú
|
<listcomp>> s z2RedisManager.get_user_messages.<locals>.<listcomp>)r r r Zlrange)r r- r Úmessagesr r r Úget_user_messages: s z RedisManager.get_user_messagesc C ó | j j d| d}| j |¡S )u è·åç¨æ·çconversation_idr% r0 ©r r r r" )r r- r r r r Úget_user_conversation_id@ ó z%RedisManager.get_user_conversation_idc C sB | j j d| }| d}| d}| j |¡ | j |¡ dS )u æ¸
é¤ç¨æ·çæ¶æ¯ç¼år% r& r0 N©r r r r )r r- Z
|
key_prefixÚkey1Úkey2r r r Úclear_user_messagesE s
|
|
|
z RedisManager.clear_user_messagesc C s2 | j j d| }| j |d¡ | j |d¡ dS )õ æ·»å æµå表ú:stream:r é´ N©r r r r r+ ©r Ú stream_idr r r r Ú
|
add_streamM s zRedisManager.add_streamc C s: | j j d| d}| jj|g|¢R | j |d¡ dS )rA rB ú:chunksrC N)r r r Zlpushr+ )r rF Úchunksr r r r Úadd_stream_chunksS s z RedisManager.add_stream_chunksc C r9 )u è·åæµårB rH )r r r ZrpoprE r r r Úget_stream_chunkY r< z RedisManager.get_stream_chunkc C ó4 | j j d| d}| j ||¡ | j |d¡ dS )õ æ·»å ai请æ±çç¶ærB ú:statusrC NrD )r rF Ústatusr r r r Úadd_stream_status^ ó z RedisManager.add_stream_statusc C r9 )rM rB rN r: rE r r r Úget_stream_statusd r< z RedisManager.get_stream_statusc C rL )rM rB r0 rC NrD )r rF r1 r r r r Úadd_stream_conversation_idi rQ z'RedisManager.add_stream_conversation_idc C r9 )rM rB r0 r: rE r r r Úget_stream_conversation_ido r< z'RedisManager.get_stream_conversation_idc C r9 )u æ£æ¥é®æ¯å¦åå¨rB rH ©r r r ÚexistsrE r r r Ú exists_streamt r< zRedisManager.exists_streamc C sD | j j d| d}d}| jj|dD ] }| j |¡ |d7 }qdS )u å é¤stream_idrB z:*r )Úmatché N)r r r Z scan_iterr )r rF ÚpatternÚcountr r r r Údelete_stream_chunksy s
|
þz!RedisManager.delete_stream_chunksc C s6 | j j d| }| j |d¡ | j || j j¡ dS )u æ·»å ä¼ä¸å¾®ä¿¡åè°msgidú:msgid:r N©r r r r r+ Z MSGID_EXPIRE©r Úmsgidr r r r Úadd_wecom_msgid ó z RedisManager.add_wecom_msgidc C s6 | j j d| }| j ||¡ | j || j j¡ dS )u æ´æ°ä¼ä¸å¾®ä¿¡åè°msgidr] Nr^ )r r` Z ai_answerr r r r Úupdate_wecom_msgid rb z RedisManager.update_wecom_msgidc C s" | j j d| }| j |¡ dS )u å é¤ä¼ä¸å¾®ä¿¡åè°msgidr] Nr= r_ r r r Údelete_wecom_msgid s z RedisManager.delete_wecom_msgidc C ó | j j d| }| j |¡S )u è·åmsgidr] r: r_ r r r Úget_wecom_msgid ó z RedisManager.get_wecom_msgidc C re )u msgidæ¯å¦åå¨r] rU r_ r r r Úexists_wecom_msgid rg z RedisManager.exists_wecom_msgidN)Ú__name__Ú
|
__module__Ú __qualname__r r r$ r/ r2 r8 r; r@ rG rJ rK rP rR rS rT rW r\ ra rc rd rf rh r r r r r s.
|
r )r r r* Úconfig_managerr r r r r r Ú<module> s
|
|
