o >@i @sddlmZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z ddl m Z dd 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-e rdd"lm.Z.e d#e.d$Z/dd)d*Z0ddd1d2Z1dd5d6Z2d+d7dd:d;Z3ddd?Z5ddCdDZ6ddGdHZ7ddKdLZ8d+d+dMdNddSd%Z9ddUdVZ:ddXdYZ;dd[d\Zddd_d`ZddjdkZ?ddodpZ@ddrdsZAddvdwZBddydzZCdd{d|ZDdd~dZEdddZFdddZGdS)) annotationsN)Iterable)Mapping)IO) TYPE_CHECKING)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$cCt|S)zF Parses a string into a TOMLDocument. Alias for parse(). )parser(r.I/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/tomlkit/api.pyloads.sr0Fdatar sort_keysboolstrc Csht|tttfst|trtt||d}z|WSty3}z dt |d}t ||d}~ww)z- Dumps a TOMLDocument into a string. ) _sort_keysz.Expecting Mapping or TOML Table or Container, z givenN) isinstancerrr rr!dict as_stringAttributeErrortype TypeError)r1r2exmsgr.r.r/dumps7s  r>fpIO[str] | IO[bytes]cCs t|S)z5 Load toml document from a file-like object. )r,read)r?r.r.r/loadI rBr2IO[str]NonecCs|t||ddS)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) rDN)writer>)r1r?r2r.r.r/dumpPs rHcC t|S)z7 Parses a string or bytes into a TOMLDocument. )r"r,r-r.r.r/r,_rCr,cCstS)z. Returns a new TOMLDocument instance. r#r.r.r.r/documentfsrJraw str | intrcC tt|S)z/Create an integer item from a number or string.)r!intrKr.r.r/integern rP str | floatrcCrM)z-Create an float item from a number or string.)r!floatrOr.r.r/float_srQrT str | boolrcCstt|tr |dkS|S)z+Turn `true` or `false` into a boolean item.true)r!r6r4rOr.r.r/booleanxsrWT)literal multilineescaperXrYrZrcCst||}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. ) _StringTypeselectrfrom_raw)rKrXrYrZtype_r.r.r/r(}s rcC$t|}t|tjstdt|S)zCreate a TOML date.z!date() only accepts date strings.)rr6 _datetimedate ValueErrorr!rKvaluer.r.r/ra rarcCr_)zCreate a TOML time.z!time() only accepts time strings.)rr6r`timerbr!rcr.r.r/rfrerfrcCr_)zCreate a TOML datetime.z)datetime() only accepts datetime strings.)rr6r`datetimerbr!rcr.r.r/rgrerg[]rcCr+)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] )rdrOr.r.r/arrays riis_super_table bool | NonercCstttd|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)rjr.r.r/tablesrlrcCstttddS)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 rr.r.r.r/ inline_tables rnr 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 rokstr | Iterable[str]rcCs$t|tr t|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..)r6r4rr)rpr.r.r/rrs rr_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" _parse_valueend parse_errorr _current)rKparservr.r.r/rds  rdsrctuple[Key, _Item]cCrI)zkParse a key-value pair from a string. :Example: >>> key_value("foo = 1") (Key('foo'), 1) )r"_parse_key_valuer~r.r.r/ key_values rr cCs t|ddS)z"Create a whitespace from a string.T)fixedrrr.r.r/ws rQrcCstdS)zCreate a newline item. )rr.r.r.r/nl%srrcCsttdd|dS)zCreate a comment item.z z# ) comment_wscomment)rrr-r.r.r/r*srencodercCst||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 appendrr.r.r/register_encoder/s rr%cCs:ttt|WddS1swYdS)zUnregister a custom encoder.N) contextlibsuppressrbr removerr.r.r/unregister_encoderDs  "r)r(r)r*r$)F)r1rr2r3r*r4)r?r@r*r$)r1rr?rEr2r3r*rF)r*r$)rKrLr*r)rKrRr*r)rKrUr*r) rKr4rXr3rYr3rZr3r*r)rKr4r*r)rKr4r*r)rKr4r*r)rh)rKr4r*r)N)rjrkr*r)r*r)r*r )rprqr*r)rKr4r*rv)r~r4r*r)r~r4r*r )r*r )r(r4r*r)rr&r*r&)rr%r*rF)H __future__rrrgr`collections.abcrrtypingrrrtomlkit._utilsrtomlkit.containerr tomlkit.exceptionsr tomlkit.itemsr r rrrrrrrrrrrvrrrrr[rrrr r!tomlkit.parserr"tomlkit.toml_documentr$r%r&r0r>rBrHr,rJrPrTrWr(rarfrirlrnrorrrdrrrrrrr.r.r.r/s