o
    ·ôjj1  ã                   @   sÀ   d dl Z d dlmZ d dlmZmZ ddddœZdd	„ Zd
dd dœfddddœfddddœfddddœfddddœfddddœfddddœfddddœfddddœfddddœfg
Zdd„ Z	dS ) é    N)Úutils)Úget_lottery_title_idsÚget_title_defé   é   é   )ÚcommonÚrareÚ	legendaryc                 C   s¤   g }t ƒ D ]}t|ƒpi }| d¡|kr| |¡ q|s!d|dœS t |¡}| j ||¡rEt |d¡}|dkr>t	 
| ||¡ d|||dœS | j ||¡ d||dœS )	NÚrarityÚ
title_none©Útyper   r   Útitle_duplicate)r   Úvaluer   ÚrebateÚ	title_new)r   r   r   )r   r   ÚgetÚappendÚrandomÚchoiceÚtitlesÚhasÚ
DUP_REBATEr   Úadd_user_pointÚunlock)Ú	dbmanagerÚuser_idr   Ú
candidatesÚtidÚdataÚtitle_idr   © r"   ú4/home/dore/onebot/Bot_Ero/plugins/lottery/rewards.pyÚdraw_title_by_rarity	   s    

€

r$   g      ?@Úpoints©r   r   g      <@g      $@g      @g      @é   gš™™™™™é?é   gš™™™™™É?é
   g      (@Ú
title_rollr   r   g      @r	   g      @r
   c                 C   s`   t   ¡ d }d}tD ] \}}||7 }||k r*|d dkr |  S t| ||d ƒ  S q
dddœS )Néd   g        r   r%   r   r   r&   )r   ÚREWARD_TABLEr$   )r   r   ÚrollÚ	thresholdÚprobÚrewardr"   r"   r#   Údraw_reward-   s   ý
r1   )
r   Úcorer   Úplugins.titler   r   r   r$   r,   r1   r"   r"   r"   r#   Ú<module>   s"    ö