xjR#ddlmZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZ ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z%ddlm&Z&ddlm'Z'ddlm(Z(ddlm)Z)dd lm*Z*dd!l+m,Z,dd"l-m.Z.e rdd#lm/Z/e d$e/%Z0dod*Z1dpdqd1Z2drd4Z3d+d5dsd8Z4dod9Z5dtd:Z6dud>Z7dvdAZ8dwdDZ9d+d+dEdFdxdKZ:dydMZ;dzdOZddZZ?dd\Z@dd`ZAddbZBddeZCddgZDddhZEddjZFddlZGddnZHdS)) annotationsN)Iterable)Mapping)IO) TYPE_CHECKING)Any)TypeVar) parse_rfc3339) Container)UnexpectedCharError)CUSTOM_ENCODERSAoT)Array)Bool)Comment)Date)DateTime) DottedKey)Float) InlineTable)Integer)Item)Key SingleKey)String) StringType)Table)Time)Trivia Whitespace)item)Parser TOMLDocument)EncoderE)boundstring str | bytesreturnr'c t|S)zF Parses a string into a TOMLDocument. Alias for parse(). )parser+s >## #DT***44666$  BDJJ9555??AAA% ~~ %%%QtDzzQQQnn"$%s"B66 C(#C##C(fpIO[str] | IO[bytes]cDt|S)z5 Load toml document from a file-like object. )r/read)rDs r1loadrHPs   r3r6IO[str]NonecN|t||dS)a 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 :Example: >>> with open("output.toml", "w") as fp: ... tomlkit.dump(data, fp) rIN)writerC)r4rDr6s r1dumprNWs)HHU49 - - -.....r3cDt|S)z7 Parses a string or bytes into a TOMLDocument. )r%r/r0s r1r/r/fs &>>   ! !!r3ctS)z. Returns a new TOMLDocument instance. r&r3r1documentrRms >>r3raw str | intrc:tt|S)z/Create an integer item from a number or string.)r$intrSs r1integerrXus C>>r3 str | floatrc:tt|S)z-Create an float item from a number or string.)r$floatrWs r1float_r\zs c  r3 str | boolrcVtt|tr|dkn|S)z+Turn `true` or `false` into a boolean item.true)r$r;r8rWs r1booleanr`s' C!5!5>v 3 ? ??r3T)literal multilineescaperarbrcrcXtj||}tj|||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. ) _StringTypeselectrfrom_raw)rSrarbrctype_s r1r+r+s*(  w 2 2E ?3v . ..r3rct|}t|tjst dt |S)zCreate a TOML date.z!date() only accepts date strings.)r r; _datetimedate ValueErrorr$rSvalues r1rkrk> #  E eY^ , ,><=== ;;r3r ct|}t|tjst dt |S)zCreate a TOML time.z!time() only accepts time strings.)r r;rjtimerlr$rms r1rqrqror3rct|}t|tjst dt |S)zCreate a TOML datetime.z)datetime() only accepts datetime strings.)r r;rjdatetimerlr$rms r1rsrss@ #  E eY/ 0 0FDEEE ;;r3[]rct|}t|tstdt ||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] zExpected an array, got )rnr;rrlr?)rSvs r1arrayrwsC c A a  ><477<<=== Hr3is_super_table bool | NonercVtttd|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)rr r!)rxs r1tabler{s " fhh~ > >>r3rcVtttdS)zCreate an inline table. :Example: >>> table = inline_table() >>> table.update({'x': 1, 'y': 2}) >>> print(table.as_string()) {x = 1, y = 2} T)new)rr r!rQr3r1 inline_tabler~s! y{{FHH$ 7 7 77r3rc tgS)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 rrQr3r1aotrs r77Nr3kstr | Iterable[str]rct|trt|Sd|D}t|dkr|dSt |S)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 c,g|]}t|SrQr).0_ks r1 zkey.. s & & &bIbMM & & &r3r)r;r8rlenr)rkeyss r1keyrsY!S|| & &A & & &D 4yyA~~Aw T??r3_Itemct|}|}|s!|t|j|S)zParse a simple value from a string. :Example: >>> value("1") 1 >>> value("true") True >>> value("[1, 2, 3]") [1, 2, 3] )char)r% _parse_valueend parse_errorr _current)rSparserrvs r1rnrnsRC[[FA ::<<L  !46? KKK Hr3srctuple[Key, _Item]cDt|S)zkParse a key-value pair from a string. :Example: >>> key_value("foo = 1") (Key('foo'), 1) )r%_parse_key_valuers r1 key_valuer"s #;; ' ' ) ))r3r#c$t|dS)z"Create a whitespace from a string.T)fixedr"rs r1wsr-s c & & &&r3c tdS)zCreate a newline item. )rrQr3r1nlr2s d88Or3rc|d}dd|D}ttd|S)zCreate a comment item. A multiline string produces one ``#``-prefixed line per line so that the result is still valid TOML. rc3(K|] }|rd|ndVdS)z# #NrQ)rlines r1 zcomment..>s1HH$5d#HHHHHHr3z ) comment_wscomment)splitjoinrr!)r+linesrendereds r1rr7sP LL  EyyHH%HHHHHH 6T8<<< = ==r3encoderc.tj||S)aAdd a custom encoder, which should be a function that will be called if the value can't otherwise be converted. The encoder should return a TOMLKit item or raise a ``ConvertError``. Example: @register_encoder def encode_custom_dict(obj, _parent=None, _sort_keys=False): if isinstance(obj, CustomDict): tbl = table() for key, value in obj.items(): # Pass along parameters when encoding nested values tbl[key] = item(value, _parent=tbl, _sort_keys=_sort_keys) return tbl raise ConvertError("Not a CustomDict") )r appendrs r1register_encoderrBs"7### Nr3r(ctjt5tj|ddddS#1swxYwYdS)zUnregister a custom encoder.N) contextlibsuppressrlr removers r1unregister_encoderrWs  Z ( (((w'''((((((((((((((((((s<AA)r+r,r-r')F)r4r5r6r7r-r8)rDrEr-r')r4r5rDrJr6r7r-rK)r-r')rSrTr-r)rSrYr-r)rSr]r-r) rSr8rar7rbr7rcr7r-r)rSr8r-r)rSr8r-r )rSr8r-r)rt)rSr8r-r)N)rxryr-r)r-r)r-r)rrr-r)rSr8r-r)rr8r-r)rr8r-r#)r-r#)r+r8r-r)rr)r-r))rr(r-rK)I __future__rrrsrjcollections.abcrrtypingrrrr tomlkit._utilsr tomlkit.containerr tomlkit.exceptionsr tomlkit.itemsr rrrrrrrrrrrrrrrrrerr r!r#r$tomlkit.parserr%tomlkit.toml_documentr'r(r)r2rCrHrNr/rRrXr\r`r+rkrqrwr{r~rrrnrrrrrrrQr3r1rs""""""$$$$$$###### ((((((''''''222222))))))!!!!!!""""""######%%%%%%!!!!!!''''''###### 333333 $$$$$$&&&&&&!!!!!!>>>>>>$%%%%%%7###A%%%%%0EJ / / / / / /""""  @@@@ //////0     $?????( 8 8 8 8     *    &****''''  >>>>*((((((r3