o
    ]yi  ã                   @   sX   d dl mZmZ d dlZd dlmZmZ d dlm	Z	 dd„ Z
dd„ Zd	d
„ Zdd„ ZdS )é    )ÚListÚDictN)Úget_db_connectionÚlogger)ÚINPUT_DB_CONFIGc              
   C   sˆ   z(t jjdi t¤Ž}| ¡ }d}| || |||f¡ | ¡  | ¡  | ¡  W d S  tyC } zt	 
d|› ¡ W Y d }~d S d }~ww )NzTINSERT INTO chat_history (session_id, mode, sender, message) VALUES (%s, %s, %s, %s)zChat DB Save Error: © ©ÚmysqlÚ	connectorÚconnectr   ÚcursorÚexecuteÚcommitÚcloseÚ	Exceptionr   Úerror)Ú
session_idÚmodeÚsenderÚmessageÚconnr   ÚsqlÚer   r   ú2/home/air/goalskill_t/back/app/models/db_module.pyÚsave_chat_log   s   €ÿr   c              
   C   sˆ   z't jjdi t¤Ž}|jdd}d}| || f¡ | ¡ }| ¡  | ¡  |W S  tyC } zt	 
d|› ¡ g W  Y d }~S d }~ww )NT©Ú
dictionaryzNSELECT sender, message FROM chat_history WHERE session_id = %s ORDER BY id ASCzChat DB Load Error: r   ©r	   r
   r   r   r   r   Úfetchallr   r   r   r   )r   r   r   r   r   Úrowsr   r   r   r   Úget_history_from_db   ó   €þr    c              
   C   s†   z't jjdi t¤Ž}| ¡ }d}| || ||f¡ | ¡  | ¡  | ¡  W d S  tyB } zt	 
d|› ¡ W Y d }~d S d }~ww )NzLINSERT INTO chat_report_jp (session_id, sender, message) VALUES (%s, %s, %s)zReport Log Save Error: r   r   )r   r   r   r   r   r   r   r   r   r   Úsave_report_log$   s   €ÿr"   c              
   C   sˆ   z't jjdi t¤Ž}|jdd}d}| || f¡ | ¡ }| ¡  | ¡  |W S  tyC } zt	 
d|› ¡ g W  Y d }~S d }~ww )NTr   zPSELECT sender, message FROM chat_report_jp WHERE session_id = %s ORDER BY id ASCzReport DB Load Error: r   r   )r   r   r   r   r   r   r   r   r   Úget_report_from_db1   r!   r#   )Útypingr   r   Úmysql.connectorr	   Úapp.corer   r   Úapp.core.configr   r   r    r"   r#   r   r   r   r   Ú<module>   s    