o i$@sddlmZmZddlZddlmZmZddlm Z ddZ ddZ d e d e fd d Zd e d e fddZd e d e fddZd e de fddZdS))ListDictN)get_db_connectionlogger) A_DB_CONFIGc Csz'tjjdit}|}d}|||||f|||WdStyB}zt d|WYd}~dSd}~ww)NzJINSERT INTO chat_history (session_id, sender, message) VALUES (%s, %s, %s)zChat DB Save Error: ) mysql connectorconnectrcursorexecutecommitclose Exceptionrerror) session_idsendermessageconnr sqlerr7/home/air/goalskill_t/back/app/models/general_module.py save_chat_logsrc Csz'tjjdit}|jdd}d}|||f|}|||WStyC}zt d|gWYd}~Sd}~ww)NT dictionaryzNSELECT sender, message FROM chat_history WHERE session_id = %s ORDER BY id ASCzChat DB Load Error: r) rr r rr r fetchallrrrr)rrr rrowsrrrrget_history_from_dbs rrreturnc Csd}zmzEtjjdit}|jdd}d}|||f|}|dur4WW|r2||dSdS|ddp;dWW|rH||SSt yo}zt d|WYd}~W|ri||dSdSd}~ww|r{||ww)NTrzr SELECT a_part_status FROM user_profile_summary WHERE session_id = %s r a_part_statuszGet A Part Status Error: r) rr r rr r fetchonergetrprintrrr rrowrrrrget_a_part_status#s>      r%statusc Csd}zFztjjdit}|}d}||||f|Wnty9}z td|WYd}~nd}~wwW|rG| | dSdS|rT| | ww)Nz INSERT INTO user_profile_summary (session_id, a_part_status) VALUES (%s, %s) ON DUPLICATE KEY UPDATE a_part_status = VALUES(a_part_status), updated_at = NOW() zUpdate A Part Status Error: r) rr r rr r r rr"r)rr&rr rrrrrupdate_a_part_status;s&   r'c Csd}zmzDtjjdit}|jdd}d}|||f|}|r8|dr8|dWW|r7||SSWW|rF||dSdSt yo}zt d|WYd}~W|ri||dSdSd}~ww|r{||ww)u유저 이름 조회NTrzm SELECT username FROM user_profile_summary WHERE session_id = %s usernamezGet Username Error: r) rr r rr r r r!rrrrr#rrr get_usernamePs>       r)r(c Csd}zGztjjdit}|}d}||||f|Wnty:}zt d|WYd}~nd}~wwW|rH| | dSdS|rU| | ww)u유저 이름 저장Nz INSERT INTO user_profile_summary (session_id, username) VALUES (%s, %s) ON DUPLICATE KEY UPDATE username = VALUES(username), updated_at = NOW() zSave Username Error: r) rr r rr r r rrrr)rr(rr rrrrr save_usernamehs&   r*)typingrrmysql.connectorrapp.corerrapp.core.configrrrstrintr%r'r)r*rrrrs