U H4f@sddlZddlZddlZddlmZddlmZddlmZm Z m Z ddl m Z m Z ddlmZddlmZddlmZmZdd lmZmZeeZGd d d ZGd d d ZddZdS)N)etree) XMLParseError)absolute_location load_externalnormalize_location)as_qname qname_attr)elements)types)AUTO_IMPORT_NAMESPACESxsd_ns)UnresolvedCustomTypeUnresolvedTypec@seZdZedZedZedZedZedZedZ edZ edZ ed Z ed Z ed Zed Zed ZedZedZedZedZedZedZedZedZedZdS)tagsschemaimportinclude annotationelement simpleType complexType simpleContentcomplexContentsequencegroupchoicealllistunion attributeany anyAttributeattributeGroup restriction extensionZ notationsN)__name__ __module__ __qualname__r rimport_rrrrrrrrrrrrrrr r!r"r#r$notationr*r*4/tmp/pip-unpacked-wheel-04z2hmy8/zeep/xsd/visitor.pyrs,rc"@s0eZdZdZddZejejdddZ ejej ddd d Z ejdd d d Z ejej dddZejejddddZddZddZd]ddZddZddZddZdd Zd!d"Zejejejej ejfd#d$d%Zd&d'Zd(d)Zd*d+Z d,d-Z!d.d/Z"d0d1Z#d2d3Z$d4d5Z%d6d7Z&d8d9Z'd:d;Z(dd?Z*d@dAZ+dBdCZ,dDdEZ-dFdGZ.dHdIZ/dJdKZ0dLdMZ1dNdOZ2d^ej3dPdQdRZ4dSdTZ5dUdVZ6dWdXZ7dYdZZ8d[d\Z9e:j;e(e:jee:j?e1e:j@ee:jAde:jBde:jCe)e:jDe*e:jEe+e:jFee:jGee:jHee:jIe'e:jJe0e:jKe2iZLdS)_ SchemaVisitorapVisitor which processes XSD files and registers global elements and types in the given schema. Notes: TODO: include and import statements can reference other nodes. We need to load these first. Always global. :param schema: :type schema: zeep.xsd.schema.Schema :param document: :type document: zeep.xsd.schema.SchemaDocument cCs||_||_t|_dSN)documentrset _includes)selfrr.r*r*r+__init__>szSchemaVisitor.__init__)qnameinstancecCs|j||dSr-)r.register_elementr1r3r4r*r*r+r5CszSchemaVisitor.register_elementN)namer4returncCs|j||dSr-)r.register_attribute)r1r7r4r*r*r+r9Fsz SchemaVisitor.register_attribute)r3r8cCs|j||dSr-)r. register_typer6r*r*r+r:KszSchemaVisitor.register_typecCs|j||dSr-)r.register_groupr6r*r*r+r;NszSchemaVisitor.register_group)r3r4r8cCs|j||dSr-)r.register_attribute_groupr6r*r*r+r<Qsz&SchemaVisitor.register_attribute_groupcCs|j||dSr-)r.register_import)r1 namespacer.r*r*r+r=VszSchemaVisitor.register_importcCs0|j|j}|s td|j||||}|S)NzNo visitor defined for %r)visitorsgettag ValueError)r1nodeparentZ visit_funcresultr*r*r+processYs  zSchemaVisitor.processcCs@t|d}|r<||}|jdkr&dStj|j||j|dSdS)Nrefz http://www.w3.org/2001/XMLSchema array_type)r _create_qnamer> xsd_elements RefAttributerAr)r1rCrIrGr*r*r+process_ref_attribute`s   z#SchemaVisitor.process_ref_attributecKs~t|d}|sdS||}|jtjkr0tj}n:|jtjkrDtj}n&|jtj krXtj }n|jtj krjtj }||j||j f|S)NrG)rrJrArrrKZ RefElementrrLrZRefGroupr"ZRefAttributeGroupr)r1rCkwargsrGclsr*r*r+process_referencens      zSchemaVisitor.process_referencecCs^|dk s t|d}|r"||j_|dd|j_|dd|j_|D]}|j||dqFdS)aVisit the xsd:schema element and process all the child elements Definition:: Content: ( (include | import | redefine | annotation)*, (((simpleType | complexType | group | attributeGroup) | element | attribute | notation), annotation*)*) :param node: The XML node :type node: lxml.etree._Element NtargetNamespaceelementFormDefault unqualifiedattributeFormDefaultrD)AssertionErrorr@r._target_namespace _element_form_attribute_formrF)r1rCZtnschildr*r*r+ visit_schemas  zSchemaVisitor.visit_schemac Cs0d}|d}|d}|r0t|jj||jj}|sP|jjsPtd|jj|j d|s`|s`d|j_ |jj ||}|rt d|||||S|s|dkrd }|st d |dS|j||jjd }|d }|r|r||krtd ||jj|j dn|s|s|jj}|jj|||d}||||S)a Definition:: Content: (annotation?) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element Nr>schemaLocationz[The attribute 'namespace' must be existent if the importing schema has no target namespace.filename sourcelineTzReturning existing schema: %r$http://www.w3.org/XML/1998/namespacezhttps://www.w3.org/2001/xml.xsdzCIgnoring import statement for namespace %r (missing schemaLocation)base_urlrQzaThe namespace defined on the xsd:import doesn't match the imported targetNamespace located at %r )Ztarget_namespace)r@rrsettingsr. _base_urlrWrlocationr__has_empty_importZ documentsZget_by_namespace_and_locationloggerdebugr=_retrieve_data _locationZcreate_new_document) r1rCrD schema_noder>rer.Z schema_tnsrr*r*r+ visit_importsh         zSchemaVisitor.visit_importc Csv|dstd|d}||jkr*dS|j||jjd}|j||jds|jdsl|jdrd|jdp|jdi}| |jt j |j |d}|D]}| |q|jD]\}} ||| q|jds|jd|jd<|}|jj} |jj} |jj} |dd|j_|d d|j_t||jj|j_|D]}|j||d qD| |j_| |j_| |j_dS) a Definition:: Content: (annotation?) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element r\zschemaLocation is requiredNrarQ)nsmaprRrSrTrU)r@NotImplementedErrorr0rir.rdaddrmattribupdaterElementrAappenditemsr/rXrYrrF) r1rCrDrerkrmnewrZkeyvalueZelement_form_defaultZattribute_form_defaultrbr*r*r+ visit_include sN         zSchemaVisitor.visit_includec CsZ|jtjk}|st|\}}nd}d}|sR|jtjkr:d}|j|||d}|rR|S|d|jj}|dksn|rt |d|jj }nt |d }t|} d} | rd} | D]:} | jtjkrqq| jtjtjfkr| rt|| |} q| st |d} | r || j} nt} |d d k}|d }tj|| |||||d }|rV||||S) a Definition:: Content: (annotation?, ( (simpleType | complexType)?, (unique | key | keyref)*)) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element r min_occurs max_occursform qualifiedr7Ntypenillabletruedefault)r7type_r{r|rr is_global)rArr_process_occurs_attrsrrPr@r.rXrrWrQNamestriprrrrrVrF _get_typetext xsd_typesAnyTyperKrrr5)r1rCrDrr{r|rEZ element_formr3childrenxsd_typerwrZ node_typerrrr*r*r+ visit_elementTs`        zSchemaVisitor.visit_element)rCrDr8cCs>|jtjk}|d}|rNtd|}|rN|d}t||j}t ||j}|sh|j ||d}|rh|S|d|j j }|dks|rt |d|j j} nt|d} |t|\} } | r|| d|} n"t |d} | r|| } nt} |d d k}|d }tj| | ||d }|r:| d k s.t|| ||S)aDeclares an attribute. Definition:: Content: (annotation?, (simpleType?)) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element z+{http://schemas.xmlsoap.org/wsdl/}arrayTypez([^\[]+)rrHr}r~r7rZuserequiredr)rrrN)rArrr@rematchgroupsrrmrrMr.rYrrWrr_pop_annotationrvisit_simple_typerrrrK AttributerVr9)r1rCrDrrIrr3rEZattribute_formr7rrtrrrrattrr*r*r+visit_attributesF           zSchemaVisitor.visit_attributec Cs|jtjkr|d}d}n|dd}d}d}t||j|jj}|t |\}}|d} | jtj kr| | |}t |||j} nB| jtj kr| | |} n(| jtjkr|| |} ntd| j| dk st|r||| | S) a Definition:: Content: (annotation?, (restriction | list | union)) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element r7TZ AnonymousFz({http://www.w3.org/2001/XMLSchema}stringrzUnexpected child: %rN)rArrr@rrmr.rWrrr#visit_restriction_simple_typer visit_listr visit_unionrVr:) r1rCrDr7r base_typer3rrtrZrr*r*r+rs*         zSchemaVisitor.visit_simple_typecCs`g}d}|jtjkr$|d}d}n|d}d}t||j|jj}d|d}t|t j f|} d} | t |\} }|r|djnd} | tj kr||d|\}} | | |||d } n| tjkr||d|}| f||d |} nd| r>d}| tjtjtjtjfkr |d}|||}|||} | || ||d } n | ||d } |r\||| | S) a Definition:: Content: (annotation?, (simpleContent | complexContent | ((group | all | choice | sequence)?, ((attribute | attributeGroup)*, anyAttribute?)))) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element z){http://www.w3.org/2001/XMLSchema}anyTyper7TFzzeep.xsd.dynamic_types)r&Z _xsd_nameNr) attributesr$r3r)r3r)rrr3r)rArrr@rrmr.rWrrZ ComplexTyperrrvisit_simple_contentrvisit_complex_contentrrrrpoprF_process_attributesr:)r1rCrDrrr7rr3Zcls_attributesZxsd_clsrrZ first_tagrrNrrZr*r*r+visit_complex_type0sP           z SchemaVisitor.visit_complex_typecCsht|}|d}|jtjkr:|||\}}}|||dS|jtjkrd|||\}}}|||dSdS)a0The complexContent element defines extensions or restrictions on a complex type that contains mixed content or elements only. Definition:: Content: (annotation?, (restriction | extension)) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element )rrr#)rrr$N)rrArr#!visit_restriction_complex_contentr$visit_extension_complex_content)r1rCrDrrZbaserrr*r*r+r|s     z#SchemaVisitor.visit_complex_contentcCsLt|}|d}|jtjkr(|||S|jtjkr@|||StddS)a'Contains extensions or restrictions on a complexType element with character data or a simpleType element as content and contains no elements. Definition:: Content: (annotation?, (restriction | extension)) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element rz!Expected restriction or extensionN)rrArr# visit_restriction_simple_contentr$visit_extension_simple_contentrV)r1rCrDrrZr*r*r+rs    z"SchemaVisitor.visit_simple_contentcCsNt|d}|r||S|t|\}}|djtjkrJ||d|SdS)a Definition:: Content: (annotation?, (simpleType?, ( minExclusive | minInclusive | maxExclusive | maxInclusive | totalDigits |fractionDigits | length | minLength | maxLength | enumeration | whiteSpace | pattern)*)) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element rrN)rrrrrArrr)r1rCrD base_namerrr*r*r+rs   z+SchemaVisitor.visit_restriction_simple_typecCst|d}||}|gfS)a Definition:: Content: (annotation?, (simpleType?, ( minExclusive | minInclusive | maxExclusive | maxInclusive | totalDigits |fractionDigits | length | minLength | maxLength | enumeration | whiteSpace | pattern)* )?, ((attribute | attributeGroup)*, anyAttribute?)) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element r)rr)r1rCrDrrr*r*r+rs  z.SchemaVisitor.visit_restriction_simple_contentc Cst|d}||}|t|\}}d}g}|rv|d} | jtjtjtjtj fkrj| d| | |}| ||}|||fS)a Definition:: Content: (annotation?, (group | all | choice | sequence)?, ((attribute | attributeGroup)*, anyAttribute?)) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element rNr rrrrrArrrrrrrFr r1rCrDrrrrrrrZr*r*r+rs     z/SchemaVisitor.visit_restriction_complex_contentc Cst|d}||}|t|\}}d}g}|rv|d} | jtjtjtjtj fkrj| d| | |}| ||}|||fS)a  Definition:: Content: (annotation?, ( (group | all | choice | sequence)?, ((attribute | attributeGroup)*, anyAttribute?))) :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element rNrrrr*r*r+rs     z-SchemaVisitor.visit_extension_complex_contentcCs:t|d}||}|t|\}}|||}||fS)a Definition:: Content: (annotation?, ((attribute | attributeGroup)*, anyAttribute?)) r)rrrrr)r1rCrDrrrrrr*r*r+r7s   z,SchemaVisitor.visit_extension_simple_contentcCsdS)aDefines an annotation. Definition:: Content: (appinfo | documentation)* :param node: The XML node :type node: lxml.etree._Element :param parent: The parent XML node :type parent: lxml.etree._Element Nr*r1rCrDr*r*r+visit_annotationJszSchemaVisitor.visit_annotationcCs(t|\}}|dd}tj|||dS)a Definition:: r\r~) isinstancerrr> localnamer r.Z is_importedrgrhrrrr(rlrXrWrf)r1r7Z import_noder*r*r+rJs0     zSchemaVisitor._create_qnamecCsHt|sdgfS|djtjkr@||dd}||ddfSd|fS)Nrry)lenrArrr)r1rtrr*r*r+rs zSchemaVisitor._pop_annotationcCsRg}|D]D}|jtjtjtjfkr:|||}||q|d|j|q|S)NzUnexpected tag `%s`)rArrr"r!rFrsr)r1rCrtrrZrr*r*r+rs  z!SchemaVisitor._process_attributescCst||jj|jdS)Nr])rr.rjr_)r1messagerCr*r*r+rs zSchemaVisitor._create_error)N)N)Mr%r&r'__doc__r2rrrKrrr5rr9r:rr;rr<r=rFrMrPr[rlrxrZ_ElementtypingUnionrLrrrrrrrrrrrrrrrrrrrrrrIOrirrJrrrrr rrrr!rrrrrrrr(rrr"r)r?r*r*r*r+r,+s   (dIc K/L!"$-!,  r,cCs:|dd}t|dd}|dkr*d}nt|}||fS)z)Process the min/max occurrence indicatorsZ maxOccurs1Z minOccursZ unbounded)r@int)rCr|r{r*r*r+rs  r)loggingrrlxmlrZzeep.exceptionsrZ zeep.loaderrrrZ zeep.utilsrrZzeep.xsdr rKr rZzeep.xsd.constr r Zzeep.xsd.types.unresolvedr r getLoggerr%rgrr,rr*r*r*r+s.     `