o
    kj                     @   sN   d dl Z d dlmZ d dlmZ g dZdd Zdeded	ed
efddZdS )    N)logger)	DbManager)	)total_games
   i-  )r   2   i.  )r   d   i/  )
total_winsr   i0  )r   r   i1  )civilian_winsr   i2  )r	   r   i3  )spy_winsr   i4  )r
   r   i5  c                  C   s(   t d} | d |   |   d S )Ndata.dba  
        CREATE TABLE IF NOT EXISTS user_game_stats (
            user_id TEXT PRIMARY KEY,
            total_games INTEGER DEFAULT 0,
            total_wins INTEGER DEFAULT 0,
            civilian_wins INTEGER DEFAULT 0,
            spy_wins INTEGER DEFAULT 0
        );
    )sqlite3connectexecutecommitclose)conn r   6/home/dore/onebot/Bot_Ero/plugins/who_is_spy/titles.py_ensure_stats_table   s   

	r   	dbmanageruser_idrolewinnerc              
   C   s  t   t|}||k}|dkr|rdnd}|dkr|rdnd}td}|d||r,dnd|||r3dnd||f |  |d|f }	|  |	sNg S |	d |	d |	d |	d	 d
}
g }tD ]\}}}|
| |kr| j	
||s| j	|| || qa|rddlm} || | |S )Ncivilian   r   spyr   aV  
        INSERT INTO user_game_stats (user_id, total_games, total_wins, civilian_wins, spy_wins)
        VALUES (?, 1, ?, ?, ?)
        ON CONFLICT(user_id) DO UPDATE SET
            total_games = total_games + 1,
            total_wins = total_wins + ?,
            civilian_wins = civilian_wins + ?,
            spy_wins = spy_wins + ?
    z^SELECT total_games, total_wins, civilian_wins, spy_wins FROM user_game_stats WHERE user_id = ?      )r   r   r	   r
   )evaluate_and_unlock_titles)r   strr   r   r   r   fetchoner   GAME_TITLE_THRESHOLDStitleshasunlockappendplugins.titler   )r   r   r   r   uid	is_winnerciv_winspy_winr   rowstatsnewlystat_key	thresholdtidr   r   r   r   grant_game_titles!   s<   
 	

r1   )	r   core.loggerr   core.database_managerr   r!   r   r   r1   r   r   r   r   <module>   s    