o lhz(@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z e r,ej Z nej Z Gddde ZGddde ZGd d d e ZGd d d eZdddZdS)zb Fast cryptographic hash of Python objects, with a special case for fast hashing of numpy arrays. N)_bytes_or_unicode PY3_OR_LATERc@eZdZdZddZdS)_ConsistentSetza Class used to ensure the hash of Sets is preserved whatever the order of its items. c Cs@zt||_WdSttjfytdd|D|_YdSw)Ncss|]}t|VqdSNhash).0er @/var/www/html/env/lib/python3.10/site-packages/joblib/hashing.py *sz*_ConsistentSet.__init__..)sorted _sequence TypeErrordecimalInvalidOperation)self set_sequencer r r __init__s z_ConsistentSet.__init__N__name__ __module__ __qualname____doc__rr r r r rs rc@r)_MyHashz7 Class used to hash objects that won't normally pickle cGs ||_dSr)args)rrr r r r0s z_MyHash.__init__Nrr r r r r-s rc@seZdZdZdddZdddZdd Zd d Zd ej fd dZ e j Z e e ee<e e ee<e e ee <e e eej<ddZddZee ee<d S)HasherzW A subclass of pickler, to do cryptographic hashing, rather than pickling. md5cCs8t|_tr dnd}tj||j|dt||_dS)N)protocol) ioBytesIOstreamrPicklerrhashlibnew_hash)r hash_namer"r r r r9s  zHasher.__init__Tc Cslz||Wntjy!}z |jd||ff7_d}~ww|j}|j||r4|jSdS)Nz"PicklingError while hashing %r: %r) dumppickle PicklingErrorrr%getvaluer)update hexdigest)robj return_digestr dumpsr r r r Bs   z Hasher.hashcCst|tjtijfr>t|dr|jj}n|j}|j}t|tt kr*t ||j}n|dur4t ||}n |jj }t |||}t ||dS)N__func__) isinstancetypes MethodTypetypepophasattrr4r__self__r,r __class__r&save)rr1 func_nameinstclsr r r r=Ms    z Hasher.savecCst|trdSt||dSr)r5rr&memoize)rr1r r r rA`s zHasher.memoizeNcCst||d}tjdkr|d=z tj||fi|WdStjyYtj||fi|t|dd}|dkrS|}|dur@|j}tj |}t ||sVt |||YdSYdSYdSw)N)namepack)r rCr__main__) dictsys version_infor& save_globalr,r-getattrrmodulesr:setattr)rr1rBrCkwargsmodulemy_namemodr r r rIks$     zHasher.save_globalc CsNz t|tt|WdSty&t|ttdd|DYdSw)Ncss |] \}}t||fVqdSrr)r kvr r r rsz)Hasher._batch_setitems..)r&_batch_setitemsiterrr)ritemsr r r rSs zHasher._batch_setitemscCst|t|dSr)r&r=r)r set_itemsr r r save_setszHasher.save_set)r)T)rrrrrr r=rAstructrCrIr&dispatchcopyr8lenobjectr,r+rSrWsetr r r r r4s       rc@s"eZdZdZd ddZddZdS) NumpyHasherz7 Special case the hasher for when numpy is loaded. rFcCsB||_tj||dddl}||_t|dr|j|_dSt|_dS)a Parameters ---------- hash_name: string The hash algorithm to be used coerce_mmap: boolean Make no difference between np.memmap and np.ndarray objects. r*rN getbuffer) coerce_mmaprrnumpynpr:r` _getbuffer memoryview)rr*rarcr r r rs    zNumpyHasher.__init__cCst||jjrT|jjsT|jdkr|}n|jjr|}n |jj r$|j }n|}|j | ||jj|jrEt||jjrE|jj}n|j}|d|j|j|jff}nt||jjre|j}|d|jff}t||dS)z Subclass the save method, to hash ndarray subclass, rather than pickling them. Off course, this is a total abuse of the Pickler class. r HASHEDN)r5rcndarraydtype hasobjectshapeflattenflags c_contiguous f_contiguousTr)r/rdviewuint8ramemmapr<stridesdescrrr=)rr1obj_c_contiguousklassr r r r=s&    zNumpyHasher.saveNrF)rrrrrr=r r r r r^s  r^rFcCsHd}||vrtd||dtjvrt||d}nt|d}||S)al Quick calculation of a hash to identify uniquely Python objects containing numpy arrays. Parameters ----------- hash_name: 'md5' or 'sha1' Hashing algorithm used. sha1 is supposedly safer, but md5 is faster. coerce_mmap: boolean Make no difference between np.memmap and np.ndarray )rsha1zAValid options for 'hash_name' are {}. Got hash_name={!r} instead.rb)r*rar_) ValueErrorformatrGrKr^rr )r1r*ravalid_hash_nameshasherr r r r s    r rw)rr,r'rGr6rXr#r_compatrr_Picklerr&r\rrrr^r r r r r s" lU