o
|
2ooh{ ã @ sb d dl mZ d dlmZ d dlmZmZmZmZm Z m
|
Z
|
d dl m Z d dl mZ G dd dZdS ) é )Ú create_engine)Ú sessionmaker)ÚBaseÚ ChatGroupÚUserÚ GroupUserÚGroupSMSAccountÚMessageHistory)Ú RedisManager)Ú global_configc @ s$ e Zd Zdd Zdd Zdd ZdS )ÚDatabaseManagerc C sT t ¡ }td|j d|j d|j d|j d|j
|
| _t | jd| _
|
t | _ d S )Nzmysql+pymysql://ú:ú@ú/)Úbind) r Ú
|
get_configr Z
|
MYSQL_USERZMYSQL_PASSWORDZ
|
MYSQL_HOSTZ
|
MYSQL_PORTZMYSQL_DBÚenginer ÚSessionr
|
Úredis)ÚselfÚconfig© r ú/E:\project\python\wecom-dify-bridge\database.pyÚ__init__ s *ÿ zDatabaseManager.__init__c C s t j | j¡ dS )u åå§åæ°æ®åºè¡¨N)r ÚmetadataZ
|
create_allr )r r r r Úinit_db s zDatabaseManager.init_dbc
|
C s | ¡ }z6z |D ]}t|||d |d d}| |¡ q| ¡ W n ty3 } z| ¡ |d}~ww W | ¡ dS | ¡ w )u ä¿åæ¶æ¯åå²å°æ°æ®åºÚroleÚcontent)Úuser_idÚconversation_idr r N)r r ÚaddZcommitÚ ExceptionZrollbackÚclose)r r Úmessagesr ÚsessionÚmsgZnew_msgÚer r r Úsave_message_history/ s$ ü þÿz$DatabaseManager.save_message_historyN)Ú__name__Ú
|
__module__Ú __qualname__r r r' r r r r r s r N)Z
|
sqlalchemyr Zsqlalchemy.ormr Zmodelsr r r r r r Ú redis_managerr
|
Úconfig_managerr r r r r r Ú<module> s
|
