o n~b|D @s,dZdZdZddlZddlZddlZddlmZddlm Z ddlm Z zddl m Z Wn ddlm Z Ydd lmZejdd krOeZeZd Zd d Znddl m ZeZeZdZddZdd Zdddddddddddd ZddeDZd d!Z d"d#Z!d$d%Z"d&d'Z#e$d(d(fd)d*Z%d+d,Z&d-d.Z'dYd0d1Z(dYd2d3Z)dZd4d5Z*Gd6d7d7eZ+Gd8d9d9e+Z,Gd:d;d;e+Z-Gdd?d?e+Z/d@Z0dAZ1dBZ2dCZ3dDZ4dEZ5dFZ6dGZ7dHZ8dIZ9dJZ:e$d(d/d(fdKdLZ;e$d(d/d/fdMdNZd\dSdTZ?e$d(fdUdVZ@dWdXZAdS)]z6 Module for deserializing/serializing to and from VDF z3.4zRossen GeorgievN)crc32)BytesIO)StringIO)Mapping)VDFDictu￾cCs |tSN)lstripBOMSliner 7/usr/local/lib/python3.10/dist-packages/vdf/__init__.py strip_boms ruÿþþÿz \ufffe\ufeffzunicode-escapecCs|t|tr tStSr)r isinstancestrr BOMS_UNICODEr r r rr$s     \?"') z\nz\tz\vz\bz\rz\fz\az\\z\?z\"z\'cCsi|]\}}||qSr r ).0kvr r r 5sr!cC t|Sr)_escape_char_mapgroupmr r r_re_escape_match7 r'cCr"r)_unescape_char_mapr$r%r r r_re_unescape_match:r(r*cCtdt|S)Nz[\n\t\v\b\r\f\a\\\?\"'])resubr'textr r r_escape=r0cCr+)Nz0(\\n|\\t|\\v|\\b|\\r|\\f|\\a|\\\\|\\\?|\\\"|\\'))r,r-r*r.r r r _unescape@r1r2Tc Cst|ts tdt|t|dstd|g}d}tjdtjd}t|dD];\}}|dkr6t |}| }|dksD|d d krEq)|d d krNd}q)|rat d t |d d|j j|d|f|d dkrt|dkrr|q)t dt |d d|j j|d |f ||} | sz|t|7}Wqtyt dt |d d|j j|d |fw| ddur| dn| d} | d} | dur| d} | dur| } | dkrd} |rt| } | dur%|r| |dvr|d| } t| |s|} |d| <n |} | |d| <| ddur$|| | ddur$d}n?| ddurW| ddurWz|t|7}WqtyVt dt |d d|j j|d |fw|r^t| n| |d| < q)t|dkr~t dt |d d|j j|d |f|S)a@ Deserialize ``s`` (a ``str`` or ``unicode`` instance containing a VDF) to a Python object. ``mapper`` specifies the Python object used after deserializetion. ``dict` is used by default. Alternatively, ``collections.OrderedDict`` can be used if you wish to preserve key order. Or any object that acts like a ``dict``. ``merge_duplicate_keys`` when ``True`` will merge multiple KeyValue lists with the same key into one instead of overwriting. You can se this to ``False`` if you are using ``VDFDict`` and need to preserve the duplicates. .Expected mapper to be subclass of dict, got %sreadlinez>Expected fp to be a file-like object supporting line iterationFz^("(?P(?:\\.|[^\\"])*)"|(?P#?[a-z0-9\-\_\\\?$%<>]+))([ \t]*("(?P(?:\\.|[^\\"])*)(?P")?|(?P(?:(? ])+)|(?P{[ \t]*)(?P})?))?)flagsr/{z$vdf.parse: expected openning bracketnamez<%s>}z+vdf.parse: one too many closing parenthasisTz+vdf.parse: unexpected EOF (open key quote?)ZqkeyNkeyZqvalvalZeblockZsblockZvq_endz1vdf.parse: unexpected EOF (open quote for value?)z/vdf.parse: unclosed parenthasis or quotes (EOF)) issubclassr TypeErrortypehasattrr,compileI enumeraterr SyntaxErrorgetattr __class____name__lenpopmatchnext StopIterationr$rstripr2rappend) fpmappermerge_duplicate_keysescapedstackZexpect_bracketZ re_keyvaluelinenor rLr<r=_mr r rparseDs        "         rXcKsRt|ts tdt|zt|}Wn ty t|}Ynwt|fi|S)zr Deserialize ``s`` (a ``str`` or ``unicode`` instance containing a JSON document) to a Python object. zExpected s to be a str, got %s)r string_typer@rA unicodeIOstrIOrX)skwargsrQr r rloadss    r^cKst|fi|S)z Deserialize ``fp`` (a ``.readline()``-supporting file-like object containing a JSON document) to a Python object. )rX)rQr]r r rloadsr_FcCsHt|ts tdt|tstdt|tstddt|||S)z7 Serialize ``obj`` to a VDF formatted ``str``. *Expected data to be an instance of``dict``"Expected pretty to be of type bool#Expected escaped to be of type boolr7)rrr@booljoin _dump_gen)objprettyrTr r rdumpss   rhcCsht|ts tdt|dstdt|tstdt|ts$tdt|||D]}||q*dS)zq Serialize ``obj`` as a VDF formatted stream to ``fp`` (a ``.write()``-supporting file-like object). r`write"Expected fp to have write() methodrarbN)rrr@rBrcreri)rfrQrgrTchunkr r rdumps     rlc csd}d}|r ||}|D]D\}}|rt|trt|}t|tr@d|||fVt||||dD]}|Vq4d|Vq|rKt|trKt|}d|||fVqdS)Nrr7z %s"%s" %s{ r6z%s} z %s"%s" "%s" )itemsrrYr0rre) datargrTlevelindentZ line_indentr<valuerkr r rres"  rec@seZdZddZdS)BASE_INTcCsd|jj|fS)Nz%s(%d))rHrI)selfr r r__repr__szBASE_INT.__repr__N)rI __module__ __qualname__rtr r r rrrs rrc@ eZdZdS)UINT_64NrIrurvr r r rrxrxc@rw)INT_64Nryr r r rr{rzr{c@rw)POINTERNryr r r rr|rzr|c@rw)COLORNryr r r rr}rzr}  cCs.t|ts tdt|tt|||||S)a9 Deserialize ``b`` (``bytes`` containing a VDF in "binary form") to a Python object. ``mapper`` specifies the Python object used after deserializetion. ``dict` is used by default. Alternatively, ``collections.OrderedDict`` can be used if you wish to preserve key order. Or any object that acts like a ``dict``. ``merge_duplicate_keys`` when ``True`` will merge multiple KeyValue lists with the same key into one instead of overwriting. You can se this to ``False`` if you are using ``VDFDict`` and need to preserve the duplicates. zExpected s to be bytes, got %s)rbytesr@rA binary_loadr)brRrS alt_formatraise_on_remainingr r r binary_loads*s rcstdrtdrtdstdt|ts tdt|td}td}td}td }dd d } |g} |sAtnt} t fd ddD]} | | kr_t | dkr]| qLn| } | t kr|rv| | dvrv| d| }n |}|| d| <| |qL| tkr| | d| <qL| tkr| dd| d| <qL| tttfvr||jd}| tkrt|}n| tkrt|}|| d| <qL| tkrt||jd| d| <qL| tkrt||jd| d| <qL| tkr ||jd| d| <qLtddt| ft | dkr%td|r@ddkr@ ddtdd| S)aN Deserialize ``fp`` (a ``.read()``-supporting file-like object containing binary VDF) to a Python object. ``mapper`` specifies the Python object used after deserializetion. ``dict` is used by default. Alternatively, ``collections.OrderedDict`` can be used if you wish to preserve key order. Or any object that acts like a ``dict``. ``merge_duplicate_keys`` when ``True`` will merge multiple KeyValue lists with the same key into one instead of overwriting. You can se this to ``False`` if you are using ``VDFDict`` and need to preserve the duplicates. readtellseekzRExpected fp to be a file-like object with tell()/seek() and read() returning bytesr3r>@rz!Unterminated cstring (offset: %d)sr~r6utf-16utf-8replaceascii) rrrFfindrrJdecoderr)rQwidebufendoffsetrkresultr r r read_stringTs2         z binary_load..read_stringcs dS)Nr6)rr rQr rzs zbinary_load..rr6r>T)rrz"Unknown data type at offset %d: %sz)Reached EOF, but Binary VDF is incompletez?Binary VDF ended at offset %d, but there is more data remainingNF)!rBr@r?rrAstructStructBIN_END BIN_END_ALTiterrJrKBIN_NONErP BIN_STRINGBIN_WIDESTRING BIN_INT32 BIN_POINTER BIN_COLORunpackrsizer|r} BIN_UINT64rx BIN_INT64r{ BIN_FLOAT32rFrreprr)rQrRrSrrint32uint64int64float32rrUZCURRENT_BIN_ENDtr<rWr=r rrr<s`       #     $$   rcCst}t||||S)z@ Serialize ``obj`` to a binary VDF formatted ``bytes``. )r binary_dumpgetvalue)rfrrr r r binary_dumpss rcCsDt|ts tdt|dstdt||dD]}||qdS)zf Serialize ``obj`` to a binary VDF formatted ``bytes`` and write it to ``fp`` filelike object z"Expected obj to be type of MappingrirjrN)rrr@rB_binary_dump_genri)rfrQrrkr r rrs   rc cs|dkr t|dkr dStd}td}td}td}|D]\}}t|tr4|d}ntdt|t|t rXt |t Vt ||d|d D]} | VqQq%t|t rjt |t ||Vq%t|tr|t|t ||Vq%t|trz |dt }tVWn|d t d }tVY|t |Vq%t|trt|t ||Vq%t|ttttfrt|trtVn t|trtVntV|t V||Vq%td t||stVdStVdS) Nrrrrrrz%dict keys must be of type str, got %sr6rrrzUnsupported type: %s)rJrrrmrrYencoder@rArrrrxrpackr{rrrfloatrr}r|intint_typerrrrr) rfrorrrrrr<rqrkr r rrsP               rcCsb|dddkr tdtd|dd\}|t|ddkr%tdt|dd||dd S) a$ Deserialize ``s`` (``bytes`` containing a VBKV to a Python object. ``mapper`` specifies the Python object used after deserializetion. ``dict` is used by default. Alternatively, ``collections.OrderedDict`` can be used if you wish to preserve key order. Or any object that acts like a ``dict``. ``merge_duplicate_keys`` when ``True`` will merge multiple KeyValue lists with the same key into one instead of overwriting. You can se this to ``False`` if you are using ``VDFDict`` and need to preserve the duplicates. NVBKVzInvalid headerrzInvalid checksumTr) ValueErrorrrrr)r\rRrSchecksumr r r vbkv_loadss  rcCs.dt|dd}t|}dtd||S)z: Serialize ``obj`` to a VBKV formatted ``bytes``. rTrrr)rdrrrr)rfrnrr r r vbkv_dumpssr)FT)FTrr)rF)B__doc__ __version__ __author__r,sysrbinasciiriorrrZcollections.abcr collectionsZ vdf.vdictr version_inforrYrrr rr[ basestringlongrrr)rmr#r'r*r0r2dictrXr^r_rhrlrerrrxr{r|r}rrrrrrrrrrrrrrrrrrr r r rs        z    l  0