6iD 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 d eefdZ"de d eefdZ#ddde ded eee ffdZ$y)N)partial) JSONEncoder) DictListOptionalUnioncastAnyTypeCallableTuple)datetimedatetime timedelta)timezone)NULLAzureJSONEncoderis_generated_modelas_attribute_dictattribute_listTypeHandlerRegistryceZdZdZdefdZy)_NullzTo create a Falsy objectreturncy)NFselfs [/home/azureuser/techstart-app/venv/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 TypeErrorr/clearr4r2rs r decoratorz:TypeHandlerRegistry.register_serializer..decoratorUsb)T*4@&&y1)$++22I|3LM \]]  " " ( ( * r")r r rstrrr2r>s`` r register_serializerz'TypeHandlerRegistry.register_serializer4sNB 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. r4rcttr|j<n4trjj |fn t djj|Sr6) r7r8r,r9r.r:r;r0r<r=s r r>z.decoratorsb)T*6B((3)$--44i5NO \]]  $ $ * * , r")r r rr?r r@s`` r register_deserializerz)TypeHandlerRegistry.register_deserializerbsTB 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]]] )r8r/r+getr-)rrEobj_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)r0r,rGr.r)rrMrIrJrKs r get_deserializerz$TypeHandlerRegistry.get_deserializers $** *++C0 0**..s3+/+H+H ' <S>*G  BI(=d  %'',,r")rN)r#r$r%r&r1rr r r r'rr?rArDrrLrOrr"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 rrpreplaceTZ_UTC isoformat astimezonerrrrbrrk)rl iso_formatteds r _datetime_as_isostrrxsr6wr62 (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 ) r7bytes bytearraybase64 b64encodedecoderxrbsuperrdefault)rrz __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"rrEcJtt|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'getattrrr)rEs 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) _visibilityrb)rs r _is_readonlyr8s*}}(( 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..IsaZ[/DTUUasr) r7rr[tuplesetr8dictitemsrrr)rrdkdvs ` r rrEsyJq%(!dE3'(tAwa_`aaa!TbcbibibklX^XZ\^,RBRSSllFXYZF[ Q1A Bbabbms-B#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..Xs+\!CVZ[C[A+\s ) rnextdir StopIteration_attr_to_rest_fieldrr _class_typekeys intersectionrb)rEflattened_itemskrs r _get_flattened_attributerOsO!#t+\s3x+\'\^bc''--/1 1==4499;<IIO$     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;rrr[rrrrrextendrrr:)rEflattened_attributeretval attr_name rest_fields r rrks 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 z*Object must be a generated model instance.r) keep_readonlyrFc3@K|]}|jk(r|ywr*) _rest_name)rrfrs r rz$as_attribute_dict..s%3==A-3sN)rr;rras_dictrrrrraddrrrrGrvalues_is_multipart_file_inputrrb)rErresultreadonly_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"%"9"9"@"@"B3//. ,1A6Nqcs6t|''1-.# ( % ODE3NOs<DF6 6F'?'F6' F30F62F33F66 G? G  G)%r~ functoolsrjsonrtypingrrrrr r r r r rrrrr__all__utcrtrrrr?rkrxrr'rrrrrrrr"r rsCPPP44   w \-\-~4%Y4%34%n(E(D$ "AB(s(88{8&TCTDT C D BGcc$c3c#(3-8S .=B<O3<OT<Od3PS8n<Or"