
    Ui                         d dl mZmZ d dlZd dlmZmZ d dlm	Z	 d Z
d Zdedefd	Zded
efdZdedefdZdedefdZdS )    )ListDictN)get_db_connectionlogger)A_DB_CONFIGc                 r   	 t          j        j        di t          }|                                }d}|                    || ||f           |                                 |                                 |                                 d S # t          $ r"}t          j
        d|            Y d }~d S d }~ww xY w)NzJINSERT INTO chat_history (session_id, sender, message) VALUES (%s, %s, %s)zChat DB Save Error:  )mysql	connectorconnectr   cursorexecutecommitclose	Exceptionr   error)
session_idsendermessageconnr   sqles          !/app/app/models/general_module.pysave_chat_logr      s    	1&5555ZsZ9:::

 1 1 1/A//0000000001s   BB
 

B6B11B6c                 t   	 t          j        j        di t          }|                    d          }d}|                    || f           |                                }|                                 |                                 |S # t          $ r#}t          j
        d|            g cY d }~S d }~ww xY 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   rowsr   s         r   get_history_from_dbr       s    &5555--^sZM***  

   /A//000						s   BB
 

B7B2,B72B7r   returnc                    d }	 t          j        j        di t          }|                    d          }d}|                    || f           |                                }|/	 |r*|                                 |                                 dS dS |                    dd          pd	 |r)|                                 |                                 S S # t          $ rI}t          d|            Y d }~|r*|                                 |                                 dS dS d }~ww xY w# |r)|                                 |                                 w w xY w)NTr   zr
            SELECT a_part_status
            FROM user_profile_summary
            WHERE session_id = %s
        r   a_part_statuszGet A Part Status Error: r	   )r
   r   r   r   r   r   fetchoner   getr   printr   r   r   r   rowr   s         r   get_a_part_statusr)   #   s   D&5555--
 	sZM***oo;  	LLNNNJJLLLLL	 	 ww**/a/
  	LLNNNJJLLLL		    -!--...qqq 	LLNNNJJLLLLL	 		  	LLNNNJJLLLL	s0   A C C 
D*!D%3D- %D**D- --Estatusc                    d }	 t          j        j        di t          }|                                }d}|                    || |f           |                                 n)# t          $ r}t          d|            Y d }~nd }~ww xY w|r*|	                                 |	                                 d S d S # |r)|	                                 |	                                 w w xY w)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()
        zUpdate A Part Status Error: r	   )
r
   r   r   r   r   r   r   r   r&   r   )r   r*   r   r   r   r   s         r   update_a_part_statusr,   ;   s   D&5555 	sZ0111 2 2 20Q00111111112  	LLNNNJJLLLLL	 	4 	LLNNNJJLLLL	s0   AA" !B9 "
B,B>B9 BB9 9-C&c                    d}	 t          j        j        di t          }|                    d          }d}|                    || f           |                                }|rH|                    d          r3|d         |r)|                                 |                                 S S 	 |r*|                                 |                                 dS dS # t          $ rN}t          j        d|            Y d}~|r*|                                 |                                 dS dS d}~ww xY w# |r)|                                 |                                 w w xY w)u   유저 이름 조회NTr   zm
            SELECT username
            FROM user_profile_summary
            WHERE session_id = %s
        usernamezGet Username Error: r	   )r
   r   r   r   r   r   r$   r%   r   r   r   r   r'   s         r   get_usernamer/   P   s   D&5555--
 	sZM***oo 	#377:&& 	#z?  	LLNNNJJLLLL	 
  	LLNNNJJLLLLL	 		    /A//000ttt 	LLNNNJJLLLLL	 		  	LLNNNJJLLLL	s*   A<C 
D3%D.<D6 .D33D6 6-E#r.   c                    d}	 t          j        j        di t          }|                                }d}|                    || |f           |                                 n.# t          $ r!}t          j	        d|            Y d}~nd}~ww xY w|r*|
                                 |
                                 dS dS # |r)|
                                 |
                                 w w xY w)u   유저 이름 저장Nz
            INSERT INTO user_profile_summary (session_id, username)
            VALUES (%s, %s)
            ON DUPLICATE KEY UPDATE
                username = VALUES(username),
                updated_at = NOW()
        zSave Username Error: r	   )r
   r   r   r   r   r   r   r   r   r   r   )r   r.   r   r   r   r   s         r   save_usernamer1   h   s   D&5555 	sZ2333 2 2 20Q00111111112  	LLNNNJJLLLLL	 	4 	LLNNNJJLLLL	s0   AA" !B> "
B,BB> BB> >-C+)typingr   r   mysql.connectorr
   app.corer   r   app.core.configr   r   r    strintr)   r,   r/   r1   r	       r   <module>r9      s               . . . . . . . . ' ' ' ' ' '
1 
1 
1  # #    0S #    *S S    0c S      r8   