File: //opt/imunify360/venv/lib/python3.11/site-packages/im360/utils/__pycache__/__init__.cpython-311.pyc
�
fn#g} � �: � d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
d dl m Z e j
e� � Z G d� de � � Z G d� de � � Zde
e d ed
e fd �Z G d � d
e� � Zdej ej ej ej ffd�Zdd�d�ZdS )� N)�Any�Callable�Type)�LazyLockc � � e Zd ZdS )� RulesLockN��__name__�
__module__� __qualname__� � �I/opt/imunify360/venv/lib/python3.11/site-packages/im360/utils/__init__.pyr r � � � � � � ��Dr r c � � e Zd ZdS )�
ModSecLockNr r
r r r r r r r �error�value�returnc � � �� � �fd�}|S )a.
Decorator that causes coroutine to return *value* if it raises *error*.
For example:
return await return_value_on_error(FileNotFoundError, X)(coro)(*args)
is equivalent to:
try:
return await coro(*args)
except FileNotFoundError:
return X
c �L �� � t j � � � � ��fd�� � }|S )Nc � � �K � �| i |��� d {V ��S # �$ r)}t � d|��� � �cY d }~S d }~ww xY w)Nz Replacing %s from %s with %s)�logger�info)�args�kwargs�e�coror r s ���r �wrapperz9return_value_on_error.<locals>.decorator.<locals>.wrapper) sz �� � � �
� !�T�4� 2�6� 2� 2�2�2�2�2�2�2�2���
�
�
�� � � :�A�t�U�K�K�K� � � � � � � �����
���s �
�A� <�A�A)� functools�wraps)r r r r s ` ��r � decoratorz(return_value_on_error.<locals>.decorator( sC ��� � � �� � � � � � � � �
� � � �r r
)r r r" s `` r �return_value_on_errorr# s* ��� � � � � � � � �r c � � � e Zd Z� fd�Z� xZS )�PasswordChangeErrorc �V �� t � � � d|� d|� �� � d S )Nz)Password change process exited with code z: )�super�__init__)�self�stderr�
returncode� __class__s �r r( z PasswordChangeError.__init__7 s< �� �
����� N�
� N� N�f� N� N�
�
�
�
�
r )r
r r r( �
__classcell__)r, s @r r% r% 6 s8 �� � � � � �
�
�
�
�
�
�
�
�
r r% � c � ��� | t |� � k r"t d| �dt |� � ��� � �t t t |� � � � s
J d� � �d� |� � � d� �fd�t | � � D � � � � �t �fd�|D � � � � r�S �L)z�
Return password with *length* char.
It is guaranteed that password have at least one character from
each given *required_char_groups*
z
Given length=zR is too short. Can't get chars from all required groups len(required_char_groups)=zgot empty char group� Tc �8 �� g | ]}t j �� � ��S r
)�secrets�choice)�.0�_�alphabets �r �
<listcomp>z,generate_strong_password.<locals>.<listcomp>] s# �� �L�L�L��G�N�8� 4� 4�L�L�Lr c 3 �P ��K � | ] �t �fd ��D � � � � V � � dS )c 3 � �K � | ]}|�v V � � d S )Nr
)r4 �c�groups �r � <genexpr>z5generate_strong_password.<locals>.<genexpr>.<genexpr>` s' �� � � �-�-�q��U�
�-�-�-�-�-�-r N)�any)r4 r; �passwords @�r r< z+generate_strong_password.<locals>.<genexpr>_ sR ��� � � �
�
�27�C�-�-�-�-�H�-�-�-� -� -�
�
�
�
�
�
r )�len�
ValueError�all�map�join�range)�length�required_char_groupsr6 r> s @@r �generate_strong_passwordrG = s ��� � ��(�)�)�)�)��
,�V�
,�
,��'�(�(�
,�
,�
�
�
�
� � � �
�
� � � � �
� � � � �w�w�+�,�,�H� � �7�7�L�L�L�L�e�F�m�m�L�L�L�M�M�� �
�
�
�
�;O�
�
�
�
�
� � �O� r )�
passwd_cmdc �h � |�!t j dt j �� � g}t |t
� � sJ � t
j g |�| �d� |� � � gdz � � dd�� � d S # t j
$ r }t |j |j
� � �d }~ww xY w)N�passwd)�path�
� T)�input�capture_output�check)�shutil�which�os�defpath�
isinstance�list�
subprocess�runrC �encode�CalledProcessErrorr% r* r+ )�username� new_passwordrH r s r �change_system_passwordr] e s� � ��� �l�8�"�*�=�=�=�>�
� �j�$� '� '�'�'� '�:��� #�j� #�(� #��*�*�l� 1� 1� 3� 3� 4�q� 8�9�9� ��
�
�
�
�
�
�� � (� :� :� :�!�!�(�A�L�9�9�9�����:���s �AB �B1�B,�,B1)�loggingr rS r2 rQ �stringrW �typingr r r �defence360agent.utilsr � getLoggerr
r r r � Exceptionr# r% �ascii_lowercase�ascii_uppercase�digits� punctuationrG r] r
r r �<module>rh s� �� ���� � � � � � � � � ����
�
�
�
�
�
�
�
� � � � � &� &� &� &� &� &� &� &� &� &� *� *� *� *� *� *� �� �8� $� $�� � � � � �� � � � � � � � �� � � ���i� � �� �� � � � �<
�
�
�
�
�)�
�
�
� �� �� ��
��� �% � % � % � % �P BF� :� :� :� :� :� :� :r