o Jh @sddlmZddlmZddlmZdefddZdefddZdefd d Z ded efd d Z defddZ defddZ ded ede de fddZde fddZde defddZde de fddZd S)!)database) HTTPException)Database lineuser_idc sNzd}tj|d|idIdHWdSty&}ztdt|dd}~ww)Nz1DELETE FROM chat WHERE lineuser_id = :lineuser_idrvalues status_codedetailrexecute Exceptionrstrrqueryer#/home/air/V2/geminiapi/db_module.pydelete_chat_historysrc Hzd}t|d|iIdHWSty#}ztdt|dd}~ww)Nz SELECT message, response FROM chat WHERE lineuser_id = :lineuser_id ORDER BY created_at ASC rrr r fetch_allrrrrrrrget_chat_historysrc r)Nz SELECT message, response FROM chat WHERE lineuser_id = :lineuser_id AND status != 99 ORDER BY created_at ASC rrr rrrrrget_chat_history_exclude_resultsrstatusc sXzd}t|||dIdH}|sgWS|WSty+}ztdt|dd}~ww)Nz SELECT message, response FROM chat WHERE lineuser_id = :lineuser_id and status = :status ORDER BY created_at ASC )rrrr r)rrrresultsrrrrget_chat_history_status&src Vzd}t|d|iIdH}|sgWS|WSty*}ztdt|dd}~ww)Nz SELECT message, response FROM chat WHERE lineuser_id = :lineuser_id AND status = 'QUIZ' ORDER BY created_at ASC rrr rrrrrrrrget_quiz_chat_history4r c r)Nz SELECT message, response FROM chat WHERE lineuser_id = :lineuser_id AND status != 1 ORDER BY created_at ASC rrr rrrrrget_non_status_one_chat_historyCr!r"messageresponsec sTzd}tj|||||ddIdHWdSty)}ztdt|dd}~ww)Nz INSERT INTO chat (lineuser_id, status, message, response, created_at) VALUES (:lineuser_id, :status, :message, :response, NOW()) )rrr#r$rrr r )rrr#r$rrrrrsave_chat_messageRsr%user_idc s^zd}t|d|iIdH}|durWdS|dWSty.}ztdt|dd}~ww)Nza SELECT lineuser_key FROM lineusers WHERE lineuser_key = :userId userId lineuser_keyrr )r fetch_onerrr)r& user_queryresultrrrr get_user_infobs r-r)returnc sdzd}t|d|iIdH}|durtddd|dWSty1}ztdt|dd}~ww)NzDSELECT lineuser_id FROM lineusers WHERE lineuser_key = :lineuser_keyr)izLineuser not foundr rr)rr*rrr)r)rr,rrrrfetch_lineuser_idqs  r/ user_namec sNzd}t|||dIdHWdSty&}ztdt|dd}~ww)Nzd INSERT INTO lineusers (lineuser_key, lineuser_name) VALUES (:userId, :displayName) )r' displayNamerr r )r&r0 insert_queryrrrrinsert_user_info|sr3N)dbrfastapir databasesrintrrrrr r"rr%r-r/r3rrrrs