\j%ddlZddlZddlmZmZmZddlmZddlm Z m Z m Z m Z m Z GddeZGddeej Zy) N)ListOptionalTuple) serializers) EXPAND_PARAM FIELDS_PARAM OMIT_PARAMWILDCARD_EXPAND_VALUES split_levelsc leZdZdZiZfdZfdZfdZdZdZ de fdZ d e d e d e e e e e ffd Zd ee dee dee d ee fdZde d ee dee dee d ef dZdee d ee dee dee d ee f dZed efdZde d ee fdZde d ee fdZdee d efdZxZS)FlexFieldsSerializerMixinz A ModelSerializer that takes additional arguments for "fields", "omit" and "expand" in order to control which fields are displayed, and whether to replace simple values with complex, nested serializations ct|jtg}t|jtg}t|jtg}|jdd}t t ||i|||_g|_ d|_ |||d|_ |s|jtng|s|jtng|s|jtngd|_|jd|jdz|jd|jdz|jd|jdzd|_y)NparentF)expandfieldsomitrrr)listpoprrr superr __init__rexpanded_fields_flex_fields_rep_applied_flex_options_base'_get_permitted_expands_from_query_param_get_query_param_value_flex_options_rep_only_flex_options_all)selfargskwargsrrrr __class__s F/root/env/lib/python3.12/site-packages/rest_flex_fields/serializers.pyrz"FlexFieldsSerializerMixin.__init__sJfjjr23fjjr23FJJz2./Hd+ '7HH !(-%# %CCLQ$22<@ 00< ' #--h7$:U:UV^:__--h7$:U:UV^:__++F3d6Q6QRX6YY" c|js-|j|j|jd|_t||S)NT)rapply_flex_fieldsrrrto_representation)rinstancer!s r"r&z+FlexFieldsSerializerMixin.to_representation<s>,,  " "4;;0K0K L,0D )w(22r#c\t|}|j||j|S)N)r get_fieldsr%r)rrr!s r"r)z$FlexFieldsSerializerMixin.get_fieldsBs+#% vt'>'>? r#c^t|d\}}t|d\}}t|d\}}|j||||} | D]} |j| |j||||} | D]4} |jj | |j | ||||| <6|S)Nrrr)r _get_fields_names_to_remover_get_expanded_field_namesrappend_make_expanded_field_serializer) rr flex_options expand_fieldsnext_expand_fields sparse_fieldsnext_sparse_fields omit_fieldsnext_omit_fields to_remove field_nameexpanded_field_namesnames r"r%z+FlexFieldsSerializerMixin.apply_flex_fieldsGs,8h9O,P) ),8h9O,P) )(4\&5I(J% %44[-Qacij # #J JJz " # $== ; 7G ) D  ' ' -??(*<>NF4L   r#c|j|}t|tr.|d}t|dkDrt j |dni}n|}i}t |tk(r|j|}t|tjr|j|d<t|tr5||d<||vr |||t<||vr |||t<||vr |||t <|di|S)zS Returns an instance of the dynamically created nested serializer. rcontextr)_expandable_fields isinstancetuplelencopydeepcopytypestr&_get_serializer_class_from_lazy_string issubclassr Serializerr<r rrr )rr9 nested_expand nested_fields nested_omit field_optionsserializer_classsettingss r"r.z9FlexFieldsSerializerMixin._make_expanded_field_serializer]s//5 mU +,Q/ :=m:Lq:Pt}}]1%56VXH, H  !S (#JJK[\  & (>(> ?"&,,HY  &(A B!%HX }$)6t)<&}$)6t)<&{"'24'8$+(++r#full_lazy_pathc|jd}|j}dj|}|j||\}}|r)|j ds|j|dz|\}}|r|St |)N.z .serializers)splitrjoin_import_serializer_classendswith Exception)rrO path_parts class_namepathrMerrors r"rFz@FlexFieldsSerializerMixin._get_serializer_class_from_lazy_string~s#))#. ^^% xx #"&"?"?j"Q% ~6&*&C&C~%z' # e # #r#rYrXreturnc tj|} t||dfS#t$r dd|d|fcYSwxYw#t$r dd|d|fcYSwxYw)NzNo module found at path: z when trying to import z No class z class found in module ) importlib import_module ImportErrorgetattrAttributeError)rrYrXmodules r"rTz2FlexFieldsSerializerMixin._import_serializer_classsx ,,T2F U6:.4 4 JNPZ[   U4TT T Us& ?<<AAr4r2next_level_omitsct|dkD}g}|st|dk(r|S|D]*}|j||||}|r|j|,|S)z Remove fields that are found in omit list, and if sparse names are passed, remove any fields not found in that list. r)rA_should_field_existr-) rr4r2rcrsparser6r7 should_exists r"r+z5FlexFieldsSerializerMixin._get_fields_names_to_removesq]#a' #k*a/   -J33K8HL   ,  -r#r7c<||vr||vryt|dkDr||vryy)a  Next level omits take form of: { 'this_level_field': [field_to_omit_at_next_level] } We don't want to prematurely omit a field, eg "omit=house.rooms.kitchen" should not omit the entire house or all the rooms, just the kitchen. FrT)rA)rr7r4r2rcs r"rez-FlexFieldsSerializerMixin._should_field_exists1  $;K)K }  !j &Er#r0ct|dk(rgS|j|r|jj}g}|D]7}||jvr|j ||||s'|j |9|S)Nr)rA_contains_wildcard_expand_valuer>keysrer-)rr0r4r2rcaccumr9s r"r,z3FlexFieldsSerializerMixin._get_expanded_field_namess }  "I  / / > 3388:M! D4222++D+}N^_ LL   r#ct|dr,t|jdr|jjS|jS)zIt's more consistent with DRF to declare the expandable fields on the Meta class, however we need to support both places for legacy reasons.Metaexpandable_fields)hasattrrnro)rs r"r>z,FlexFieldsSerializerMixin._expandable_fieldss8 4 WTYY8K%L99.. .%%%r#fieldc|jrgSt|dr|jjdsgS|jdjj |}|s7|jdjj dj |}|r"t|dk(r|djdS|xsgS)zS Only allowed to examine query params if it's the root serializer. r<requestz{}[]r;r,) rrpr<get query_paramsgetlistformatrArR)rrqvaluess r"rz0FlexFieldsSerializerMixin._get_query_param_values ;;ItY't||/?/? /JIi(55==eD\\),99AA&--PUBVWF c&kQ&!9??3' '|r# expand_paramc|j|}d|jvrB|jd}|j|r|Stt |t |zS|S)z If a list of permitted_expands has been passed to context, make sure that the "expand" fields from the query params comply. permitted_expands)rr<rjrset)rrzrr|s r"rzAFlexFieldsSerializerMixin._get_permitted_expands_from_query_paramsc ,,\: $,, . $ -@ A 33F;((CK#.?*@@AA r# expand_valuescttytt|ttz}t|dkDS)NFr)r rr}rA)rr~intersecting_valuess r"rjz9FlexFieldsSerializerMixin._contains_wildcard_expand_values7 ! )"3}#5rrrj __classcell__)r!s@r"r r s  D3  ,,BS UU%(U x}hsm+ ,U #YCys)  c 6#YCy  s)   .Cy#YCy  s)  c 4&D&&CDI(CDQTI$,T#Y,4,r#r c eZdZy)FlexFieldsModelSerializerN)rrrr=r#r"rr!sr#r)rBr]typingrrrrest_frameworkrrest_flex_fieldsrrr r r objectr ModelSerializerrr=r#r"rsB ((&N,N,b 9;;V;V r#