o 'hK@sUdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZmZmZmZmZmZmZddlmZmZmZmZddlmZddlmZmZddlm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*dd l+m,Z,dd l-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ej9ej:ej;ejdZ?GdddZ@GdddZAe reeeBe fZCneZCdZDGdddeCZEeEeAjFdZGdeHd< eEeAjIdZJdeHd< eEeAjKdZLdeHd< eLZMdeHd< dd&d'ZNdd*d+ZOeMfdd-d.ZPeMfdd1d2ZQeMfdd5d6ZRdd9d:ZSdd>> from bson.json_util import loads >>> loads( ... '[{"foo": [1, 2]}, {"bar": {"hello": "world"}}, {"code": {"$scope": {}, "$code": "function x() { return 1; }"}}, {"bin": {"$type": "80", "$binary": "AQIDBA=="}}]' ... ) [{'foo': [1, 2]}, {'bar': {'hello': 'world'}}, {'code': Code('function x() { return 1; }', {})}, {'bin': Binary(b'...', 128)}] Example usage with :const:`RELAXED_JSON_OPTIONS` (the default): .. doctest:: >>> from bson import Binary, Code >>> from bson.json_util import dumps >>> dumps( ... [ ... {"foo": [1, 2]}, ... {"bar": {"hello": "world"}}, ... {"code": Code("function x() { return 1; }")}, ... {"bin": Binary(b"")}, ... ] ... ) '[{"foo": [1, 2]}, {"bar": {"hello": "world"}}, {"code": {"$code": "function x() { return 1; }"}}, {"bin": {"$binary": {"base64": "AQIDBA==", "subType": "00"}}}]' Example usage (with :const:`CANONICAL_JSON_OPTIONS`): .. doctest:: >>> from bson import Binary, Code >>> from bson.json_util import dumps, CANONICAL_JSON_OPTIONS >>> dumps( ... [ ... {"foo": [1, 2]}, ... {"bar": {"hello": "world"}}, ... {"code": Code("function x() { return 1; }")}, ... {"bin": Binary(b"")}, ... ], ... json_options=CANONICAL_JSON_OPTIONS, ... ) '[{"foo": [{"$numberInt": "1"}, {"$numberInt": "2"}]}, {"bar": {"hello": "world"}}, {"code": {"$code": "function x() { return 1; }"}}, {"bin": {"$binary": {"base64": "AQIDBA==", "subType": "00"}}}]' Example usage (with :const:`LEGACY_JSON_OPTIONS`): .. doctest:: >>> from bson import Binary, Code >>> from bson.json_util import dumps, LEGACY_JSON_OPTIONS >>> dumps( ... [ ... {"foo": [1, 2]}, ... {"bar": {"hello": "world"}}, ... {"code": Code("function x() { return 1; }", {})}, ... {"bin": Binary(b"")}, ... ], ... json_options=LEGACY_JSON_OPTIONS, ... ) '[{"foo": [1, 2]}, {"bar": {"hello": "world"}}, {"code": {"$code": "function x() { return 1; }", "$scope": {}}}, {"bin": {"$binary": "AQIDBA==", "$type": "00"}}]' Alternatively, you can manually pass the `default` to :func:`json.dumps`. It won't handle :class:`~bson.binary.Binary` and :class:`~bson.code.Code` instances (as they are extended strings you can't provide custom defaults), but it will be faster as there is less recursion. .. note:: If your application does not need the flexibility offered by :class:`JSONOptions` and spends a large amount of time in the `json_util` module, look to `python-bsonjs `_ for a nice performance improvement. `python-bsonjs` is a fast BSON to MongoDB Extended JSON converter for Python built on top of `libbson `_. `python-bsonjs` works best with PyMongo when using :class:`~bson.raw_bson.RawBSONDocument`. ) annotationsN) TYPE_CHECKINGAnyCallableMappingMutableMappingOptionalSequenceTupleTypeUnioncast)ALL_UUID_SUBTYPES UUID_SUBTYPEBinaryUuidRepresentation)Code) CodecOptionsDatetimeConversion) _MAX_UTC_MS EPOCH_AWARE DatetimeMS_datetime_to_millis_millis_to_datetime)DBRef) Decimal128)Int64)MaxKey)MinKey)ObjectId)Regex)RE_TYPE Timestamp)utc)ilmsuxc@eZdZdZ dZ dZdS)DatetimeRepresentationrN)__name__ __module__ __qualname__LEGACY NUMBERLONGISO8601r5r5J/var/www/html/olx_land/venv/lib/python3.10/site-packages/bson/json_util.pyr,s  r,c@r+)JSONModerr-r.N)r/r0r1r2RELAXED CANONICALr5r5r5r6r7s r7lcseZdZUded<ded<ded<ded<ded<dfd d Zdddejfdfdd Zd fdd Zd!fdd Z d"ddZ Z S)# JSONOptionsint json_modeboolstrict_number_longdatetime_representation strict_uuidzType[MutableMapping[str, Any]]document_classargsrkwargscstdS)a Encapsulates JSON options for :func:`dumps` and :func:`loads`. :param strict_number_long: If ``True``, :class:`~bson.int64.Int64` objects are encoded to MongoDB Extended JSON's *Strict mode* type `NumberLong`, ie ``'{"$numberLong": "" }'``. Otherwise they will be encoded as an `int`. Defaults to ``False``. :param datetime_representation: The representation to use when encoding instances of :class:`datetime.datetime`. Defaults to :const:`~DatetimeRepresentation.LEGACY`. :param strict_uuid: If ``True``, :class:`uuid.UUID` object are encoded to MongoDB Extended JSON's *Strict mode* type `Binary`. Otherwise it will be encoded as ``'{"$uuid": "" }'``. Defaults to ``False``. :param json_mode: The :class:`JSONMode` to use when encoding BSON types to Extended JSON. Defaults to :const:`~JSONMode.LEGACY`. :param document_class: BSON documents returned by :func:`loads` will be decoded to an instance of this class. Must be a subclass of :class:`collections.MutableMapping`. Defaults to :class:`dict`. :param uuid_representation: The :class:`~bson.binary.UuidRepresentation` to use when encoding and decoding instances of :class:`uuid.UUID`. Defaults to :const:`~bson.binary.UuidRepresentation.UNSPECIFIED`. :param tz_aware: If ``True``, MongoDB Extended JSON's *Strict mode* type `Date` will be decoded to timezone aware instances of :class:`datetime.datetime`. Otherwise they will be naive. Defaults to ``False``. :param tzinfo: A :class:`datetime.tzinfo` subclass that specifies the timezone from which :class:`~datetime.datetime` objects should be decoded. Defaults to :const:`~bson.tz_util.utc`. :param datetime_conversion: Specifies how UTC datetimes should be decoded within BSON. Valid options include 'datetime_ms' to return as a DatetimeMS, 'datetime' to return as a datetime.datetime and raising a ValueError for out-of-range values, 'datetime_auto' to return DatetimeMS objects when the underlying datetime is out-of-range and 'datetime_clamp' to clamp to the minimum and maximum possible datetimes. Defaults to 'datetime'. See :ref:`handling-out-of-range-datetimes` for details. :param args: arguments to :class:`~bson.codec_options.CodecOptions` :param kwargs: arguments to :class:`~bson.codec_options.CodecOptions` .. seealso:: The specification for Relaxed and Canonical `Extended JSON`_. .. versionchanged:: 4.0 The default for `json_mode` was changed from :const:`JSONMode.LEGACY` to :const:`JSONMode.RELAXED`. The default for `uuid_representation` was changed from :const:`~bson.binary.UuidRepresentation.PYTHON_LEGACY` to :const:`~bson.binary.UuidRepresentation.UNSPECIFIED`. .. versionchanged:: 3.5 Accepts the optional parameter `json_mode`. .. versionchanged:: 4.0 Changed default value of `tz_aware` to False. N)super__init__)selfrBrC __class__r5r6rEs6zJSONOptions.__init__NclsType[JSONOptions]Optional[bool] Optional[int]returncsz|dd|d<|dr|dt|d<|tjtjtjdfvr#tdttt j |g|Ri|}|t jt j t j fvrAtd||_|jt j kro|rPtd|dtjfvr[td|dvrctd d|_tj|_d |_|S|jt j kr|dvr}td |dtjfvrtd |dvrtd d |_tj|_d |_|Sd|_tj|_d|_|dur||_|dur||_|dur||_|S) Ntz_awareFtzinfoznJSONOptions.datetime_representation must be one of LEGACY, NUMBERLONG, or ISO8601 from DatetimeRepresentation.zQJSONOptions.json_mode must be one of LEGACY, RELAXED, or CANONICAL from JSONMode.zr?r@)rIr>r?r@r<rBrCrFrGr5r6rR1sl     zJSONOptions.__new__strcs d|j|j|j|jtS)Nz[strict_number_long={!r}, datetime_representation={!r}, strict_uuid={!r}, json_mode={!r}, {})formatr>r?r@r<rD_arguments_repr)rFrGr5r6rUtszJSONOptions._arguments_reprdict[Any, Any]cs*t}||j|j|j|jd|S)Nr>r?r@r<)rD _options_dictupdater>r?r@r<)rF options_dictrGr5r6rXs zJSONOptions._options_dictcKs@|}dD] }||t||||<q||tdi|S)a Make a copy of this JSONOptions, overriding some options:: >>> from bson.json_util import CANONICAL_JSON_OPTIONS >>> CANONICAL_JSON_OPTIONS.tz_aware True >>> json_options = CANONICAL_JSON_OPTIONS.with_options(tz_aware=False, tzinfo=None) >>> json_options.tz_aware False .. versionadded:: 3.12 rWNr5)rXrPgetattrrYr:)rFrCoptsoptr5r5r6 with_optionss   zJSONOptions.with_options)rBrrCr)rIrJr>rKr?rLr@rKr<r;rBrrCrrMr:)rMrS)rMrV)rCrrMr:) r/r0r1__annotations__rEr7r8rRrUrXr^ __classcell__r5r5rGr6r:s :C  r:)r<LEGACY_JSON_OPTIONSCANONICAL_JSON_OPTIONSRELAXED_JSON_OPTIONSDEFAULT_JSON_OPTIONSobjrrBrCrMrScOs*|dt}tjt||g|Ri|S)aQHelper function that wraps :func:`json.dumps`. Recursive function that handles all BSON types including :class:`~bson.binary.Binary` and :class:`~bson.code.Code`. :param json_options: A :class:`JSONOptions` instance used to modify the encoding of MongoDB Extended JSON types. Defaults to :const:`DEFAULT_JSON_OPTIONS`. .. versionchanged:: 4.0 Now outputs MongoDB Relaxed Extended JSON by default (using :const:`DEFAULT_JSON_OPTIONS`). .. versionchanged:: 3.4 Accepts optional parameter `json_options`. See :class:`JSONOptions`. json_options)poprdjsondumps _json_convert)rerBrCrfr5r5r6ris rir(Union[str, bytes, bytearray]csP|dtjturfdd|d<nfdd|d<tj|g|Ri|S)aHelper function that wraps :func:`json.loads`. Automatically passes the object_hook for BSON type conversion. Raises ``TypeError``, ``ValueError``, ``KeyError``, or :exc:`~bson.errors.InvalidId` on invalid MongoDB Extended JSON. :param json_options: A :class:`JSONOptions` instance used to modify the decoding of MongoDB Extended JSON types. Defaults to :const:`DEFAULT_JSON_OPTIONS`. .. versionchanged:: 4.0 Now loads :class:`datetime.datetime` instances as naive by default. To load timezone aware instances utilize the `json_options` parameter. See :ref:`tz_aware_default_change` for an example. .. versionchanged:: 3.5 Parses Relaxed and Canonical Extended JSON as well as PyMongo's legacy format. Now raises ``TypeError`` or ``ValueError`` when parsing JSON type wrappers with values of the wrong type or any extra keys. .. versionchanged:: 3.4 Accepts optional parameter `json_options`. See :class:`JSONOptions`. rfc t|SN) object_hookrerfr5r6 zloads..rncrlrm)object_pairs_hook)pairsrpr5r6rqrrrs)rgrdrAdictrhloads)r(rBrCr5rpr6rvs  rvrfcsnt|drfdd|DSt|dr%t|ttfs%fdd|DSzt|WSty6|YSw)z]Recursive helper method that converts BSON types so they can be converted into json. itemscsi|] \}}|t|qSr5rj).0kvrpr5r6 sz!_json_convert..__iter__csg|]}t|qSr5rx)ryr{rpr5r6 sz!_json_convert..)hasattrrw isinstancerSbytesdefault TypeErrorrerfr5rpr6rjs   rjrtSequence[Tuple[str, Any]]cCst|||Srm)rnrA)rtrfr5r5r6rs srsdctMapping[str, Any]cCs4d}|D] }|tvr|}nq|rt|||S|Srm) _PARSERS_SET_PARSERS)rrfmatchrzr5r5r6rnsrndocdummy0cCsJ|d}t|ttfs |Sd}|ddD] }|t|dO}qt||S)N$regexr$options)rrSrrP _RE_OPT_TABLEr )rrpatternflagsr]r5r5r6_parse_legacy_regexs rUnion[Binary, uuid.UUID]cCsdt|dkr td|t|dtstd||jtjkr+tt |dSt |dS)z*Decode a JSON legacy $uuid to Python UUID.r-zBad $uuid, extra field(s): $uuidz$uuid must be a string: ) lenrrrSuuid_representationr UNSPECIFIEDr from_uuiduuidUUIDrrfr5r5r6_parse_legacy_uuid)s  rdatasubtyper;cCsn|tvr(|j}t||}|tjkr|S|tkrtj}n|tjkr#tj}||S|dkr2t t j |St||SNr) rrrrrrSTANDARD PYTHON_LEGACYas_uuidr rr)rrrfr binary_valuer5r5r6_binary_or_uuid5s      rcCsht|dtrd|d|d<t|dd}|dkr%t|dddd}t|d}t|||S)N$type%02xl$binary)rr;base64 b64decodeencoder)rrfrrr5r5r6_parse_legacy_binaryJs rcCs|d}|d}|d}t|tstd|t|tr#t|dkr*td|t|dkr7td|t|}t|t|d|S) NrrsubTypez!$binary base64 must be a string: r.z7$binary subType must be a string at most 2 characters: z=$binary must include only "base64" and "subType" components: r) rrSrrrrrrr;)rrfbinaryb64rrr5r5r6_parse_canonical_binaryTs  r$Union[datetime.datetime, DatetimeMS]c Cs|d}t|dkrtd|t|trrSr;rr5r5r6 _encode_int64as rcCs|Srmr5rerr5r5r6 _encode_noophsrcCsd}|jtj@r |d7}|jtj@r|d7}|jtj@r |d7}|jtj@r*|d7}|jtj@r4|d7}|jtj@r>|d7}t|j t rH|j }n|j d}|j t jkrY||d Sd ||d iS) Nrr%r&r'r(r)r*zutf-8)rrr)rr)rre IGNORECASELOCALE MULTILINEDOTALLUNICODEVERBOSErrrSrr<r7r2)rerfrrr5r5r6 _encode_regexls&          r"cCsB|jtjkrt |krtkrnndt|iSdt|iS|S)Nrr)r<r7r9 _INT32_MAXrSrr5r5r6 _encode_ints   r$cCs`|jtjkr.t|rddiSt|r |dkrdnd}d|iS|jtjkr.dtt|iS|S)NrNaNrInfinityz -Infinity) r<r7r2mathisnanisinfr9rSrepr)rerfrepresentationr5r5r6 _encode_floats    r,datetime.datetimecCs|jtjkrN|js|jtd}|jdusJ|tkrN|j|}|j|j |j fdkr-d}n| d}t |j d}|r@d|fnd}dd | d ||iSt|}|jtjkr\d|iSdd t|iiS) Nr)rrrrz%ziz.%03drrz{}{}{}rr)r?r,r4rOrr$r utcoffsetdaysr microsecondsstrftimer;rrTrr2rS)rerfoff tz_stringmillisfracsecsr5r5r6rs"     rcCs t|d|Sr)rrr5r5r6 _encode_bytes r6rcCst||j|Srm)rrrr5r5r6_encode_binary_objr8 uuid.UUIDcCs.|jrtj||jd}t||j|Sd|jiS)N)rr)r@rrrrrhex)rerfbinvalr5r5r6 _encode_uuids r=cC dt|iS)NrrSrr5r5r6_encode_objectidr7r@cCsd|j|jdiS)Nr )r r%)timeincrr5r5r6_encode_timestampsrCcCr>)Nrr?rr5r5r6_encode_decimal128r7rDrcCst||dS)Nrp)rjrrr5r5r6 _encode_dbrefsrEdummy1cCddiS)Nrr-r5rrFr5r5r6_encode_minkeyrIcCrG)Nrr-r5rHr5r5r6_encode_maxkeyrJrKz-dict[Type, Callable[[Any, JSONOptions], Any]] _ENCODERSz,dict[int, Callable[[Any, JSONOptions], Any]]_MARKERS _type_markerccs|]}|VqdSrmr5)ryr r5r5r6 srOcCsz tt|||WStyYnwt|dr/|j}|tvr/t|}|tt|<|||StD]}t||rIt|}|tt|<|||Sq1td|)NrNz%r is not JSON serializable) rLrKeyErrorrrNrM_BUILT_IN_TYPESrr)rerfmarkerfuncbaser5r5r6rs$       rcCst|Srm)rror5r5r6 _get_str_sizerJrUcCsdtt|S)Nr)rrSrAror5r5r6_get_datetime_sizesrVr cCdt|jS)N)rrror5r5r6_get_regex_size#r9rYcCrW)N")r collectionror5r5r6_get_dbref_size'r9r\ zdict[Any, int]_CONSTANT_SIZE_TABLEzdict[Any, Callable[[Any], int]]_VARIABLE_SIZE_TABLEmax_size current_sizecCs(||kr|St|}zt|WStyYnwzt||WSty)Ynw|tkrQ|jrG|dt|j||t|t|j7}|S|dt|7}|S|tkrx| D]\}}|t|||7}|t|||7}||kru|SqY|St |dr|D]}|t|||7}||kr|Sq|S)z!Recursively finds size of objectsrr}) rrarPrbrrget_sizerrurwr)rercrdobj_typerzr{r%r5r5r6re>sH   "  re max_lengthTuple[Any, int]cCs|dkrdS|}t|dr2i}|D]\}}t||\}}|r$|||<|dkr-||fSq||fSt|dr`t|ttfs`g}|D]}t||\}}|rR|||dkr[||fSqB||fSt||S)zMRecursively truncate documents as needed to fit inside max_length characters.rrrwr})rrw_truncate_documentsrrSrappend _truncate)rerg remaining truncatedrzr{ truncated_vr5r5r6rigs4   rirlcCsRt||}||kr|||fSz|d|}Wn ty"|}Ynw|||fSrm)rer)rerlsizermr5r5r6rks    rk)rerrBrrCrrMrS)r(rkrBrrCrrMr)rerrfr:rMr)rtrrfr:rMr)rrrfr:rMr)rrrrrMr)rrrfr:rMr)rrrr;rfr:rMr)rrrfr:rMr)rrrrrMr)rrrrrMrS)rrrrrMr)rrrrrMr)rrrrrMr;)rrrrrMr)rrrrrMr)rrrrrMr)rrrrrMr)rrrrrMr)rrrrrMr#)rrrr;rfr:rMr)rerrfr:rMru)rerrfr:rMru)rerrfr:rMr)rerrrrMr)rer;rfr:rMr)rerrfr:rMr)rer-rfr:rMru)rerrfr:rMru)rerrfr:rMru)rer:rfr:rMru)rerrrrMru)rer#rrrMru)rerrfr:rMru)rrrFrrMru)rerrMr;)rer-rMr;)rer rMr;)rerrMr;)r)rerrcr;rdr;rMr;)rerrgr;rMrh)rerrlr;rMrh)__doc__ __future__rrrrhr'rrtypingrrrrrrr r r r r bson.binaryrrrr bson.coderbson.codec_optionsrrbson.datetime_msrrrrr bson.dbrefrbson.decimal128r bson.int64r bson.max_keyr bson.min_keyr bson.objectidr bson.regexr bson.sonr!bson.timestampr# bson.tz_utilr$ILMSUXrr,r7rS _BASE_CLASSr#r:r2rar_r9rbr8rcrdrirvrjrsrnrrrrrrrrrrrrrrrrrr r rrsetrrrrrrr"r$r,rr6r8r=r@rCrDrErIrKr=rrr;rrrLrM_typrrNtuplerQrrUrVrYr\rarbrerirkr5r5r5r6s W4            $+1  "    E                                           )