U H4fod@s.dZddlZddlZddlZddlZddlmZmZmZej dkrPddl m Z n ddl m Z ddl mZddlmZmZdd lmZmZdd lmZmZdd lmZdd lmZmZmZmZd dddgZGdddeZ Gddde e!Z"Gdd d e"Z#Gddde"Z$Gddde"Z%Gddde Z&dS)a@ zeep.xsd.elements.indicators ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Indicators are a collection of elements. There are four available, these are All, Choice, Group and Sequence. Indicator -> OrderIndicator -> All -> Choice -> Sequence -> Group N) OrderedDict defaultdictdeque))cached_property)threaded_cached_property)etree)UnexpectedElementErrorValidationError)NotSet SkipValue)AnyElement)Base)NamePrefixGeneratorUniqueNameGeneratorcreate_prefixed_namemax_occurs_iterAllChoiceGroupSequencecsBeZdZdZfddZeddZeddZd d d ZZ S) Indicatorz#Base class for the other indicatorscsd|jjtfS)Nz<%s(%s)>) __class____name__super__repr__selfr@/tmp/pip-unpacked-wheel-04z2hmy8/zeep/xsd/elements/indicators.pyr,szIndicator.__repr__cCs&tdd|jD}|jr"d|iS|S)NcSsg|]\}}||jfqSr!) default_value).0nameelementr!r!r" 2sz+Indicator.default_value.._value_1)relementsaccepts_multiplervaluesr!r!r"r#/s zIndicator.default_valuecCs tdSNNotImplementedErrorrr!r!r"r)9szIndicator.elementscCs tdSr-r.rr% min_occurs max_occursr!r!r"clone=szIndicator.clone)r0r0) r __module__ __qualname____doc__rpropertyr#r)r4 __classcell__r!r!r r"r)s   rcseZdZdZdZdfdd ZdddZedd Zed d Z d d Z dddZ ddZ ddZ ddZddZdddZZS) OrderIndicatorz0Base class for All, Choice and Sequence classes.Nr0cs,||_||_t|dk r(||dSr-)r2r3r__init__extend)rr)r2r3r r!r"r;Fs  zOrderIndicator.__init__cCs|jt|||dS)N)r)r2r3)rlistr1r!r!r"r4Ms zOrderIndicator.clonecCs<g}|jD],\}}|dkr(||jq |||fq |Sz:List of tuples containing the element name and the elementN)elements_nestedr<r)append)rresultr%elmr!r!r"r)Rs zOrderIndicator.elementscCsg}t}t}|D]}t|ttttfrn|jrD|| |fq|j D]\}}| |}qJ|d|fqt|t tfr|| |fq| |j }|||fq|Sr>)rr isinstancerrrrr*r@get_namer)Z create_namer attr_name)rrA generatorZ generator_2rBsub_nameZsub_elmr%r!r!r"r?]s  zOrderIndicator.elements_nestedcCs||js |g}t}|D]\}d}|jD]B\}}t|trX|j|krf||jdk rf|d7}q$|||7}q$||qt|S)Return the number of values which are accepted by this choice. If not all required elements are available then 0 is returned. rNr0) r*setr?rCrr%acceptaddmax)rr,resultsvaluenumr%r&r!r!r"rJss   zOrderIndicator.acceptrcCsd|jD]\}}t|trtdqi}|jD],\}}|t|krDqZ||||<|d7}q,|||fS)N0Choice elements only work with keyword argumentsr0)r?rCr TypeErrorr)len)rargsindexr%rBrAr&r!r!r" parse_argss     zOrderIndicator.parse_argsc CsT|jr|st|r ||kr iS|js*t||}t|tsD|g}g}t|j|D]}zt|}Wnt k rt dYnXt }|j D](\}} | |||} | dk r|| q|rt d|t|df||qT||i}ttd|s|||S|jrtt }|jD](\} } | || |} | r"|| q"|SdS)aApply the given kwarg to the element. The available_kwargs is modified in-place. Returns a dict with the result. :param kwargs: The kwargs :type kwargs: dict :param name: The name as which this type is registered in the parent :type name: str :param available_kwargs: The kwargs keys which are still available, modified in place :type available_kwargs: set :rtype: dict z3A list of dicts is expected for unbounded SequencesN+%s() got an unexpected keyword argument %r.r)r*AssertionErrorgetrCr=rr3rIkeysAttributeErrorrQrr) parse_kwargsupdater@anyfilterremover?) rkwargsr%available_kwargs item_kwargsrAZ item_value subresultZ item_namer&rNelm_name sub_resultr!r!r"r[sP        zOrderIndicator.parse_kwargscCs"t|D]\}}|||<q|Sr-) enumerateresolve)rirBr!r!r"rgszOrderIndicator.resolvec Cst|ts|g}n|}|||t|j|D]l}|jD]`\}}|rj||kr`||}||g}qrt}|}n|}|}|tkr|q8|dk s|js8| |||q8q.dS)z.Create subelements in the given parent object.N) rCr=validaterr3r?r r is_optionalrender) rparentrN render_pathr,r%r&Z element_valueZ child_pathr!r!r"rks$   zOrderIndicator.rendercCs"|D]}|tkrtd|dqdS)Nz No value setpath)r r )rrNrmitemr!r!r"riszOrderIndicator.validateTcCsrg}|jD]H\}}t|tr2||j|ddq |j|dd}|d||fq d|}|jrnd|fS|S)NF standalonez%s: %sz, z[%s])r?rCrr@ signaturejoinr*)rschemarrpartsr%r&rNpartr!r!r"rss   zOrderIndicator.signature)Nr0r0)r0r0)r)NT)rr5r6r7r%r;r4rr)r?rJrUr[rgrkrirsr9r!r!r r"r:As    Gr:cs,eZdZdZd fdd Zd ddZZS) rzlAllows the elements in the group to appear (or not appear) in any order in the containing element. Nr0Fcst|||||_dSr-)rr;_consume_other)rr)r2r3Z consume_otherr r!r"r;sz All.__init__c Cst}dd|jD}t}tt}t|D],\} } | j|kr,|| || j| q,t |ddD] } || =qf|jD],\}} | | j } | rx| j | ||d||<qx|j r|rt||d<||S)Consume matching xmlelements :param xmlelements: Dequeue of XML element objects :type xmlelements: collections.deque of lxml.etree._Element :param schema: The parent XML schema :type schema: zeep.xsd.Schema :param name: The name of the parent element :type name: str :param context: Optional parsing context (for inline schemas) :type context: zeep.xsd.context.XmlParserContext :rtype: dict or None cSsh|]\}}|jqSr!)qname)r$__r&r!r!r" 0sz(All.parse_xmlelements..T)reversecontextZ _raw_elements)rr)rIrrrftagrKr@sortedrXrzparse_xmlelementsrxr=clear) r xmlelementsrur%rrAZ expected_tagsZ consumed_tagsr,rhrBr&Z sub_elementsr!r!r"r!s,      zAll.parse_xmlelements)Nr0r0F)NN)rr5r6r7r;rr9r!r!r r"rsc@sneZdZdZdddZeddZeddZdd d Zd d Z ddZ ddZ ddZ ddZ dddZd S)rz@Permits one and only one of the elements contained in the group.rcCs|r tddS)NrP)rQrrSrTr!r!r"rUMszChoice.parse_argscCsdS)NTr!rr!r!r"rjQszChoice.is_optionalcCstSr-)rrr!r!r"r#UszChoice.default_valueNc Cs g}t|jD]}|sqg}|jD]v\}} t|} z| j| |||d} Wntk rdYq$YnXt| trx|| i} t|t| } | r$| | | fq$|sg}qt |t ddd}|r| |ddt |ddD] } |qqqq|jr ||i}n|r|dni}|S)ry)rrur%rrTkeyr}r0)rr3r?copyrr rCrrRr@roperator itemgetterrangepopleftr*)rrrur%rrA_unusedoptionsZ element_namer&Zlocal_xmlelementsreZ num_consumedrhr!r!r"rYsD      zChoice.parse_xmlelementscCs|r||kr|jst||p g}||g}t|tr@|g}|D]}|D]}t|tr||krj||n|}||r||qDqLt|tr||qDqL|j |krL| |j }||j |iqDqLt d|| fqD|js|r|dnd}n|jriSi}d} |j D]j\}} t|} | ||| } | rt| s`|| || n| s|| || d} q| r|jD]\} } || dqni}|r|jr||i}|S)aProcesses the kwargs for this choice element. Returns a dict containing the values found. This handles two distinct initialization methods: 1. Passing the choice elements directly to the kwargs (unnested) 2. Passing the choice elements into the `name` kwarg (_value_1) (nested). This case is required when multiple choice elements are given. :param name: Name of the choice element (_value_1) :type name: str :param element: Choice element object :type element: zeep.xsd.Choice :param kwargs: dict (or list of dicts) of kwargs for initialization :type kwargs: list / dict zJNo complete xsd:Sequence found for the xsd:Choice %r. The signature is: %srNFT)r*rWr_rCdictr:rJr@rr%rXrQrsr?rr[r]r,intersection_updater\r) setdefault)rr`r%rar,rArNr& choice_valuefoundchoiceZ temp_kwargsrcZ choice_namer!r!r"r[sd                  zChoice.parse_kwargscCsJ|js |g}||||D](}||}|r|\}}||||qdS)zRender the value to the parent element tree node. This is a bit more complex then the order render methods since we need to search for the best matching choice element. N)r*ri_find_element_to_renderrk)rrlrNrmrprAr&rr!r!r"rks  z Choice.rendercCs>d}|D]}||}|r|d7}q|s:|js:td|ddS)Nrr0zMissing choice valuesrn)rrjr )rrNrmrrprAr!r!r"ris   zChoice.validatecst}|jD]j\}t|trb|jrFtfdd|Dr`|dqv|krv|rv|dq ||}||q |rt|SdS)rHc3s|]}|ko|VqdSr-r!)r$rpr%r!r" sz Choice.accept..r0r) rIr?rCrr*allrKrJrL)rr,numsr&rOr!rr"rJs     z Choice.acceptc Csg}|jD]\}}t|trh|j|krz||j}Wntk rL|}YnX|dk r|d||fq |dk rz ||}Wqttfk r|}YqXn|}||}|r ||||fq |rt|t ddd}|dddSdS)zReturn a tuple (element, value) for the best matching choice. This is used to decide which choice child is best suitable for rendering the available data. Nr0rTr) r?rCrr%KeyErrorr@rQrJrrr)rrNmatchesr%r&rZscorer!r!r"rs,      zChoice._find_element_to_renderTc Csvg}|jD]H\}}t|tr6|d|j|ddq |d||j|ddfq dd|}|jrrd|fS|S)Nz{%s}Frqz{%s: %s}z(%s)z | z%s[])r?rCr:r@rsrtr*)rrurrrvr%r&rwr!r!r"rs?s  zChoice.signature)r)NN)NT)rr5r6r7rUr8rjr#rr[rkrirJrrsr!r!r!r"rJs    >T $c@seZdZdZdddZdS)rzoRequires the elements in the group to appear in the specified sequence within the containing element. Nc Csg}|jr|stt|jD]}|s(qt}|jD]h\}} z| j||||d} Wn"tk rt|jj rld} YnXt | t r| | n| ||<|s4qq4|r| |q|js|r|dSdS||iS)ryr~Nr)r*rWrr3rr)rr settingsstrictrCr:r\r@) rrrur%rrArZ item_resultrdr&Zitem_subresultr!r!r"rTs:     zSequence.parse_xmlelements)NN)rr5r6r7rr!r!r!r"rNscseZdZdZdfdd ZddZddZed d Zdd d Z d dZ d ddZ ddZ d!ddZ ddZddZd"ddZZS)#rzyGroups a set of element declarations so that they can be incorporated as a group into complex type definitions. r0cs6t||_||_|r |jnd|_||_||_dSr-)rr;childrz localnamer%r3r2)rr%rr3r2r r!r"r;s  zGroup.__init__cCs|Sr-)rsrr!r!r"__str__sz Group.__str__cos|jD] }|VqdSr-)r)rrSr`rpr!r!r"__iter__s zGroup.__iter__cCs|jrd|jfgS|jjS)Nr()r*rr)rr!r!r"r)s zGroup.elementscCs|jd|j||dS)N)r%rr2r3)rrr1r!r!r"r4s z Group.clonecCs |j|S)rH)rrJr+r!r!r"rJsz Group.acceptrcCs|j||Sr-)rrUrr!r!r"rUszGroup.parse_argsc Cs|jr||kriS||||}g}|jjr4dnd}t|j|D]J}t|}|j|||} |rtd|t |df| rD| | qD|r||i}n|j|||}|S)Nr(rVr) r*r_rrr3rIrYr[rQr=r@) rr`r%rarbrArGZ sub_kwargsZavailable_sub_kwargsrcr!r!r"r[s4    zGroup.parse_kwargsNc CsPg}t|jD]&}||jj||||d|sq6q|jsH|rH|dS||iS)ryr~r)rr3r@rrr*)rrrur%rrArr!r!r"rs zGroup.parse_xmlelementscCs4t|ts|g}n|}|D]}|j|||qdSr-)rCr=rrk)rrlrNrmr,r!r!r"rks  z Group.rendercCs|j|_|Sr-)rrgrr!r!r"rgs z Group.resolveTcCs<t|j|}|r(d||jj|ddfS|jj|ddSdS)Nz%s(%s)Frq)rrzrrs)rrurrr%r!r!r"rss zGroup.signature)r0r0)r0r0)r)NN)NT)rr5r6r7r;rrrr)r4rJrUr[rrkrgrsr9r!r!r r"rs     )'r7rrsystyping collectionsrrr version_info functoolsrrlxmlr Zzeep.exceptionsr r Zzeep.xsd.constr r Zzeep.xsd.elementsrrZzeep.xsd.elements.baserZzeep.xsd.utilsrrrr__all__rr=r:rrrrr!r!r!r"s0      W38