o
    àý°ju  ã                   @   s   d dl m Z  G dd„ dƒZdS )é    )Úarrayc                   @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ ZdS )Ú
KeyManagerc                 C   s
   i | _ d S )N)Úkeys)Úself© r   ú‹/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/impacket/Dot11KeyManager.pyÚ__init__   s   
zKeyManager.__init__c                 C   s    t |tttfƒstdƒ‚t|ƒS )Nz-BSSID datatype must be a tuple, list or array)Ú
isinstanceÚlistÚtupler   Ú	Exception©r   Úbssidr   r   r   Ú__get_bssid_hasheable_type   s   z%KeyManager.__get_bssid_hasheable_typec                 C   s&   |   |¡}|| jvr|| j|< dS dS )NTF©Ú%_KeyManager__get_bssid_hasheable_typer   ©r   r   Úkeyr   r   r   Úadd_key   s
   


zKeyManager.add_keyc                 C   s   |   |¡}|| j|< dS )NTr   r   r   r   r   Úreplace_key$   s   

zKeyManager.replace_keyc                 C   s"   |   |¡}|| jv r| j| S dS )NFr   r   r   r   r   Úget_key*   s   


zKeyManager.get_keyc                 C   s6   |   |¡}t|tƒstdƒ‚|| jv r| j|= dS dS )NzBSSID datatype must be a listTF)r   r	   r
   r   r   r   r   r   r   Ú
delete_key1   s   


zKeyManager.delete_keyN)	Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r   r   r   r      s    r   N)r   r   r   r   r   r   Ú<module>   s   