U ð¤ùd^ã@s†ddlZddlZddlmZddlmZddlmZe e ¡Z e  d¡Z Gdd„deƒZ Gd d „d e ƒZGd d „d eƒZddd„ZdS)éN)Ú OrderedDict)Úisclassé)Ú RestErrorz\{|\}|\,|[\w_:\-\*]+c@seZdZdZdS)Ú MaskErrorz#Raised when an error occurs on maskN©Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r ú4/tmp/pip-unpacked-wheel-dt_sn2ih/flask_restx/mask.pyrsrc@seZdZdZdS)Ú ParseErrorz#Raised when the mask parsing failedNrr r r r rsrcsJeZdZdZd‡fdd„ Zdd„Zdd „Zd d „Zd d „Zdd„Z ‡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 sj||_t|tƒr*tt|ƒ ¡| |¡nrszMask.apply..)ÚdefaultÚ attributer)rzMask is inconsistent with modelÚ__dict__)r*r%rÚlistÚtupleÚsetZNestedÚListZ PolymorphÚcloneÚtypeZRawr3r4rÚ issubclassrrrÚhasattrr5Ú filter_data)rÚdatar%r r1r r.fs&     ÿþ ý z Mask.applycCsÀi}| ¡D]€\}}|dkr q q t|tƒrj| |d¡}|jrH|dkrHq qŒ|dkrZd||<qŒ| |¡||<q |jr|||kr|q q | |d¡||<q d| ¡kr¼| ¡D]\}}||kr¢|||<q¢|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)ÚitemsrrÚgetrr.Úkeys)rr?ÚoutÚfieldÚcontentÚnestedÚkeyÚvaluer r r r>†s&      zMask.filter_datacCsd d dd„| ¡Dƒ¡¡S)Nz{{{0}}}rcSs0g|](\}}t|tƒr(d |t|ƒf¡n|‘qS)r*)rrÚjoinr)r/ÚkÚvr r r r2©sÿz Mask.__str__..)ÚformatrJrAr1r r r Ú__str__¦sþÿÿz Mask.__str__)NF) rr r r rrr r.r>rNÚ __classcell__r r rr rs /   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 )rr.)r?rrr r r r.±s r.)F)ÚloggingÚreÚ collectionsrÚinspectrÚerrorsrÚ getLoggerrÚlogÚcompiler!rrrr.r r r r Ús