XhdZddlZddlZddlmZ ddlmZmZmZm Z m Z m Z m Z ddl mZddlmZmZej(ded efd Zd Zeefd ZddZeefdZddZy#e$rddlmZmZmZmZm Z m Z m Z m Z YUwxYw)z*serialization utilities for apply messagesN)chain) CannedObjectcan can_sequenceistypesequence_typesuncanuncan_sequence)PICKLE_PROTOCOL)r rrrrrr r ) MAX_BYTES MAX_ITEMSzHipykernel.serialize is deprecated. It has moved to ipyparallel.serialize) stacklevelc6g}t|tr|jrzt|jD]b\}}t ||kDr!d|j|<|j |5t|t sF|j|j|<d|S)z/extract buffers larger than a certain thresholdN) isinstancerbuffers enumeratelenappend memoryviewtobytes)obj thresholdribufs G/var/www/html/myenv/lib/python3.12/site-packages/ipykernel/serialize.py_extract_buffersr0s~G#|$ , /FAs3x)#!% As#C,!$ A / Nct|trL|jr?t|jD]&\}}| |j d|j|<(yyy)zrestore buffers extracted byNr)rrrrpop)rrrrs r_restore_buffersr!@sP#|$ , 0FAs{!(Q A 0*5$rcg}t|tr"s3x.'@C  BA NN+A/?@ A B T s3x.8 ACF A NN+A/?@ ADG  3x'.>?@ NN1fll49: Nrct|}|jd}tj|}t |t r5t |tkr#|D]}t||t||}||fSt |trHt |tkr6i}t|D]"}||}t||t||||<$||fSt||t||}||fS)a"reconstruct an object serialized by serialize_object from data buffers. Parameters ---------- buffers : list of buffers/bytes g : globals to be used when uncanning Returns ------- (newobj, bufs) : unpacked object, and the list of remaining unused buffers. r) listr r'loadsrrrr r!r r$r%r )rgbufspobjcannedr,newobjr-s rdeserialize_objectr7ns =D 88A;D \\$ F fn%#f+ *A &A Q % &* 4<  #f+ "9 $Aq A Q %a F1I $ 4< &vq! 4<rcttjfd|D}tj }ttjfd|D}t t |t ||}tjt|tg} | jtj|t| j|| j|| S)apack up a function, args, and kwargs to be sent over the wire Each element of args/kwargs will be canned for special treatment, but inspection will not go any deeper than that. Any object whose data is larger than `threshold` will not have their data copied (only numpy arrays and bytes/buffers support zero-copy) Message will be a list of bytes/buffers of the format: [ cf, pinfo, , ] With length at least two + len(args) + len(kwargs) c38K|]}t|ywNr.).0argr)r*s r z%pack_apply_message..sdX[,S2BNSdsc3>K|]}t|ywr:r;)r<keyr)r*kwargss rr>z%pack_apply_message..s% PS VC[*:N K s)nargs narg_bufskw_keys) r0r from_iterabler%keysr$rr'r(rr rr#) fargsrAr)r*arg_bufsrD kwarg_bufsinfomsgs ``` rpack_apply_messagerMs  d_cddHV[[]#G  W^  J c$i3x=' JD <<A 0 1CJJv||D/23JJxJJz Jrct|}t|dk\sJd|jd}tt j ||}|jd}t j |}|d|d||dd}}g} t |dD]"} t||\} }| j| $t| } |rJdi} |dD]}t||\}}|| |<|rJd || | fS) zdunpack f,args,kwargs from buffers packed by pack_apply_message() Returns: original f,args,kwargsrznot enough buffers!rNrCrBz#Shouldn't be any arg bufs left overrDz%Shouldn't be any kwarg bufs left over) r0rr r r'r1ranger7rtuple)r3r2copypfrGpinforKrIrJ args_list_r=rHrAr@kwargs runpack_apply_messagerWs& :D t9>000> !B fll2"A HHQKE << D 3${"34d4 ;L;N6OjHI 4= !*8Q7 X  D>>>< FI.z1=zs BBB> dF?rr:)NT)__doc__r'warnings itertoolsripyparallel.serialize.canningrrrrrr r ipyparallel.serialize.serializer ImportErroripykernel.pickleutiljupyter_client.sessionr r warnDeprecationWarningrr!r.r7rMrWrrrcs0 @8  N%.  0,5Y#L@:CS\"J{     sA$$B?B