a Z%b!@sxdZddlmZddlmZddlZddlmZddlZddl m Z m Z m Z m Z mZmZmZddZGd d d eZdS) z0.4)contextmanager)partialN) copyfileobj) text_typePY2reraiseStringIOBytesIOIteratorwrapscsfdd}|S)Ncstfdd}|S)NcsN|g|Ri|}r0t|ts0||jSsJt|trJ||jS|SN) isinstancerdecodeencodingencode)selfargskwargsrv)enabledf]/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/data/__init__.py_s   z*enable_unicode..wrapper.._)r )rrr)rrwrapper szenable_unicode..wrapperr)rrrrrenable_unicode s rc@seZdZdZdZdZdZdZd)ddZddZ ddZ d d Z d d Z d dZ ddZddZddZddZeddZedddZedddZeddd Zd!d"Zd#d$Zed*d'd(ZdS)+DataaDynamically converts between various forms of passed in input data. Exactly one of ``arg``, ``data`` or ``file`` must be not-``None``. :param arg: Dynamic argument. If a bytestring, will be interpreted as raw bytes. A unicode string will be interpreted as text and any object that has a ``read()`` method as a file-like. Any instance of ``Data`` will be passed through and rendered unusable. :param encoding: The data's encoding. Will be used for every conversion from bytestrings to text (unicode) if necessary and the other way around. :param data: Buffer argument. If unicode string, will be interpreted as text, otherwise as bytestring. :param file: File argument. Any object with a ``read()`` method will be treated as file-like. Everything else is considered a filename.NcCs||||f|_|||gddkr*td|durt||jrv|jpH|j}|jpT|j}|j }d|_|_|_|_nt |dr|}n|}|durt|t r||_q||_n2|durt |dr||_t |ddr|j }n||_|pd|_ dS)Nz'Must supply exactly one of data or filereadrutf8) orig_argscount ValueErrorr __class__datatextfilefilenamerhasattrrgetattr)rargrr%r'rrr__init__1s.       z Data.__init__cCs|jdur|jS|jdur(|j|jS|jdur:|S|jdurxt|jd}|WdS1sn0Yt ddS)z>Returns the data as bytes (on Python3) or string (on Python2).NrbBroken Data, all None.) r%r&rrr'readbr(openrr#)rrrrr __bytes__Rs    &zData.__bytes__cCs|S)zhContext manager support. If data is a file-like, will close it upon exiting the context manager.rrrrr __enter__cszData.__enter__cCs |dSr )close)rZexception_typeZexception_value tracebackrrr__exit__hsz Data.__exit__cCs|S)zdIterator support. Returns lines (similar to file objects) using :meth:`~data.Data.readline`.rr2rrr__iter__ksz Data.__iter__cCs|}|st|Sr )readline StopIteration)rchunkrrr__next__psz Data.__next__cCstr |S|S)z@Returns the data as unicode (on Python3) or string (on Python2).)rr1 __unicode__r2rrr__str__vsz Data.__str__cCs2|jdur|jS|jdur"|S||jS)zReturns the data as unicode.N)r&r'rr1rrr2rrrr<|s   zData.__unicode__cCshdd}|jj}|jdur0d|||j|jS|jdurPd|||j|jSd||jp`|j|jS)NcSs(t|dkrt|St|dddS)Nz...)lenrepr)bufrrrheads zData.__repr__..headz{}(data={}, encoding={!r})z{}(file={!r}, encoding={!r}))r$__name__r%formatrr&r'r()rrBcnamerrr__repr__s  z Data.__repr__cCs,t|dddur|jdurdS|jdS)z?Closes input if based on open filelike. Otherwise does nothing._streamN)r*r'streamr4r2rrrr4sz Data.closecCsvt|dsp|jdur|j|_nR|jdur8t|jd|_n8|jdurPt|j|_n |jdurht|j|_nt d|jS)zpReturns a stream object (:func:`file`, :class:`~io.BytesIO` or :class:`~StringIO.StringIO`) on the data.rGNr-r.) r)r'rGr(r0r&rr%r r#r2rrrrHs      z Data.streamTcOs|jj|i|S)z`Read method, implements same interface as :func:`file.read`. Always returns ``unicode``.rHrrrrrrrrsz Data.readFcOs|jj|i|S)z>Like :meth:`~data.Data.read`, but returns bytestrings instead.rIrJrrrr/sz Data.readbcOs |jj|S)z4Return one line from stream. Always returns unicode.)rHr8rJrrrr8sz Data.readlinecOs"ttt|jg|Ri|dS)z9Return list of all lines. Always returns list of unicode.)listiterrr8rJrrr readlinesszData.readlinescCs|}t|dr|jdur8t|jdddur8t|j|q|jdurzt|jd}t||Wdq1sn0Yq||n4t|d}||WdS1s0YdS)zSave data to file. Will copy by either writing out the data or using :func:`shutil.copyfileobj`. :param file: A file-like object (with a ``write`` method) or a filename.writeNrr-wb) r)r'r*rr(r0rOr1save_to)rr'destinpoutrrrrQs   * z Data.save_torKtmpccstj|||dd}zn||||d|VWzt|jWqty~}z|j dkrjt |WYd}~qd}~00nJzt|jWn6ty}z|j dkrt |WYd}~n d}~000dS)a$Saves data to temporary file and returns the relevant instance of :func:`~tempfile.NamedTemporaryFile`. The resulting file is not deleted upon closing, but when the context manager exits. Other arguments are passed on to :func:`~tempfile.NamedTemporaryFile`. F)suffixprefixdirdeleterrN) tempfileNamedTemporaryFilerQflushseekosunlinknameOSErrorerrnor)rrVrWrXrUerrr temp_saveds*     zData.temp_saved)NNNN)rKrUN)rC __module__ __qualname____doc__r%r&r'r(r,r1r3r6r7r;r=r<rFr4propertyrHrrr/r8rNrQrrdrrrrrs6 !     r) __version__ contextlibr functoolsrr^shutilrrZsixrrrrr r r rrrrrrs   $