U Te@sJddlmZddlZddlZddlmZddlmZddlm Z ddlm Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z$ddlm%Z%ddlm&Z&ddlm'Z'ddlm(Z(ddlm)Z)dd l*m+Z+dd!l,m-Z-d"d#d$d%d&Z.dd(d)d*d+d,d-Z/d.d#d/d0d1Z0d'd2d(d3d)d4d5d6d7Z1d"d#d$d8d9Z2d#d:d;d<Z3d=d>d?d@dAZ4dBdCd?dDdEZ5d*dFd?dGdHZ6d'd'dIdJd*d)d)d)dKdLdMdNZ7d*dOd?dPdQZ8d*dRd?dSdTZ9d*dUd?dVdWZdd*dXd?dYdZZ:dd[d\d]d^d_Z;d`d:dadbZd*dkd?dldmZ?d*dndodpdqZ@d*drdodsdtZAdrd:dudvZBd*dwd$dxdyZCe dzed{ZDdzdzd|d}d~ZEdd4d|ddZFdS)) annotationsN)Mapping)IO)Iterable)TypeVar) parse_rfc3339) Container)UnexpectedCharError)CUSTOM_ENCODERSAoT)Array)Bool)Comment)Date)DateTime) DottedKey)Encoder)Float) InlineTable)Integer)Item)Key) SingleKey)String) StringType)Table)Time)Trivia Whitespaceitem)Parser TOMLDocumentz str | bytesr%)stringreturncCst|S)zF Parses a string into a TOMLDocument. Alias for parse(). )parser&r*j/mounts/lovelace/software/anaconda3/envs/qiime2-amplicon-2024.2/lib/python3.8/site-packages/tomlkit/api.pyloads(sr,Frboolstr)data sort_keysr'c Cspt|ts$t|tr$tt||d}z |WStk rj}zdt|d}t||W5d}~XYnXdS)z- Dumps a TOMLDocument into a string. ) _sort_keysz%Expecting Mapping or TOML Container, z givenN) isinstancerrr"dict as_stringAttributeErrortype TypeError)r/r0exmsgr*r*r+dumps1s r:zIO[str] | IO[bytes])fpr'cCs t|S)z5 Load toml document from a file-like object. )r(read)r;r*r*r+loadAsr=r0zIO[str]None)r/r;r0r'cCs|t||ddS)z Dump a TOMLDocument into a writable file stream. :param data: a dict-like object to dump :param sort_keys: if true, sort the keys in alphabetic order r>N)writer:)r/r;r0r*r*r+dumpHsrAcCs t|S)z7 Parses a string or bytes into a TOMLDocument. )r#r(r)r*r*r+r(Rsr()r'cCstS)z. Returns a new TOMLDocument instance. r$r*r*r*r+documentYsrBz str | intr)rawr'cCs tt|S)z/Create an integer item from a number or string.)r"intrCr*r*r+integerasrFz str | floatrcCs tt|S)z-Create an float item from a number or string.)r"floatrEr*r*r+float_fsrHrcCs t|dkS)z+Turn `true` or `false` into a boolean item.truer!rEr*r*r+booleanksrJT)literal multilineescaper)rCrKrLrMr'cCst||}t|||S)aBCreate a string item. By default, this function will create *single line basic* strings, but boolean flags (e.g. ``literal=True`` and/or ``multiline=True``) can be used for personalization. For more information, please check the spec: ``__. Common escaping rules will be applied for basic strings. This can be controlled by explicitly setting ``escape=False``. Please note that, if you disable escaping, you will have to make sure that the given strings don't contain any forbidden character or sequence. ) _StringTypeselectrZfrom_raw)rCrKrLrMtype_r*r*r+r&ps r&rcCs$t|}t|tjstdt|S)zCreate a TOML date.z!date() only accepts date strings.)rr2 _datetimedate ValueErrorr"rCvaluer*r*r+rRs rRrcCs$t|}t|tjstdt|S)zCreate a TOML time.z!time() only accepts time strings.)rr2rQtimerSr"rTr*r*r+rVs rVrcCs$t|}t|tjstdt|S)zCreate a TOML datetime.z)datetime() only accepts datetime strings.)rr2rQdatetimerSr"rTr*r*r+rWs rWr cCs|dkr d}t|S)zCreate an array item for its string representation. :Example: >>> array("[1, 2, 3]") # Create from a string [1, 2, 3] >>> a = array() >>> a.extend([1, 2, 3]) # Create from a list >>> a [1, 2, 3] Nz[])rUrEr*r*r+arrays rXz bool | Noner)is_super_tabler'cCstttd|S)aFCreate an empty table. :param is_super_table: if true, the table is a super table :Example: >>> doc = document() >>> foo = table(True) >>> bar = table() >>> bar.update({'x': 1}) >>> foo.append('bar', bar) >>> doc.append('foo', foo) >>> print(doc.as_string()) [foo.bar] x = 1 F)rrr)rYr*r*r+tablesrZrcCstttddS)zCreate an inline table. :Example: >>> table = inline_table() >>> table.update({'x': 1, 'y': 2}) >>> print(table.as_string()) {x = 1, y = 2} T)new)rrrr*r*r*r+ inline_tables r\r cCstgS)zCreate an array of table. :Example: >>> doc = document() >>> aot = aot() >>> aot.append(item({'x': 1})) >>> doc.append('foo', aot) >>> print(doc.as_string()) [[foo]] x = 1 r r*r*r*r+aots r]zstr | Iterable[str]r)kr'cCs$t|trt|Stdd|DS)aCreate a key from a string. When a list of string is given, it will create a dotted key. :Example: >>> doc = document() >>> doc.append(key('foo'), 1) >>> doc.append(key(['bar', 'baz']), 2) >>> print(doc.as_string()) foo = 1 bar.baz = 2 cSsg|] }t|qSr*)key).0_kr*r*r+ szkey..)r2r.rr)r^r*r*r+r_s r__ItemcCs,t|}|}|s(|jt|jd|S)zParse a simple value from a string. :Example: >>> value("1") 1 >>> value("true") True >>> value("[1, 2, 3]") [1, 2, 3] )char)r#Z _parse_valueendZ parse_errorr _current)rCparservr*r*r+rUs rUztuple[Key, _Item])srcr'cCs t|S)zkParse a key-value pair from a string. :Example: >>> key_value("foo = 1") (Key('foo'), 1) )r#Z_parse_key_valuerir*r*r+ key_value srkr cCs t|ddS)z"Create a whitespace from a string.T)fixedrrjr*r*r+wssrmcCstdS)zCreate a newline item. )rmr*r*r*r+nlsrorcCsttdd|dS)zCreate a comment item.z z# )Z comment_wscomment)rrr)r*r*r+rp srpE)bound)encoderr'cCst||S)zAdd a custom encoder, which should be a function that will be called if the value can't otherwise be converted. It should takes a single value and return a TOMLKit item or raise a ``TypeError``. )r appendrsr*r*r+register_encoder(s rvrc Cs$ttt|W5QRXdS)zUnregister a custom encoder.N) contextlibsuppressrSr removerur*r*r+unregister_encoder1s rz)F)N)N)G __future__rrwrWrQcollections.abcrtypingrrrZtomlkit._utilsrZtomlkit.containerrZtomlkit.exceptionsr Z tomlkit.itemsr r r rrrrrrrrrrrcrrrrrNrrrr r"Ztomlkit.parserr#Ztomlkit.toml_documentr%r,r:r=rAr(rBrFrHrJr&rRrVrXrZr\r]r_rUrkrmrorprqrvrzr*r*r*r+s|