pickleutil.pyc
1 o 2 0��ce1 � @ s� U d Z ddlZddlZejdedd� ddlZddladdlZddlm Z ddl 3 m Z ddl m Z dd lmZ eZeZtjZd:d 4 d �Ze� Zd d � Zdd� Zdd� ZG dd� d�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G d d � d e�Z G d d � d e�Z!G d d!� d!e!�Z"G d"d#� d#e!�Z#d:d$d%�Z$d&d'� Z%d(d)� Z&d*d+� Z'd,d-� Z(e)e*e+fZ,d.d/� Z-d:d0d1�Z.d:d2d3�Z/d:d4d5�Z0d6e e e e1e!ee#ee ee'iZ2eeur�e"e2e< ed7d8� e3e/iZ4ej5eej6f e7d9< e2�� Z8e4�� Z9dS );z>Pickle related utilities. Perhaps this should be called 'can'.� Nz@ipykernel.pickleutil is deprecated. It has moved to ipyparallel.� )� 5 stacklevel)� FunctionType)�codeutil)� 6 get_logger)� import_itemc s � fdd�}t |jd �S )zUthe type of a closure cell doesn't seem to be importable, 7 so just create one 8 c � � S �N� r 9 ��ar 10 ��C:\Users\Jacks.GUTTSPC\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\ipykernel\pickleutil.py�inner# � z _get_cell_type.<locals>.innerr )�type� __closure__)r r r 11 r r �_get_cell_type s r c C s2 t | t�rtd�}t| j|j| j| j�} d| _| S )z�decorator for making functions appear as interactively defined. 12 This results in the function being linked to the user_ns as globals() 13 instead of the module globals(). 14 �__main__)� 15 isinstancer � 16 __import__�__code__�__dict__�__name__� __defaults__� 17 __module__)�fZmainmodr 18 r 19 r � interactive0 s 20 �r c C �F ddl } | azddlm} W n ty Y nw | |_t�td� dS )zquse dill to expand serialization support 21 22 adds support for object methods and closures to serialization. 23 r N�� serialize)�dill�pickle� ipykernelr � ImportError�can_map�popr )r r r 24 r 25 r �use_dillE s �r& c C r )zxuse cloudpickle to expand serialization support 26 27 adds support for object methods and closures to serialization. 28 r Nr )� cloudpickler! r" r r# r$ r% r )r' r r 29 r 30 r �use_cloudpickle^ s �r( c @ s e Zd Zddd�Zddd�ZdS )� CannedObjectNc C sL |pg | _ t�|�| _t|�| _|D ]}t| j|tt||��� qg | _dS )a� can an object for safe pickling 31 32 Parameters 33 ---------- 34 obj 35 The object to be canned 36 keys : list (optional) 37 list of attribute names that will be explicitly canned / uncanned 38 hook : callable (optional) 39 An optional extra callable, 40 which can do additional processing of the uncanned object. 41 42 Notes 43 ----- 44 large data may be offloaded into the buffers list, 45 used for zero-copy transfers. 46 N)�keys�copy�obj�can�hook�setattr�getattr�buffers)�selfr, r* r. �keyr 47 r 48 r �__init__y s 49 50 51 zCannedObject.__init__c C s\ |d u ri }| j }| jD ]}t||tt||�|�� q | jr+t| j|�| _| �||� | j S r )r, r* r/ �uncanr0 r. )r2 �gr, r3 r 52 r 53 r � 54 get_object� s 55 zCannedObject.get_object)NNr �r r � __qualname__r4 r7 r 56 r 57 r 58 r r) x s 59 r) c @ s* e Zd ZdZdd� Zdd� Zd dd�ZdS ) 60 � Referencez/object for wrapping a remote reference by name.c C s& t |t�s td| ��|| _g | _d S )Nzillegal name: %r)r �str� TypeError�namer1 )r2 r= r 61 r 62 r r4 � s 63 64 zReference.__init__c C s 65 d| j S )Nz<Reference: %r>)r= )r2 r 66 r 67 r �__repr__� s 68 zReference.__repr__Nc C s |d u ri }t | j|�S r )�evalr= )r2 r6 r 69 r 70 r r7 � s zReference.get_objectr )r r r9 �__doc__r4 r> r7 r 71 r 72 r 73 r r: � s 74 r: c @ s" e Zd ZdZdd� Zddd�ZdS )� 75 CannedCellzCan a closure cellc C s t |j�| _d S r )r- � cell_contents)r2 �cellr 76 r 77 r r4 � s zCannedCell.__init__Nc s" t | j|�� � fdd�}|jd S )Nc r r r 78 r 79 �rB r 80 r r � r z$CannedCell.get_object.<locals>.innerr )r5 rB r )r2 r6 r r 81 rD r r7 � s 82 zCannedCell.get_objectr )r r r9 r@ r4 r7 r 83 r 84 r 85 r rA � s rA c @ �& e Zd Zdd� Zdd� Zddd�ZdS ) �CannedFunctionc C s~ | � |� |j| _| |jrdd� |jD �| _nd | _| |j}|r-tdd� |D ��| _nd | _|jp4d| _ |j 86 | _ 87 g | _ d S )Nc S � g | ]}t |��qS r 88 �r- )�.0�fdr 89 r 90 r � 91 <listcomp>� � z+CannedFunction.__init__.<locals>.<listcomp>c s s � | ]}t |�V qd S r rH �rI rC r 92 r 93 r � <genexpr>� s � z*CannedFunction.__init__.<locals>.<genexpr>r ) � _check_typer �coder �defaultsr �tuple�closurer �moduler r1 )r2 r rS r 94 r 95 r r4 � s 96 97 zCannedFunction.__init__c C � t |t�s J d��d S )NzNot a function type)r r �r2 r, r 98 r 99 r rO � � zCannedFunction._check_typeNc s� | j �d�st| j � tj| j j� � d u ri � | jr(t� fdd�| jD ��}nd }| jr:t� fdd�| jD ��}nd }t | j 100 � | j ||�}|S )N�__c 3 � � | ]}t |� �V qd S r �r5 )rI Zcfd�r6 r 101 r rN � � � z,CannedFunction.get_object.<locals>.<genexpr>c 3 rY r rZ rM r[ r 102 r rN � r\ ) rT � 103 startswithr �sys�modulesr rQ rR rS r rP r )r2 r6 rQ rS ZnewFuncr 104 r[ r r7 � s 105 zCannedFunction.get_objectr �r r r9 r4 rO r7 r 106 r 107 r 108 r rF � s rF c @ rE ) � CannedClassc C s� | � |� |j| _t|t� | _i | _|j�� D ]\}}|dvr't |�| j|< q| jr.g }n|� 109 � }dd� |dd � D �| _ g | _ d S )N)� __weakref__r c S rG r 110 rH )rI �cr 111 r 112 r rK � rL z(CannedClass.__init__.<locals>.<listcomp>� ) rO r r= r r Z old_style� _canned_dictr �itemsr- �mro�parentsr1 )r2 �cls�k�vrg r 113 r 114 r r4 � s 115 � 116 zCannedClass.__init__c C rU )NzNot a class type)r � 117 class_typerV r 118 r 119 r rO � rW zCannedClass._check_typeNc s0 t � fdd�| jD ��}t| j|t| j� d��S )Nc 3 rY r rZ )rI �pr[ r 120 r rN r\ z)CannedClass.get_object.<locals>.<genexpr>r[ )rR rh r r= � 121 uncan_dictre )r2 r6 rh r 122 r[ r r7 s zCannedClass.get_objectr r` r 123 r 124 r 125 r ra � s ra c @ s e Zd Zdd� Zddd�ZdS )� CannedArrayc C s� ddl m} |j| _|jjr|jjn|jj| _d| _t|j�dkr%d| _n |jdkr.d| _n|jjrBt dd� |jj� 126 � D ��rBd| _| jrOt � |t �g| _d S ||d d�}t|�g| _d S ) Nr )�ascontiguousarrayFT�Oc s s � | ] \}}|d kV qdS )rq Nr 127 )rI �dt�szr 128 r 129 r rN s � z'CannedArray.__init__.<locals>.<genexpr>��dtype)�numpyrp �shaperu �fields�descrr; �pickled�sum�any�valuesr! �dumps�PICKLE_PROTOCOLr1 �buffer)r2 r, rp r 130 r 131 r r4 s 132 " zCannedArray.__init__Nc C s< ddl m} | jd }| jrt�|�S ||| jd��| j�S )Nr )� 133 frombufferrt ) rv r� r1 rz r! �loadsru �reshaperw )r2 r6 r� �datar 134 r 135 r r7 s 136 137 138 zCannedArray.get_objectr r8 r 139 r 140 r 141 r ro s ro c @ sB e Zd Zedejeeejf defdd��Z dd� Z 142 d 143 dd �Z dS ) � CannedBytes�buf�returnc C s( t | t�r | �� S t | t�st| �S | S )z+Cast a buffer or memoryview object to bytes)r � 144 memoryview�tobytes�bytes)r� r 145 r 146 r �wrap) s 147 148 149 zCannedBytes.wrapc C s |g| _ d S r )r1 rV r 150 r 151 r r4 2 s zCannedBytes.__init__Nc C s | j d }| �|�S )Nr )r1 r� )r2 r6 r� r 152 r 153 r r7 5 s 154 155 zCannedBytes.get_objectr ) r r r9 � staticmethod�typing�Unionr� r� � SupportsBytesr� r4 r7 r 156 r 157 r 158 r r� ( s 159 "r� c @ � e Zd ZeZdS )� CannedBufferN)r r r9 r� r� r 160 r 161 r 162 r r� : � r� c @ r� )�CannedMemoryViewN)r r r9 r� r� r 163 r 164 r 165 r r� > r� r� c C s� t � }|�d� t| �� �D ]3\}}t|t�rAzt|�}W n ty9 |r2||vr2|jd|dd� | � |� Y qw | � |�| |< qdS )z(import any string-keys in a type mappingzImporting canning mapz canning class not importable: %rT)�exc_infoN) 166 r �debug�listrf r r; r � Exception�errorr% )�mapping�original�logr3 �_ri r 167 r 168 r �_import_mappingG s 169 170 ���r� c C s6 t |t�r|D ] }t| �|u r dS qdS t| �|u S )zNlike isinstance(obj, check), but strict 171 172 This won't catch subclasses. 173 TF)r rR r )r, �checkri r 174 r 175 r �istypeX s 176 � r� c C sX d}t �� D ]\}}t|t�rd} n t| |�r || � S q|r*tt t� t| �S | S )z prepare an object for picklingFT)r$ rf r r; r� r� �_original_can_mapr- )r, � import_neededri Zcannerr 177 r 178 r r- f s 179 180 � 181 r- c C s t | t�r| jdkrt| �S | S )Nr )r rl r ra )r, r 182 r 183 r � can_class{ s r� c C s4 t | t�ri }| �� D ] 184 \}}t|�||< q |S | S )zcan the *values* of a dict)r� �dictrf r- )r, �newobjrj rk r 185 r 186 r �can_dict� s 187 r� c C s( t | t�rt| �}|dd� | D ��S | S )z can the elements of a sequencec S rG r 188 rH �rI �ir 189 r 190 r rK � rL z can_sequence.<locals>.<listcomp>�r� �sequence_typesr )r, �tr 191 r 192 r � can_sequence� s 193 r� c C s\ d}t �� D ]\}}t|t�rd} n t| |�r || |� S q|r,tt t� t| |�S | S )zinvert canningFT)� uncan_maprf r r; r� �_original_uncan_mapr5 )r, r6 r� ri Zuncannerr 194 r 195 r r5 � s 196 197 � 198 199 r5 c C s6 t | t�ri }| �� D ] \}}t||�||< q |S | S r )r� r� rf r5 )r, r6 r� rj rk r 200 r 201 r rn � s 202 rn c s, t | t�rt| �}|� fdd�| D ��S | S )Nc s g | ]}t |� ��qS r 203 rZ r� r[ r 204 r rK � s z"uncan_sequence.<locals>.<listcomp>r� )r, r6 r� r 205 r[ r �uncan_sequence� s 206 r� z numpy.ndarrayc C s 207 | � |�S r )r7 )r, r6 r 208 r 209 r �<lambda>� s 210 r� r� r ):r@ r� �warnings�warn�DeprecationWarningr+ r! r^ �typesr �ipyparallel.serializer � traitlets.logr � traitlets.utils.importstringr r� r� r rl �DEFAULT_PROTOCOLr r � cell_typer r& r( r) r: rA rF ra ro r� r� r� r� r� r- r� r� r� rR �setr� r� r5 rn r� r� r$ r� r� �Dict�Any�__annotations__r� r� r 211 r 212 r 213 r �<module> sp � 214 (,! 215 216 217 218 219 220 ��