o i_J @sddlZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZmZmZmZddlmZgdZejZGdd d ZeZ Gd d d Zd ed efddZde eeeefd efddZGdddeZde d efddZde d efddZ ddde ded e fdd Z!d!e d"ed efd#d$Z"de d eefd%d&Z#de d eefd'd(Z$ddde ded eee ffd)d*Z%d+e d,ed efd-d.Z&dS)/N)partial) JSONEncoder) DictListOptionalUnioncastAnyTypeCallableTuple)datetimedatetime timedelta)timezone)NULLAzureJSONEncoderis_generated_modelas_attribute_dictattribute_listTypeHandlerRegistryget_backcompat_attr_namec@seZdZdZdefddZdS)_NullzTo create a Falsy objectreturncCsdS)NFselfrrO/home/app/Keep/.python/lib/python3.10/site-packages/azure/core/serialization.py__bool__sz_Null.__bool__N)__name__ __module__ __qualname____doc__boolrrrrrrsrc @seZdZdZdddZdeeeege ffdeeege e effgeege e effffddZ deeeege ffdeeee e efgefgeee e efgefffd d Z d edeeege e efffd d Zdedeee e efgeffddZdS)rzUA registry for custom serializers and deserializers for specific types or conditions.rNcCs(i|_i|_g|_g|_i|_i|_dSN)_serializer_types_deserializer_types_serializer_predicates_deserializer_predicates_serializer_cache_deserializer_cacherrrr__init__-s  zTypeHandlerRegistry.__init__ conditioncs@dttgtttffdttgtttffffdd }|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_funcrcFttr |j<ntrj|fntdj|SNz:Condition must be a type or a callable predicate function.) isinstancetyper&callabler(append TypeErrorr*clearr.r-rrr decoratorW   z:TypeHandlerRegistry.register_serializer..decorator)r r rstrrr-r9rr8rregister_serializer6s 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. r.rcr/r0) r1r2r'r3r)r4r5r+r6r7r8rrr9r:z.decorator)r r rr;r r<rr8rregister_deserializerds@! z)TypeHandlerRegistry.register_deserializerobjcCs^t|}||jvr|j|S|jt|}|s(|jD] \}}||r'|}nq||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]]] )r2r*r&getr()rr?obj_typehandler predicate pred_handlerrrrget_serializers   z"TypeHandlerRegistry.get_serializerclscCsf||jvr |j|S|j|}|s"|jD] \}}||r!|}nq|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)r+r'r@r)r)rrFrBrCrDrrrget_deserializers    z$TypeHandlerRegistry.get_deserializer)rN)r r!r"r#r,rr r r r$rr;r=r>rrErGrrrrr*s  0 .4 (.,rtdrc Cs|}t|d\}}t|d\}}t|d\}}ttt|||f\}}}t|d}d}|r3d|}d}|p8|}|rB|d|7}|pE|}|rO|d|7}z|r\d t|}n d |}|d }Wnt ytd |}Ynw|d |7}d ||S)auConverts a datetime.timedelta object into an ISO 8601 formatted string, e.g. 'P4DT12H30M05S' Function adapted from the Tin Can Python project: https://github.com/RusticiSoftware/TinCanPython :param td: The timedelta object to convert :type td: datetime.timedelta :return: An ISO 8601 formatted string representing the timedelta object :rtype: str <z%sDTz{:02}Hz{:02}Mz{:02}z%09.6f0z{}SP) total_secondsdivmodlistmapintroundformat is_integerrstripAttributeError) rHsecondsminuteshoursdaysZdate_strZtime_strZ bigger_existsZseconds_stringrrr_timedelta_as_isostrs6     r^dtcCst|dr(t|dr(tt|}|js|jtd}n|t}|ddSzttt t f|}|WSt yHtt |}t |YSw)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)hasattrrr rbreplaceTZ_UTC isoformat astimezonerrrrYrr^)r_Z iso_formattedrrr_datetime_as_isostrs      rics*eZdZdZdedeffdd ZZS)rzHA JSON encoder that's capable of serializing datetime objects and bytes.orcsJt|ttfrt|Szt|WStyYnwtt | |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 ) r1bytes bytearraybase64 b64encodedecoderirYsuperrdefault)rrj __class__rrrqs  zAzureJSONEncoder.default)r r!r"r#r rq __classcell__rrrrrrsrr?cCstt|ddp 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 Z _is_modelF_attribute_map)r$getattrrd)r?rrrr/srpcCs$z|jdgkWStyYdSw)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)Z _visibilityrY)rwrrr _is_readonly:s  ryFexclude_readonlyvr{csx|dus t|tr dSt|tttfr t|fdd|DSt|tr0fdd|DSt|r:t |dS|S)Nc3s|] }t|dVqdS)rzN_as_attribute_dict_value).0xrzrr Kz+_as_attribute_dict_value..csi|] \}}|t|dqS)rzr})rZdkZdvrzrr Msz,_as_attribute_dict_value..rz) r1rrRtuplesetr2dictitemsrr)r|r{rrzrr~Gs r~ rest_fielddefault_attr_namecCs$z|jp|WSty|YSw)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 )Z_original_tsp_namerY)rrrrr_get_backcompat_nameQs   rc Csd}zt|tddt|Dd}Wn tyYdSw|dur$dS|jD]!\}}zt|jj t|r@|WSWq)t yJYq)wdS)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] Ncss|] }d|vr|VqdS)Z__flattened_itemsNr)rarrrrhrz+_get_flattened_attribute..) rvnextdir StopIteration_attr_to_rest_fieldrr _class_typekeys intersectionrY)r?Zflattened_itemskr|rrr_get_flattened_attribute_s""   rcCspt|stdt|drt|jSt|}g}|jD]\}}||kr0| t |j q| |q|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.ru) rr5rdrRrurrrrextendrrr4)r?flattened_attributeretval attr_namerrrrrys  rc svt|stdt|dr|j| dSzi}t}i}t|}|jD]*\}}|r5t|r5| |j ||krJ|j jD] \}} ||| j <q?q%|||j <q%|D]R\} |r_|vr_qT|kry| D]\}} t | |d|| ||<qgqTd} ztfdd|jDj} Wn tyYnw| r| nt | |d|| <qT|WSty} ztd| d} ~ ww) 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.ru)Z keep_readonlyrzFc3s|] }|jkr|VqdSr%) _rest_name)rrfrrrrsz$as_attribute_dict..N)rr5rdas_dictrrrrryaddrrr~r@rvaluesZ_is_multipart_file_inputrrY) r?r{resultZreadonly_propsZ rest_to_attrrrrfkfvr|Zis_multipart_file_inputexcrrrrsR         rmodelrcCs~t|stdt|}|jD]+\}}||kr t||S||kr<|jjD]\}}||kr;t||Sq*q|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)rr5rrrrr)rrrZfield_attr_namerrrrrrrs r)'rm functoolsrjsonrtypingrrrrrr r r r r rrrr__all__utcrfrrrr;r^rirr$rryr~rrrrrrrrrs0  ,   7   $=