�
���g[ � �h � d dl mZ ddlmZ ee� � Z G d� d� � Z G d� d� � ZdS ) � )�
import_module� )�
get_loggerc � � e Zd ZdZdd�ZdS )�_PatchedModuleObjzMSet all the modules components as attributes of the _PatchedModuleObj object.Nc �� � |pg }|�B|j D ]:}||v s|� d� � s t | |t ||� � � � �;t |t
� � r|j n|| _ d S )N�__)�__dict__�
startswith�setattr�getattr�
isinstancer �_original_module)�self�module�attrs�keys �g/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/datasets/utils/patching.py�__init__z_PatchedModuleObj.__init__ s � �� ��� � � ��
=�
=���%�<�<�s�~�~�d�';�';�<��D�#�w�v�s�';�';�<�<�<��;E�f�N_�;`�;`� l�� 7� 7�fl�� � � � �N)�__name__�
__module__� __qualname__�__doc__r � r r r r s4 � � � � � �W�W�m� m� m� m� m� mr r c �< � e Zd ZdZg Zd defd�Zd� Zd� Zd� Z d� Z
dS )
�patch_submoduleaF
Patch a submodule attribute of an object, by keeping all other submodules intact at all levels.
Example::
>>> import importlib
>>> from datasets.load import dataset_module_factory
>>> from datasets.streaming import patch_submodule, xjoin
>>>
>>> dataset_module = dataset_module_factory("snli")
>>> snli_module = importlib.import_module(dataset_module.module_path)
>>> patcher = patch_submodule(snli_module, "os.path.join", xjoin)
>>> patcher.start()
>>> assert snli_module.os.path.join is xjoin
N�targetc � � || _ || _ || _ |� d� � d | _ i | _ |pg | _ d S )N�.r )�objr �new�splitr �originalr )r r" r r# r s r r zpatch_submodule.__init__( sE � ���� �� �����<�<��$�$�Q�'�����
��[�b��
�
�
r c �J � | j � d� � �^ }}t t |� � � � D �]H} t d� |d |dz � � � � � }n# t $ r Y �>w xY w| j � � � D ]�}t | j |� � }||u s t |t � � r�|j |u r�|| j
|< t | j |t || j �� � � � t | j |� � }||dz d � D ]F}t ||t t ||d � � | j �� � � � t ||� � }�Gt ||| j � � ���J|r� t t d� |� � � � |� � } n# t" t f$ r Y d S w xY w| j � � � D ]Q}t | j |� � | u r8t | j |� � | j
|< t | j || j � � �Rd S |t% � � d v r?t% � � d | | j
|< t | j || j � � d S t' d|� d�� � �)Nr! r )r � __builtins__zTried to patch attribute z instead of a submodule.)r r$ �range�lenr �join�ModuleNotFoundErrorr" �__dir__r
r r r r% r r r# �AttributeError�globals� RuntimeError)
r �
submodules� target_attr�i� submodule�attr�obj_attr�patchedr �
attr_values
r � __enter__zpatch_submodule.__enter__0 s� � �#'�;�#4�#4�S�#9�#9� ��[� �s�:���'�'� <� <�A�
� )�#�(�(�:�g��A��g�3F�*G�*G� H� H� � ��&�
�
�
���
���� ��(�(�*�*�
<�
<��"�4�8�T�2�2�� �y�(�(� �x�):�;�;� )�@H�@Y�]f�@f�@f�*2�D�M�$�'��D�H�d�,=�h�d�j�,Y�,Y�,Y�Z�Z�Z� %�d�h�� 5� 5�G� )�!�a�%�'�'� 2� 8� 8�� ���.?���QT�VZ�@[�@[�cg�cm�.n�.n�.n�o�o�o�")�'�3�"7�"7����G�[�$�(�;�;�;��!
<�. � b�
� $�]�3�8�8�J�3G�3G�%H�%H�+� V� V�
�
��"�$7�8�
�
�
����
���� ��(�(�*�*�
6�
6�� �4�8�T�*�*�j�8�8�*1�$�(�D�*A�*A�D�M�$�'��D�H�d�D�H�5�5�5��
6�
6� �G�I�I�n� 5�
5�
5�)0���>�)B�;�)O�D�M�+� &� �D�H�k�4�8� 4� 4� 4� 4� 4� � `�;� `� `� `�a�a� as# �-A,�,
A9�8A9�0F8 �8G
� G
c � � t | j � � D ]0}t | j || j � |� � � � �1d S r )�listr% r r" �pop)r �exc_infor4 s r �__exit__zpatch_submodule.__exit__f sL � ��� �'�'� =� =�D� �D�H�d�D�M�$5�$5�d�$;�$;� <� <� <� <� =� =r c �b � | � � � | j � | � � dS )zActivate a patch.N)r8 �_active_patches�append�r s r �startzpatch_submodule.startj s/ � � ������ � �#�#�D�)�)�)�)�)r c � � | j � | � � n# t $ r Y dS w xY w| � � � S )zStop an active patch.N)r? �remove�
ValueErrorr= rA s r �stopzpatch_submodule.stopo sS � � � � � '� '�� -� -� -� -��� � � ��4�4� ���� �}�}��� s � �
+�+r ) r r r r r? �strr r8 r= rB rF r r r r r s� � � � � � �� � �O�!� !�C� !� !� !� !�4b� 4b� 4b�l=� =� =�*� *� *�
� � � � r r N)� importlibr �loggingr r �loggerr r r r r �<module>rK s� �� #� #� #� #� #� #� � � � � � �
��H� � �� m� m� m� m� m� m� m� m�b � b � b � b � b � b � b � b � b � b r