o h@sddlmZmZddlmZddlmZddlmZddZ deefdd Z d ee fd d Z d e d ee fddZ d!dedee fddZde deefddZde de de de fddZde fddZd S)")ListDict) HTTPException)logger)database_yokohamac szzd}d}tj|dIdHtj|dIdHtdWdSty<}ztd|tdd|d d}~ww) u // book_rankingsとkeywordsテーブルの全データを削除（初期化）します。 // TRUNCATE TABLEを使い、データを高速に削除しIDもリセットします。 zTRUNCATE TABLE book_rankings;zTRUNCATE TABLE keywords;)queryNz8Tables 'book_rankings' and 'keywords' have been cleared.zError clearing tables:  DB Error:  status_codedetailrexecuterinfo Exceptionerrorr)query_rankingsquery_keywordser$/home/air/yokohama/back/db_module.pyclear_ranking_datasrrankingsc s|sdSdd|D}zd}tj||dIdHtdt|dWdStyC}ztd|td d |d d}~ww) uo // スクレイピングで取得した複数の書籍の基本情報をMySQL DBに保存します。 NcSs"g|] }|d|dddqS)titlesourcerakuten)rr)get).0rrrr sz&save_book_rankings..zk INSERT IGNORE INTO book_rankings (title, source) VALUES (:title, :source) rvalueszSaved or ignored z book rankings in MySQL.z4Error during bulk insert of book rankings to MySQL: rr r )r execute_manyrrlenrrr)rvalues_to_insertrrrrrsave_book_rankingssr%keywordsc sr|sdSzdd|D}d}tj||dIdHWdSty8}ztd|tdd|d d}~ww) u // [修正] キーワードリストを`keywords`テーブルに保存します。 // `link_keywords_to_book`からこの機能だけを分離しました。 NcSsg|]}d|iqSkeywordr)rkwrrrr5z&save_keywords_list..z7INSERT IGNORE INTO keywords (keyword) VALUES (:keyword)r z%Error saving keywords list to MySQL: rr r )rr"rrrr)r&r!rrrrrsave_keywords_list,sr+rc sz|sdSd|}zd}tj|||ddIdHWdSty<}ztd|d|tdd |d d}~ww) u // [新規] 書籍のタイトルを基準に、`book_rankings`テーブルの`keyword`カラムを更新します。 // これが本来この関数名が担うべき役割です。 N,zV UPDATE book_rankings SET keyword = :keyword WHERE title = :title )r(rr!zError linking keywords to 'z ' in MySQL: rr r )joinrrrrrr)rr& keywords_strrrrrrlink_keywords_to_book=s  r0limitreturnc shzd}tj|d|idIdH}dd|DWSty3}ztd|gWYd}~Sd}~ww)uh // `keywords`テーブルから重複なくランダムにキーワードを取得します。 z SELECT keyword FROM keywords WHERE keyword IS NOT NULL AND keyword != '' GROUP BY keyword ORDER BY RAND() LIMIT :limit r2r-NcSsg|]}|dqSr'r)rrowrrrr\r*z'get_random_keywords..z Error fetching random keywords: r fetch_allrrr)r2rresultsrrrrget_random_keywordsOsr8 session_idc s`zd}tj|d|idIdHWSty/}ztd|d|gWYd}~Sd}~ww)uR // 指定されたsession_idの会話履歴をDBから取得します。 z SELECT user_message, assistant_response, status FROM chat_history WHERE session_id = :session_id AND status != '99' ORDER BY sort_order ASC r9r-Nz+Error fetching chat history for session_id : r5r9rrrrrget_chat_historyfsr< chat_tokenstatusmessageresponsec sz/d}tj|d|idIdH}|dpdd}d}||t|||d }tj||dIdHWdStyP} ztd |d | td d | dd} ~ ww)u= // 1回の対話ターンをDBに保存します。 zTSELECT MAX(sort_order) as max_order FROM chat_history WHERE session_id = :session_idr9r-N max_orderrz INSERT INTO chat_history (session_id, sort_order, status, user_message, assistant_response) VALUES (:session_id, :sort_order, :status, :user_message, :assistant_response) )r9 sort_orderr> user_messageassistant_responsez&Error saving chat turn for session_id r:rr r )r fetch_onestrrrrrr) r=r>r?r@sort_order_querylast_order_result next_order insert_queryr!rrrrsave_chat_messagevs$rLc sxzd}tj|d|idIdHtd|dWdSty;}ztd|d|td d |d d}~ww) uS // 指定されたsession_idの会話履歴をすべて削除します。 z7DELETE FROM chat_history WHERE session_id = :session_idr9r-NzChat history for session z has been cleared.z(Error deleting chat history for session r:rr r r r;rrrdelete_chat_historysrMN)r1)typingrrfastapirconfigrdbrrr%rGr+r0intr8r<rLrMrrrrs