]j_J ddlZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZmZmZmZddlmZgdZej*ZGdd ZeZ Gd d Zd ed efdZde eeeefd efdZGddeZde d efdZde d efdZ ddde ded e fdZ!de ded efdZ"de d eefdZ#de d eefd Z$ddde ded eee ffd!Z%d"e d#ed efd$Z&y)%N)partial) JSONEncoder) DictListOptionalUnioncastAnyTypeCallableTuple)datetimedatetime timedelta)timezone)NULLAzureJSONEncoderis_generated_modelas_attribute_dictattribute_listTypeHandlerRegistryget_backcompat_attr_nameceZdZdZdefdZy)_NullzTo create a Falsy objectreturncy)NFselfs B/root/env/lib/python3.12/site-packages/azure/core/serialization.py__bool__z_Null.__bool__sN)__name__ __module__ __qualname____doc__boolr"rr#r!rrs"$r#rc 0eZdZdZd dZdeeeege ffdeeege e effgeege e effffdZ deeeege ffdeeee e efgefgeee e efgefffdZ dedeeege e efffd Zd edeee e efgeffd Zy) rzUA registry for custom serializers and deserializers for specific types or conditions.rNcXi|_i|_g|_g|_i|_i|_yN)_serializer_types_deserializer_types_serializer_predicates_deserializer_predicates_serializer_cache_deserializer_cachers r!__init__zTypeHandlerRegistry.__init__-s2799; TV#VX%ACCE r# conditioncdttgtttffdttgtttffffd }|S)aDecorator to register a serializer. The handler function is expected to take a single argument, the object to serialize, and return a dictionary representation of that object. Examples: .. code-block:: python @registry.register_serializer(CustomModel) def serialize_single_type(value: CustomModel) -> dict: return value.to_dict() @registry.register_serializer(lambda x: isinstance(x, BaseModel)) def serialize_with_condition(value: BaseModel) -> dict: return value.to_dict() # Called manually for a specific type def custom_serializer(value: CustomModel) -> Dict[str, Any]: return {"custom": value.custom} registry.register_serializer(CustomModel)(custom_serializer) :param condition: A type or a callable predicate function that takes an object and returns a bool. :type condition: Union[Type, Callable[[Any], bool]] :return: A decorator that registers the handler function. :rtype: Callable[[Callable[[Any], Dict[str, Any]]], Callable[[Any], Dict[str, Any]]] :raises TypeError: If the condition is neither a type nor a callable. handler_funcrcttr|j<n4trjj |fn t djj|SNz:Condition must be a type or a callable predicate function.) isinstancetyper,callabler.append TypeErrorr0clearr5r3r s r! decoratorz:TypeHandlerRegistry.register_serializer..decoratorWsb)T*4@&&y1)$++22I|3LM \]]  " " ( ( * r#)r r rstrr r3r?s`` r!register_serializerz'TypeHandlerRegistry.register_serializer6sNB HcUDcN-B$C RUQVX\]`be]eXfQfHg r#cdtttttfgtfdtttttfgtfffd }|S)a<Decorator to register a deserializer. The handler function is expected to take two arguments: the target type and the data dictionary, and return an instance of the target type. Examples: .. code-block:: python @registry.register_deserializer(CustomModel) def deserialize_single_type(cls: Type[CustomModel], data: dict) -> CustomModel: return cls(**data) @registry.register_deserializer(lambda t: issubclass(t, BaseModel)) def deserialize_with_condition(cls: Type[BaseModel], data: dict) -> BaseModel: return cls(**data) # Called manually for a specific type def custom_deserializer(cls: Type[CustomModel], data: Dict[str, Any]) -> CustomModel: return cls(custom=data["custom"]) registry.register_deserializer(CustomModel)(custom_deserializer) :param condition: A type or a callable predicate function that takes an object and returns a bool. :type condition: Union[Type, Callable[[Any], bool]] :return: A decorator that registers the handler function. :rtype: Callable[[Callable[[Type, Dict[str, Any]], Any]], Callable[[Type, Dict[str, Any]], Any]] :raises TypeError: If the condition is neither a type nor a callable. r5rcttr|j<n4trjj |fn t djj|Sr7) r8r9r-r:r/r;r<r1r=r>s r!r?z.decoratorsb)T*6B((3)$--44i5NO \]]  $ $ * * , r#)r r rr@r rAs`` r!register_deserializerz)TypeHandlerRegistry.register_deserializerdsTB HdDcN-CS-H$I hX\^bcfhkck^lWmorWrNs r#objct|}||jvr|j|S|jjt|}|s |jD]\}}||s|}n||j|<|S)aGets the appropriate serializer for an object. It first checks the type dictionary for a direct type match. If no match is found, it iterates through the predicate list to find a match. Results of the lookup are cached for performance based on the object's type. :param obj: The object to serialize. :type obj: any :return: The serializer function if found, otherwise None. :rtype: Optional[Callable[[Any], Dict[str, Any]]] )r9r0r,getr.)r rFobj_typehandler predicate pred_handlers r!get_serializerz"TypeHandlerRegistry.get_serializers9 t-- -))(3 3((,,T#Y7+/+F+F ' <S>*G  ,3x(r#clsc||jvr|j|S|jj|}|s |jD]\}}||s|}n|r t ||nd|j|<|j|S)aXGets the appropriate deserializer for a class. It first checks the type dictionary for a direct type match. If no match is found, it iterates through the predicate list to find a match. Results of the lookup are cached for performance based on the class. :param cls: The class to deserialize. :type cls: type :return: A deserializer function bound to the specified class that takes a dictionary and returns an instance of that class, or None if no deserializer is found. :rtype: Optional[Callable[[Dict[str, Any]], Any]] N)r1r-rHr/r)r rNrJrKrLs r!get_deserializerz$TypeHandlerRegistry.get_deserializers $** *++C0 0**..s3+/+H+H ' <S>*G  BI(=d  %'',,r#)rN)r$r%r&r'r2rr r r r(rr@rBrErrMrPrr#r!rr*s/_F,tXseTk%::;, 8SE4S>123XseT#s(^>S5TT U,\,tXseTk%::;, 8T4S>2C7898T4PSUXPX>DZ\_D_;`` a,\#(8SE4S>H $$ 1 01s.AD D('D(dtct|drt|drvtt|}|js%|j t j }n#|jt j }|j ddS ttttf|}|j S#t$rtt|}t|cYSwxYw)aCConverts a datetime.(datetime|date|time|timedelta) object into an ISO 8601 formatted string. :param dt: The datetime object to convert :type dt: datetime.datetime or datetime.date or datetime.time or datetime.timedelta :return: An ISO 8601 formatted string representing the datetime object :rtype: str yearhour)tzinfoz+00:00Z)hasattrr rrqreplaceTZ_UTC isoformat astimezonerrrrcrrl)rm iso_formatteds r!_datetime_as_isostrrysr6wr62 (B yyJJfJ5??AMMM&1;;=M$$Xs33( %d #R (||~ ( )R #B''(s,B==$C$#C$c,eZdZdZdedeffd ZxZS)rzHA JSON encoder that's capable of serializing datetime objects and bytes.orct|ttfr#tj|j S t |S#t$rYnwxYwtt|+|S)zOverride the default method to handle datetime and bytes serialization. :param o: The object to serialize. :type o: any :return: A JSON-serializable representation of the object. :rtype: any ) r8bytes bytearraybase64 b64encodedecoderyrcsuperrdefault)r r{ __class__s r!rzAzureJSONEncoder.defaultsc a%+ ,##A&--/ / &q) )   %t4Q77s A AA)r$r%r&r'r r __classcell__)rs@r!rrsR 8 8 8 8r#rrFcJtt|ddxs t|dS)zCheck if the object is a generated SDK model. :param obj: The object to check. :type obj: any :return: True if the object is a generated SDK model, False otherwise. :rtype: bool _is_modelF_attribute_map)r(getattrrs)rFs r!rr/s% [%0RGCAQ4R SSr#pcB |jdgk(S#t$rYywxYw)zCheck if an attribute is readonly. :param any p: The property to check. :return: True if the property is readonly, False otherwise. :rtype: bool readF) _visibilityrc)rs r! _is_readonlyr:s*}}(( s  Fexclude_readonlyvrc R|t|tryt|tttfrt |fd|DSt|t r/|jDcic]\}}|t|c}}St|r t|S|Scc}}w)Nc38K|]}t|yw)rN)_as_attribute_dict_value).0xrs r! z+_as_attribute_dict_value..KsaZ[/DTUUasr) r8rr\tuplesetr9dictitemsrrr)rrdkdvs ` r!rrGsyJq%(!dE3'(tAwa_`aaa!TbcbibibklX^XZ\^,RBRSSllFXYZF[ Q1A Bbabbms-B# rest_fielddefault_attr_namecF |jxs|S#t$r|cYSwxYw)aGet the backcompat name for an attribute. :param any rest_field: The rest field to get the backcompat name from. :param str default_attr_name: The default attribute name to use if no backcompat name :return: The backcompat name. :rtype: str )_original_tsp_namerc)rrs r!_get_backcompat_namerQs.!,,A0AA !  !s   c pd} t|tdt|Dd}|y|jj D]O\}} t |jjjjt |r|cSQy#t$rYywxYw#t$rYmwxYw)zGet the name of the flattened attribute in a generated TypeSpec model if one exists. :param any obj: The object to check. :return: The name of the flattened attribute if it exists, otherwise None. :rtype: Optional[str] Nc3*K|] }d|vs| yw)__flattened_itemsNr)ras r!rz+_get_flattened_attribute..hs+\!CVZ[C[A+\s ) rnextdir StopIteration_attr_to_rest_fieldrr _class_typekeys intersectionrc)rFflattened_itemskrs r!_get_flattened_attributer_sO!#t+\s3x+\'\^bc''--/1 1==4499;<II#oJ^_`     s$&BAB) B&%B&) B54B5cdt|s tdt|dr#t|jj St |}g}|jjD]@\}}||k(r%|jt|j0|j|B|S)zGet a list of attribute names for a generated SDK model. :param obj: The object to get attributes from. :type obj: any :return: A list of attribute names. :rtype: List[str] z$Object is not a generated SDK model.r) rr<rsr\rrrrrextendrrr;)rFflattened_attributeretval attr_namers r!rrys c ">??s$%C&&++-..237F!$!8!8!>!>!@% : ) + MM.)?)?@ A MM) $ % Mr#c t|s tdt|dr|j| S i}t }i}t |}|j jD]}\}}|r&t|r|j|j||k(r<|jj jD]\}} ||| j<o|||j<|jD]\ } |r |vr |k(r9| jD]%\}} t| |||j||<'Kd} t fd|j jDj } | r| n t| |||j <|S#t"$rY3wxYw#t$$r} td| d} ~ wwxYw)a Convert an object to a dictionary of its attributes. Made solely for backcompatibility with the legacy `.as_dict()` on msrest models. .. deprecated::1.35.0 This function is added for backcompat purposes only. :param any obj: The object to convert to a dictionary :keyword bool exclude_readonly: Whether to exclude readonly properties :return: A dictionary containing the object's attributes :rtype: dict[str, any] :raises TypeError: If the object is not a generated model instance *Object must be a generated model instance.r) keep_readonlyrFc3BK|]}|jk(s|ywr+) _rest_name)rrfrs r!rz$as_attribute_dict..s"3!Z[I[3sN)rr<rsas_dictrrrrraddrrrrHrvalues_is_multipart_file_inputrrc)rFrresultreadonly_props rest_to_attrrrrfkfvris_multipart_file_inputexcrs @r!rrs c "DEEs$%{{-=)={>>'O 6s;%(%<%<%B%B%D @ !IzL$<"":#8#89"i/(44HHNNP5FB24L/57@ Z223 @IIK DAqA$7''ggiwFB7OPReu7vF<++B34w+0'.23%(%<%<%C%C%E3/..,1A6Nqcs6t|''1-. $ % ODE3NOs<DF6 6F'?'F6' F30F62F33F66 G? G  Gmodelrc@t|s tdt|}|jj D]_\}}||k(rt ||cS||k(s|j jj D]\}}||k(s t ||ccSa|S)aGet the backcompat attribute name for a given attribute. This function takes an attribute name and returns the backcompat name (original TSP name) if one exists, otherwise returns the attribute name itself. :param any model: The model instance. :param str attr_name: The attribute name to get the backcompat name for. :return: The backcompat attribute name (original TSP name) or the attribute name itself. :rtype: str r)rr<rrrrr)rrrfield_attr_namerrrs r!rrs e $DEE359','@'@'F'F'H 8# i '' I> > / 1$00DDJJL 8B?/B77 8 8 r#)'r functoolsrjsonrtypingrrrrr r r r r rrrrr__all__utcrurrrr@rlryrr(rrrrrrrrrr#r!rsxPPP44   w \-\-~4%Y4%34%n(E(D$ "AB(s(88{8&TCTDT C D BGcc$c3c !S !S !S !#(3-4S .=B:O3:OT:Od3PS8n:OzCCCr#