o !i^@sddlZddlZddlmZddlmZddlmZee Z e dZ GdddeZ Gd d d e ZGd d d eZdddZdS)N) OrderedDict)isclass) RestErrorz\{|\}|\,|[\w_:\-\*]+c@eZdZdZdS) MaskErrorz#Raised when an error occurs on maskN__name__ __module__ __qualname____doc__r r U/var/www/html/crm_dreinet/venv_linux/lib/python3.10/site-packages/flask_restx/mask.pyrrc@r) ParseErrorz#Raised when the mask parsing failedNrr r r rrrrcsJeZdZdZdfdd ZddZdd Zd d Zd d ZddZ Z 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 NFc sv||_t|trtt|||dSt|ttfr+tt|j|fi|dS||_tt|jdi|dS)Nr ) skip isinstancestrsuperr__init__parsedictr)selfmaskrkwargs __class__r rr"s z Mask.__init__cCs|sdS||}|}d}g}t|D]B}|dkr4||vr"tdt|jd||<||||}n |dkrC|s>td|}n|dkrP|dvrOtd nd ||<|}q|r]td dS) 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 rNzUnexpected commaTMissing closing bracket)cleanLEXERfindallrrrappendpop)rrfieldspreviousstacktokenr r rr-s4    z Mask.parsecCs@|dd}|ddkr|ddkrtd|dd}|S) zRemove unnecessary characters rrrr!r)replacestripr)rrr r rr"\s    z Mask.cleancsddlm}t|tttfrfdd|DSt||j|j|jfr'| St ||j kr8|j |j |j dS||j krC|j dSt||j sSt|rWt||j rWtdt|ttfsft|drf|j}|S) 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'csg|]}|qSr )apply).0drr r rszMask.apply..)default attributer)rzMask is inconsistent with model__dict__)r,r'rlisttuplesetNestedList PolymorphclonetypeRawr5r6r issubclassrrrhasattrr7 filter_data)rdatar'r r3rr0fs&       z Mask.applycCsi}|D]=\}}|dkrqt|tr3||d}|jr"|dur"q|dur+d||<q||||<q|jr;||vr;q||d||<qd|vr[|D] \}}||vrZ|||<qN|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)itemsrrgetrr0keys)rrDoutfieldcontentnestedkeyvaluer r rrCs(     zMask.filter_datacCsdddd|DS)Nz{{{0}}}r cSs0g|]\}}t|trd|t|fn|qS)r,)rrjoinr)r1kvr r rr4sz Mask.__str__..)formatrOrFr3r r r__str__sz Mask.__str__)NF) r r r r rrr"r0rCrS __classcell__r r rrrs /   rFcCst|||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 )rr0)rDrrr r rr0s r0)F)loggingre collectionsrinspectrerrorsr getLoggerr logcompiler#rrrr0r r r rs