o g@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 extensionnotationN)__name__ __module__ __qualname__r rimport_rrrrrrrrrrrrrrr r!r"r#r$r%r*r*I/home/ubuntu/webapp/venv/lib/python3.10/site-packages/zeep/xsd/visitor.pyrs. rc@sdeZdZdZddZdejdejfddZ dejdej d d fd d Z dejd d fd dZ dejdej fddZdejdejd d fddZddZddZd^ddZddZddZddZdd Zd!d"Zd#ejd$ejd ejej ejffd%d&Zd'd(Zd)d*Zd+d,Z d-d.Z!d/d0Z"d1d2Z#d3d4Z$d5d6Z%d7d8Z&d9d:Z'd;d<Z(d=d>Z)d?d@Z*dAdBZ+dCdDZ,dEdFZ-dGdHZ.dIdJZ/dKdLZ0dMdNZ1dOdPZ2d^dQej3fdRdSZ4dTdUZ5dVdWZ6dXdYZ7dZd[Z8d\d]Z9ie:j;e(e:jee:j?e1e:j@ee:jAd e:jBd e:jCe)e:jDe*e:jEe+e:jFee:jGee:jHee:jIe'e:jJe0e:jKe2ZLd S)_ 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__>s zSchemaVisitor.__init__qnameinstancecC|j||dSr-)r.register_elementr1r3r4r*r*r+r6CzSchemaVisitor.register_elementnamereturnNcCr5r-)r.register_attribute)r1r9r4r*r*r+r;Fz SchemaVisitor.register_attributecCr5r-)r. register_typer7r*r*r+r=Kr8zSchemaVisitor.register_typecCr5r-)r.register_groupr7r*r*r+r>Nr8zSchemaVisitor.register_groupcCr5r-)r.register_attribute_groupr7r*r*r+r?Qr<z&SchemaVisitor.register_attribute_groupcCr5r-)r.register_import)r1 namespacer.r*r*r+r@Vr8zSchemaVisitor.register_importcCs0|j|j}|std|j||||}|S)NzNo visitor defined for %r)visitorsgettag ValueError)r1nodeparent visit_funcresultr*r*r+processYs  zSchemaVisitor.processcCs@t|d}|r||}|jdkrdStj|j||j|dSdS)Nrefz http://www.w3.org/2001/XMLSchema array_type)r _create_qnamerA xsd_elements RefAttributerDr)r1rFrMrKr*r*r+process_ref_attribute`s    z#SchemaVisitor.process_ref_attributecKst|d}|s dS||}|jtjkrtj}n|jtjkr"tj}n|jtj kr,tj }n |jtj kr5tj }||j||j fi|S)NrK)rrNrDrrrO RefElementrrPrRefGroupr"RefAttributeGroupr)r1rFkwargsrKclsr*r*r+process_referencens      zSchemaVisitor.process_referencecCs^|dusJ|d}|r||j_|dd|j_|dd|j_|D] }|j||dq#dS)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 unqualifiedattributeFormDefaultrG)rCr._target_namespace _element_form_attribute_formrJ)r1rFtnschildr*r*r+ visit_schemas  zSchemaVisitor.visit_schemac Cs*d}|d}|d}|rt|jj||jj}|s(|jjs(td|jj|j d|s0|s0d|j_ |jj ||}|rHt d|||||S|sP|dkrPd }|sZt d |dS|j||jjd }|d }|r||r|||kr|td ||jj|j d|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 NrAschemaLocationz[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_urlrXzaThe namespace defined on the xsd:import doesn't match the imported targetNamespace located at %r )target_namespace)rCrrsettingsr. _base_urlr]rlocationrf_has_empty_import documentsget_by_namespace_and_locationloggerdebugr@_retrieve_data _locationcreate_new_document) r1rFrG schema_noderArmr. schema_tnsrr*r*r+ visit_imports^         zSchemaVisitor.visit_importc Cst|ds td|d}||jvrdS|j||jjd}|j||jds{|jds6|jdr{d|jdpA|jdi}| |jt j |j |d}|D]}| |qS|jD] \}} ||| q`|jdsy|jd|jd<|}|jj} |jj} |jj} |dd|j_|d d|j_t||jj|j_|D] }|j||d q| |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 rczschemaLocation is requiredNrhrX)nsmaprYrZr[r\)rCNotImplementedErrorr0rsr.rladdryattribupdaterElementrDappenditemsr/r^r_rrJ) r1rFrGrmrvrynewrakeyvalueelement_form_defaultattribute_form_defaultrir*r*r+ visit_include sJ          zSchemaVisitor.visit_includec CsR|jtjk}|st|\}}nd}d}|s)|jtjkrd}|j|||d}|r)|S|d|jj}|dks7|r@t |d|jj }n t |d }t|} d} | rsd} | D]} | jtjkr_qV| jtjtjfvrr| rlJ|| |} qV| st |d} | r|| j} nt} |d d k}|d }tj|| |||||d }|r||||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 qualifiedr9Ntypenillabletruedefault)r9type_rrrr is_global)rDrr_process_occurs_attrsrrWrCr.r^rr]rQNamestriprrrrrJ _get_typetext xsd_typesAnyTyperOr~r6)r1rFrGrrrrI element_formr3childrenxsd_typerra node_typerrrr*r*r+ visit_elementTs^        zSchemaVisitor.visit_elementrFrGcCs:|jtjk}|d}|r'td|}|r'|d}t||j}t ||j}|s4|j ||d}|r4|S|d|j j }|dksB|rKt |d|j j} nt|d} |t|\} } | rg|| d|} nt |d} | rt|| } nt} |d d k}|d }tj| | ||d }|r| d usJ|| ||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([^\[]+)rrLrrr9ruserequiredr)rrrN)rDrrrCrematchgroupsrryrrQr.r_rr]rr_pop_annotationrvisit_simple_typerrrrO Attributer;)r1rFrGrrMrr3rIattribute_formr9rrrrrrattrr*r*r+visit_attributes@            zSchemaVisitor.visit_attributec Cs|jtjkr|d}d}n|dd}d}d}t||j|jj}|t |\}}|d} | jtj krB| | |}t |||j} n!| jtj krO| | |} n| jtjkr\|| |} ntd| j| dusiJ|rq||| | 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 r9T AnonymousFz({http://www.w3.org/2001/XMLSchema}stringrzUnexpected child: %rN)rDrrrCrryr.r]rrr#visit_restriction_simple_typer visit_listr visit_unionAssertionErrorr=) r1rFrGr9r base_typer3rrrarr*r*r+rs*         zSchemaVisitor.visit_simple_typecCsZg}d}|jtjkr|d}d}n|d}d}t||j|jj}d|d}t|t j f|} d} | t |\} }|rA|djnd} | tj kr[||d|\}} | | |||d } nH| tjkrs||d|}| d ||d |} n0| rd}| tjtjtjtjfvr|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}anyTyper9TFzzeep.xsd.dynamic_types)r' _xsd_nameNr) attributesr$r3r)r3r)rrr3rr*)rDrrrCrryr.r]rr ComplexTyperrrvisit_simple_contentrvisit_complex_contentrrrrpoprJ_process_attributesr=)r1rFrGrrr9rr3cls_attributesxsd_clsrr first_tagrrUrrar*r*r+visit_complex_type0sJ           z SchemaVisitor.visit_complex_typecCsht|}|d}|jtjkr|||\}}}|||dS|jtjkr2|||\}}}|||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)rrDrr#!visit_restriction_complex_contentr$visit_extension_complex_content)r1rFrGrrabaserrr*r*r+r|s    z#SchemaVisitor.visit_complex_contentcCsHt|}|d}|jtjkr|||S|jtjkr |||Std)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 extension)rrDrr# visit_restriction_simple_contentr$visit_extension_simple_contentr)r1rFrGrrar*r*r+rs    z"SchemaVisitor.visit_simple_contentcCsNt|d}|r ||S|t|\}}|djtjkr%||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)rrrrrDrrr)r1rFrG 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)r1rFrGrrr*r*r+rs  z.SchemaVisitor.visit_restriction_simple_contentc Ct|d}||}|t|\}}d}g}|r;|d} | jtjtjtjtj fvr5| 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 rrrrrDrrrrrrrJr r1rFrGrrrrrrrar*r*r+rs      z/SchemaVisitor.visit_restriction_complex_contentc Cr)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)r1rFrGrrrrrr*r*r+r7s  z,SchemaVisitor.visit_extension_simple_contentcCdS)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*r1rFrGr*r*r+visit_annotationJszSchemaVisitor.visit_annotationcCs(t|\}}|dd}tj|||dS)a Definition:: rr?r@rJrQrWrbrxrr_ElementtypingUnionrPrrrrrrrrrrrrrrrrrrrrrrIOrsrrNrrrrr rrrr!rrrrrrrr)rrr"r%rBr*r*r*r+r,+s   (dIb K/L!"$-!,       r,cCs@|dd}t|dd}|dkrd}||fSt|}||fS)z)Process the min/max occurrence indicators maxOccurs1 minOccurs unbounded)rCint)rFrrr*r*r+rs r)loggingrrlxmlrzeep.exceptionsr zeep.loaderrrr zeep.utilsrrzeep.xsdr rOr rzeep.xsd.constr r zeep.xsd.types.unresolvedr r getLoggerr&rqrr,rr*r*r*r+s0      `