U
    ânldI  ã                   @   sP   d dl Z d dlZd dlZd dlmZ dd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dS )é    N)Úget_storagec                   C   s   d  tjtjtj dd�¡S )NÚ é   )Úk)ÚjoinÚrandomÚchoicesÚstringÚascii_uppercaseÚdigits© r   r   ú5/var/www/html/LibreTranslate/libretranslate/secret.pyÚgenerate_secret   s    r   c                  C   s.   t ƒ } |  d¡}|  d|¡ |  dtƒ ¡ d S )NÚsecret_1Úsecret_0)r   Úget_strÚset_strr   )Úsr   r   r   r   Úrotate_secrets
   s    
r   c                 C   s"   t ƒ }| | d¡kp | | d¡kS ©Nr   r   ©r   r   )Úsecretr   r   r   r   Úsecret_match   s    r   c                   C   s   t ƒ  d¡S )Nr   r   r   r   r   r   Úget_current_secret   s    r   c                 C   sF   | j rB| jrBtƒ }| d¡s*| dtƒ ¡ | d¡sB| dtƒ ¡ d S r   )Úapi_keysÚrequire_api_key_secretr   Úexistsr   r   )Úargsr   r   r   r   Úsetup   s    

r   )
Úatexitr   r	   Úlibretranslate.storager   r   r   r   r   r   r   r   r   r   Ú<module>   s   