o
    ii+                     @   s<   d dl mZmZ d dlZd dlmZmZ dd Zdd Z	dS )    )ListDictN)get_db_connectionloggerc              
   C   sz   z!t  }| }d}||| ||f |  |  |  W d S  ty< } ztd|  W Y d }~d S d }~ww )NzJINSERT INTO chat_history (session_id, sender, message) VALUES (%s, %s, %s)zChat DB Save Error: )r   cursorexecutecommitclose	Exceptionr   error)
session_idmodesendermessageconnr   sqle r   2/home/air/goalskill_t/back/app/models/db_module.pysave_chat_log   s   r   c              
   C   s|   z!t  }|jdd}d}||| f | }|  |  |W S  ty= } 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   fetchallr	   r
   r   r   )r   r   r   r   r   rowsr   r   r   r   get_history_from_db   s   r   )
typingr   r   mysql.connectormysqlapp.corer   r   r   r   r   r   r   r   <module>   s
    