U 4Je@sddZddlmZddlZddlmZmZmZmZm Z ddl m Z Gdddej Z Gd d d ZdS) zImplementation of the DbDoc.) annotationsN)AnyDictKeysViewOptionalUnion)ProgrammingErrorc@s eZdZdZdddddZdS)ExprJSONEncoderzgA :class:`json.JSONEncoder` subclass, which enables encoding of :class:`mysqlx.ExprParser` objects.objectstr)oreturncCst|dr|Stj||S)Nexpr)hasattrjson JSONEncoderdefault)selfr r0/tmp/pip-unpacked-wheel-7_167w8m/mysqlx/dbdoc.pyr,s zExprJSONEncoder.defaultN)__name__ __module__ __qualname____doc__rrrrrr (sr c@seZdZdZdddddZddd d Zddd d Zdd ddddZdd dddZdddddZ d"dddddZ ddddZ ddd d!Z dS)#DbDoczRepresents a generic document in JSON format. Args: value (object): The value can be a JSON string or a dict. Raises: ValueError: If ``value`` type is not a basestring or dict. zUnion[str, Dict[str, Any]]None)valuercCs@t|tr||_n*t|tr*t||_ntdt|dS)NzUnable to handle type: ) isinstancedict__dict__r rloads ValueErrortype)rrrrr__init__=s   zDbDoc.__init__r )rcCs|SN)as_strrrrr__str__Esz DbDoc.__str__cCs t|jSr%)reprr r'rrr__repr__HszDbDoc.__repr__r)indexrrcCs|dkrtd||j|<dS)N_idzCannot modify _id)r r )rr+rrrr __setitem__KszDbDoc.__setitem__)r+rcCs |j|Sr%r )rr+rrr __getitem__PszDbDoc.__getitem__bool)itemrcCs ||jkSr%r.)rr1rrr __contains__SszDbDoc.__contains__Nz Optional[str])doc_idrcCs.|j}|r||d<nd|kr&|d=t|S)aeReturns a new copy of a :class:`mysqlx.DbDoc` object containing the `doc_id` provided. If `doc_id` is not provided, it will be removed from new :class:`mysqlx.DbDoc` object. Args: doc_id (Optional[str]): Document ID Returns: mysqlx.DbDoc: A new instance of DbDoc containing the _id provided r,)r copyr)rr3Znew_dictrrrr4Vs  z DbDoc.copyz KeysView[str]cCs |jS)zJReturns the keys. Returns: `list`: The keys. )r keysr'rrrr5hsz DbDoc.keyscCstj|jtdS)zSerialize :class:`mysqlx.DbDoc` to a JSON formatted ``str``. Returns: str: A JSON formatted ``str`` representation of the document. .. versionadded:: 8.0.16 )cls)rdumpsr r r'rrrr&psz DbDoc.as_str)N) rrrrr$r(r*r-r/r2r4r5r&rrrrr3s r)r __future__rrtypingrrrrrerrorsr rr rrrrrs