E6i^ddlZddlZddlmZddlmZddlmZeje Z ejdZ GddeZ Gd d e ZGd d eZdd Zy)N) OrderedDict)isclass) RestErrorz\{|\}|\,|[\w_:\-\*]+ceZdZdZy) MaskErrorz#Raised when an error occurs on maskN__name__ __module__ __qualname____doc__S/home/azureuser/techstart-app/venv/lib/python3.12/site-packages/flask_restx/mask.pyrr-rrceZdZdZy) ParseErrorz#Raised when the mask parsing failedNr rrrrrrrrcBeZdZdZdfd ZdZdZdZdZdZ xZ S) Maskz Hold a parsed mask. :param str|dict|Mask mask: A mask, parsed or not :param bool skip: If ``True``, missing fields won't appear in result c ||_t|tr$tt||j |yt|ttfrtt||fi|y||_tt|di|y)Nr) skip isinstancestrsuperr__init__parsedictr)selfmaskrkwargs __class__s rrz Mask.__init__"si dC $ & ( JJt  t[1 2 $ &t 6v 6DI $ & 0 0rc|sy|j|}|}d}g}tj|D]}|dk(r?||vr tdt |j ||<|j |||}n<|dk(r|s td|j}n|dk(r|dvrtd d ||<|}|r td y) a Parse a fields mask. Expect something in the form:: {field,nested{nested_field,another},last} External brackets are optionals so it can also be written:: field,nested{nested_field,another},last All extras characters will be ignored. :param str mask: the mask string to parse :raises ParseError: when a mask is unparseable/invalid N{zUnexpected opening bracket)r}zUnexpected closing bracket,)r%r#NzUnexpected commaTMissing closing bracket)cleanLEXERfindallrrrappendpop)rrfieldspreviousstacktokens rrz Mask.parse-s" zz$]]4( E|6)$%ABB#'TYY#7x  V$)#$%ABB#//$%788 $u H# & 67 7 rc|jddj}|ddk(r|ddk7r td|dd}|S) zRemove unnecessary characters rr#r$r&r)replacestripr)rrs rr'z Mask.clean\sM||D"%++- 7c>Bx3 !:;;":D rcddlm}t|ttt fr|Dcgc]}|j |c}St||j|j|jfr|j|St||jk(r(|j|j|j|S||jk(r|j|St||js!t|r!t!||jr t#dt|t$t&fst)|dr |j*}|j-|Scc}w)z Apply a fields mask to the data. :param data: The data or model to apply mask on :raises MaskError: when unable to apply the mask r)r,)default attributer)rzMask is inconsistent with model__dict__)r2r,rlisttuplesetapplyNestedList PolymorphclonetypeRawr7r8r issubclassrrrhasattrr9 filter_data)rdatar,ds rr=z Mask.applyfs  dT5#. /+/0aDJJqM0 0 v}}fkk6;K;KL M::d# # $Z6:: %::dlldnnSW:X X VZZ ::4:( ( tVZZ (t}4,=> >D4"5674;T==D%%%1sE!ci}|jD]\}}|dk(r t|tr>|j|d}|jr|=|d||<E|j |||<Z|jr||vrk|j|d||<d|j vr"|jD]\}}||vs |||<|S)z Handle the data filtering given a parsed mask :param dict data: the raw data to filter :param list mask: a parsed mask to filter against :param bool skip: whether or not to skip missing fields *N)itemsrrgetrr=keys)rrGoutfieldcontentnestedkeyvalues rrFzMask.filter_datas"jjl 3NE7|GT*%.99^!%CJ!(v!6CJuD0!XXeT2E  3 $))+ "jjl % Uc>$CH % rcdjdj|jDcgc]3\}}t|trdj|t |fn|5c}}Scc}}w)Nz{{{0}}}r%r2)formatjoinrKrrr)rkvs r__str__z Mask.__str__sg HH!% 1-7q$,?BGGQAK(QF   s8A.)NF) r r r r rrr'r=rFrY __classcell__)r!s@rrrs) 1-8^&@@ rrc8t||j|S)a Apply a fields mask to the data. :param data: The data or model to apply mask on :param str|Mask mask: the mask (parsed or not) to apply on data :param bool skip: If rue, missing field won't appear in result :raises MaskError: when unable to apply the mask )rr=)rGrrs rr=r=s d  ! !$ ''r)F)loggingre collectionsrinspectrerrorsr getLoggerr logcompiler(rrrr=rrrrdsc #g! *+    T ;T n (r