.PhGFdZddlZddlmZmZddlmZmZmZddl m Z ddl m Z ddl mZddlmZd Z Gd d eZGd d eZGddeZdZGddZdS)z Module ``ioutils`` implements a number of helper classes and functions which are useful when dealing with input, output, and bytestreams in a variety of ways. N)BytesIOIOBase)ABCMetaabstractmethodabstractproperty)EINVAL) EncodedFile) TemporaryFile) zip_longestiUSceZdZdZeZd+dZd,dZed-dZ edZ ed.d Z ed,d Z ed.d Z d ZedZedZedZedZedZdZdZeeeZedZdZdZdZedZedZedZdZ d,dZ!dZ"dZ#d Z$d!Z%d"Z&e&Z'd#Z(d$Z)d%Z*d&Z+d'Z,d(Z-d)Z.d*Z/dS)/ SpooledIOBaseai A base class shared by the SpooledBytesIO and SpooledStringIO classes. The SpooledTemporaryFile class is missing several attributes and methods present in the StringIO implementation. This brings the api as close to parity as possible so that classes derived from SpooledIOBase can be used as near drop-in replacements to save memory. @KLNc"||_||_dSN) _max_size_dir)selfmax_sizedirs O/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/boltons/ioutils.py__init__zSpooledIOBase.__init__Hs! c:|jrt|dn|dS)z$Raise a ValueError if file is closedNzI/O operation on closed file.)closed ValueError)rmsgs r _checkClosedzSpooledIOBase._checkClosedLs8 ; 6 # =<1466 6 6 6rcdS)z!Read n characters from the bufferNrns rreadzSpooledIOBase.readQrcdS)zWrite into the bufferNr rss rwritezSpooledIOBase.writeUr$rrcdS)z"Seek to a specific point in a fileNr rposmodes rseekzSpooledIOBase.seekYr$rcdS)zReturns the next available lineNr rlengths rreadlinezSpooledIOBase.readline]r$rcdS)z=Returns a list of all lines from the current position forwardNr rsizehints r readlineszSpooledIOBase.readlinesar$rcb||D]}||dS)zv Write lines to the file from an interable. NOTE: writelines() does NOT add line separators. N)rr()rlineslines r writelineszSpooledIOBase.writelinesesD   D JJt      rcdS)z5Roll file-like-object over into a real temporary fileNr rs rrolloverzSpooledIOBase.rolloveror$rcdS)zReturn the current positionNr r;s rtellzSpooledIOBase.tellsr$rcdS)zShould return a flo instanceNr r;s rbufferzSpooledIOBase.bufferwr$rcdS)z>Returns whether the file has been rolled to a real file or notNr r;s r_rolledzSpooledIOBase._rolled{r$rcdS)zReturns the length of the dataNr r;s rlenzSpooledIOBase.lenr$rc|jjSrr@ softspacer;s r_get_softspacezSpooledIOBase._get_softspaces {$$rc||j_dSrrF)rvals r_set_softspacezSpooledIOBase._set_softspaces # rc|jSr)r@r;s r_filezSpooledIOBase._files {rc4|jSr)r@closer;s rrOzSpooledIOBase.closes{  """rc\||jSr)rr@flushr;s rrQzSpooledIOBase.flushs( {  """rc\||jSr)rr@isattyr;s rrSzSpooledIOBase.isattys( {!!###rc|jjSr)r@rr;s rrzSpooledIOBase.closeds {!!rc*|Sr)r>r;s rr+zSpooledIOBase.possyy{{rc*|Sr)getvaluer;s rbufzSpooledIOBase.bufs}}rc\||jSr)r<r@filenor;s rrZzSpooledIOBase.filenos# {!!###rcX|||jS|dkrttd|}|||j||kr||dSdS)z Truncate the contents of the buffer. Custom version of truncate that takes either no arguments (like the real SpooledTemporaryFile) or a single argument that truncates the value to a certain index location. NrzNegative size not allowed)rr@truncateOSErrorrr>r-)rsizer+s rr\zSpooledIOBase.truncates  <;'')) ) !88&"=>> >iikk $  :: IIcNNNNN :rc||}|d|}|||S)z!Return the entire files contents.r)rr>r-r#rr+rJs rrWzSpooledIOBase.getvaluesQ iikk ! iikk # rcdSNTr r;s rseekablezSpooledIOBase.seekabletrcdSrbr r;s rreadablezSpooledIOBase.readablerdrcdSrbr r;s rwritablezSpooledIOBase.writablerdrcR||}|s||j}|jdt j||jkrt|j||SNr)rr1r@r>r-osSEEK_END StopIteration)rr8r+s r__next__zSpooledIOBase.__next__s }} &+""$$C K  Q , , ,dk&&((((##   %%% rc|jSrrDr;s r__len__zSpooledIOBase.__len__s xrc.||Srrr;s r__iter__zSpooledIOBase.__iter__  rc.||Srrsr;s r __enter__zSpooledIOBase.__enter__rurc8|jdSr)rMrO)rargss r__exit__zSpooledIOBase.__exit__s rc,t||jr|}|} |d|dd}t ||D]\}}||krd}n|||||S#t $rP ||n#t $rYnwxYw ||n#t $rYnwxYwwxYwdS)NrTF) isinstance __class__r>r-r Exception)rotherself_pos other_poseq self_line other_lines r__eq__zSpooledIOBase.__eq__sX eT^ , , yy{{H I  !  1 -8u-E-E)Iz J.."/ (### 9%%%    IIh'''' DJJy)))) D usTA6B77 DCD C%"D$C%%D)C?>D? D  D D  Dc.|| Sr)r)rrs r__ne__zSpooledIOBase.__ne__ s;;u%%%%rcdSrbr r;s r__bool__zSpooledIOBase.__bool__rdrcR |dS#t$rYdSwxYw)z;Can fail when called at program exit so suppress traceback.N)rOr~r;s r__del__zSpooledIOBase.__del__s:  JJLLLLL    DD s  &&)rNrrr)0__name__ __module__ __qualname____doc__r __metaclass__rrrr#r(r-r1r5r9r<r>rr@rBrDrHrKpropertyrGrMrOrQrSrr+rXrZr\rWrcrfrhrnnextrqrtrwrzrrrrr rrr r =s@M6666 000^0$$^$111^1...^.LLL^LDD^D**^*+++MMM---%%%$$$88I X######$$$""X"XX$$$,    D:&&&rr ceZdZdZddZdZddZddZdd Zd Z e d Z e d Z e d Z dZdS)SpooledBytesIOa SpooledBytesIO is a spooled file-like-object that only accepts bytes. On Python 2.x this means the 'str' type; on Python 3.x this means the 'bytes' type. Bytes are written in and retrieved exactly as given, but it will raise TypeErrors if something other than bytes are written. Example:: >>> from boltons import ioutils >>> with ioutils.SpooledBytesIO() as f: ... f.write(b"Happy IO") ... _ = f.seek(0) ... isinstance(f.getvalue(), bytes) True rc^||j|Sr)rr@r#r!s rr#zSpooledBytesIO.read*s* {"""rcv|t|ts4tdt |j|t|z|j kr| |j |dS)Nzbytes expected, got {}) rr|bytes TypeErrorformattyperr>rDrr<r@r(r&s rr(zSpooledBytesIO.write.s !U## 4;;Q   99;;Q 4> 1 1 MMOOO !rrc`||j||Sr)rr@r-r*s rr-zSpooledBytesIO.seek9s, {T***rNc||r|j|S|jSr)rr@r1r/s rr1zSpooledBytesIO.readline=sF   *;''// /;'')) )rc6|j|Sr)r@r5r3s rr5zSpooledBytesIO.readlinesDs{$$X...rc6|jst|j}|j}||j|||j||_ dSdS)z$Roll the StringIO over to a TempFilerN) rBr rr@r>r(rWr-rO_bufferrtmpr+s rr<zSpooledBytesIO.rolloverGs| DI...C+""$$C IIdk**,, - - - HHSMMM K     DLLL   rc8t|jt Sr)r|r@rr;s rrBzSpooledBytesIO._rolledQsdk73333rcf |jS#t$rt|_YnwxYw|jSr)rAttributeErrorrr;s rr@zSpooledBytesIO.bufferUsA %<  % % %"99DLLL %|s  ))cP|}|jrA|dtj|j}n4|dtj|}|||S)z Determine the length of the filer)r>rBr-rkfstatrZst_sizerlr`s rrDzSpooledBytesIO.len]s}iikk <  IIaLLL(4;;==))1CC IIa % % %))++C # rc\||jSr)rr@r>r;s rr>zSpooledBytesIO.telljs( {!!!rrrr)rrrrr#r(r-r1r5r<rrBr@rDr>r rrrrs ####   ++++****////44X4X  X """""rrceZdZdZfdZddZdZdZddZdd Z dd Z e d Z e d Z dZdZe dZxZS)SpooledStringIOu9 SpooledStringIO is a spooled file-like-object that only accepts unicode values. On Python 2.x this means the 'unicode' type and on Python 3.x this means the 'str' type. Values are accepted as unicode and then coerced into utf-8 encoded bytes for storage. On retrieval, the values are returned as unicode. Example:: >>> from boltons import ioutils >>> with ioutils.SpooledStringIO() as f: ... f.write(u"— Hey, an emdash!") ... _ = f.seek(0) ... isinstance(f.read(), str) True cHd|_tj|i|dSrj)_tellsuperr)rrykwargsr}s rrzSpooledStringIO.__init__s, $)&)))))rrc||jj||}|t |z|_|Sr)rr@readerr#r>rDr)rr"rets rr#zSpooledStringIO.readsM k %%a++YY[[3s88+  rc"|t|ts4tdt |j|}|jt| dz|j kr| |j | d|t|z|_dS)Nzstr expected, got {}utf-8)rr|strrrrrr>r@rDencoderr<r(r)rr' current_poss rr(zSpooledStringIO.writes !S!! 299Q  iikk ;    AHHW$5$5 6 6 6$. H H MMOOO !((7++,,, 3q66) rc||z} ||krnO|tz|kr|||z n(|t}|tz }|snV|S)z8Traverse from current position to the right n codepoints)READ_CHUNK_SIZEr#)rcurrent_positionr"destrs r_traverse_codepointsz$SpooledStringIO._traverse_codepointss!# 4'' /1D88 $!11222ii00  /  ' * rrcZ||tjkr8|jd|d|||_n|tjkrG|}|||||z|_ne|tj krB|jd|j |z }|d|||_ntd|d|S)z/Traverse from offset to the specified codepointrzInvalid whence (z, should be 0, 1, or 2)) rrkSEEK_SETr@r-rrSEEK_CURr>rlrDr)rr+r, start_pos dest_positions rr-zSpooledStringIO.seeks  2;   K  Q     % %a - - -DJJ R[  I  % %diikk3 7 7 7"SDJJ R[ K  Q    HsNM  % %a 7 7 7&DJJ@4@@@ yy{{rNc||j|d}|t |z|_|S)Nr)rr@r1decoder>rDr)rr0rs rr1zSpooledStringIO.readlinesV k""6**11'::YY[[3s88+  rcd|j|D}|td|Dz|_|S)Nc8g|]}|dS)r)r.0xs r z-SpooledStringIO.readlines..s$JJJQqxx  JJJrc34K|]}t|VdSrrprs r z,SpooledStringIO.readlines..s(&;&;!s1vv&;&;&;&;&;&;r)r@r5r>sumr)rr4rs rr5zSpooledStringIO.readlinessUJJ$+*?*?*I*IJJJYY[[3&;&;s&;&;&;#;#;;  rc |jS#t$r%ttd|_YnwxYw|jS)Nr data_encoding)rrr rr;s rr@zSpooledStringIO.buffersS I<  I I I&wyyHHHDLLL I|s  ,88cBt|jjt Sr)r|r@streamrr;s rrBzSpooledStringIO._rolledsdk0'::::rcT|jstt|jd}|j}||j|||j ||_ dSdS)z"Roll the buffer over to a TempFilerrrN) rBr r rr@r>r(rWr-rOrrs rr<zSpooledStringIO.rollovers| m :::,3555C+""$$C IIdk**,, - - - HHSMMM K     DLLL  rc8||jS)zReturn the codepoint position)rrr;s rr>zSpooledStringIO.tells zrc|j}|jdd} |t}|sn|t |z }0|j||S)z.Determine the number of codepoints in the filer)r@r>r-r#rrD)rr+totalrs rrDzSpooledStringIO.lensk    ))O,,C  SXX E    rrrr)rrrrrr#r(rr-r1r5rr@rBr<r>rD __classcell__)r}s@rrros"***** * * *40  X;;X;      X     rrct|ddrdSt|ddr; t|trdSn#t$rYnwxYwdS)NencodingFTrW)getattrr|rWrr~)fileobjs ris_text_fileobjrsw E**tw E** '**,,c22 t     D  5s'A AAc4eZdZdZdZddZejfdZdS)MultiFileReaderajTakes a list of open files or file-like objects and provides an interface to read from them all contiguously. Like :func:`itertools.chain()`, but for reading files. >>> mfr = MultiFileReader(BytesIO(b'ab'), BytesIO(b'cd'), BytesIO(b'e')) >>> mfr.read(3).decode('ascii') u'abc' >>> mfr.read(3).decode('ascii') u'de' The constructor takes as many fileobjs as you hand it, and will raise a TypeError on non-file-like objects. A ValueError is raised when file-like objects are a mix of bytes- and text-handling objects (for instance, BytesIO and StringIO). ctd|Dstdtd|Drd|_n/td|Drt dd|_||_d|_dS) Nc g|]>}tt|ddott|dd?S)r#Nr-)callablerrfs rrz,MultiFileReader.__init__..s^JJJ;<WQ55667WQ5566JJJrzCMultiFileReader expected file-like objects with .read() and .seek()c,g|]}t|Sr rrs rrz,MultiFileReader.__init__..#s 555q""555rc,g|]}t|Sr rrs rrz,MultiFileReader.__init__..&s 777/!$$777rzEAll arguments to MultiFileReader must handle bytes OR text, not a mixrr)allr_joineranyr _fileobjs_index)rfileobjss rrzMultiFileReader.__init__sJJ@HJJJKK 9899 9 55H555 6 6 DLL 77h777 8 8 9:: :DL! rNc|s)|jd|jDSg}|dkr|jt |jkr||j|j|t |d}||kr|xjdz c_||z}|dkr|jt |jk|j|S)zRead up to the specified *amt*, seamlessly bridging across files. Returns the appropriate type of string (bytes or text) for the input, and returns an empty string when the files are exhausted. c3>K|]}|VdSr)r#rs rrz'MultiFileReader.read..6s*$F$F!QVVXX$F$F$F$F$F$Frrr)rjoinrrrDappendr#)ramtpartsgots rr#zMultiFileReader.read/s  G<$$$F$Ft~$F$F$FFF FAgg$+DN(;(;;; LL 499#>> ? ? ?eBi..CSyy q 3JC Agg$+DN(;(;;; |  '''rc|tjkrtd|dkrtd|jD]}|ddS)zwEnables setting position of the file cursor to a given *offset*. Currently only supports ``offset=0``. z0MultiFileReader.seek() only supports os.SEEK_SETrz;MultiFileReader only supports seeking to start at this timeN)rkrNotImplementedErrorrr-)roffsetwhencers rr-zMultiFileReader.seek@sv R[ %BDD D Q;;%MOO O  A FF1IIII  rr) rrrrrr#rkrr-r rrrr s\ "(((("#%+      rr)rrkiorrabcrrrerrnorcodecsr tempfiler itertoolsr rr rrrrr rrrsF   """"""!!!!!!YYYYYFYYYxS"S"S"S"S"]S"S"S"lMMMMMmMMM`   >>>>>>>>>>r