o g,G@sddlZddlZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZddl mZdd l mZdd lmZdd lmZdd lmZeeZGd d d ZGdddZGdddZdS)N) OrderedDict)etree) exceptionsns) load_external)Settings)const)elements)types)builtins) SchemaVisitorc@s@eZdZdZdd,d-Z!d.d/Z"d0d1Z#d2d3Z$d4d5Z%d6d7Z&d8d9Z'd=d:d;Z(dS)?Schemaz-A schema is a collection of schema documents.NcCs`|pt|_||_t|_i|_i|_|t|t s&|dur#|gng}n|}| ||dS)z :param node: :param transport: :param location: :param settings: The settings object N) rsettings _transport_SchemaContainer documents_prefix_map_auto_prefix_map_custom_load_default_documents isinstancelist add_documents)selfnode transportlocationrnodesrH/home/ubuntu/webapp/venv/lib/python3.10/site-packages/zeep/xsd/schema.py__init__s  zSchema.__init__cCs|j}|r d|j|jfSdS)Nzz ) root_document _location_target_namespace)rmain_docrrr__repr__/szSchema.__repr__cs2i|jfdd|jDS)Ncs"i|] \}}|vr||qSrvalues).0kvretvalrr =s"z%Schema.prefix_map..)updaterritemsrrr*r prefix_map8s  zSchema.prefix_mapcCstdd|jDdS)Ncss|]}|js|VqdSN) _is_internal)r'docrrr Csz'Schema.root_document..)nextrr/rrrr AzSchema.root_documentcCstdd|jDS)zABoolean to indicate if this schema contains any types or elementscss|]}|jVqdSr1)is_empty)r'documentrrrr4Hsz"Schema.is_empty..)allrr/rrrr7EszSchema.is_emptycCs |jSr1)rget_all_namespacesr/rrr namespacesJs zSchema.namespacesccDt}|jD]}|jD]}|j|vr|V||jqqdS)zYYield all globla xsd.Type objects :rtype: Iterable of zeep.xsd.Element N)setr _elementsr&qnameadd)rseenr8elementrrrr N   zSchema.elementsccr<)z]Yield all global xsd.Type objects :rtype: Iterable of zeep.xsd.ComplexType N)r=r_typesr&r?r@)rrAr8type_rrrr \rCz Schema.types schema_nodesrreturncCsDg}|D] }|||}||q|D]}|q||_dSr1)create_new_documentappendresolve_create_prefix_mapr)rrFr resolve_queuerr8rrrrjs   zSchema.add_documentsurlcCs,t||j|jd}|j||d}|dS)N)r)rM)rrrrHrJ)rrM schema_noder8rrradd_document_by_urlws zSchema.add_document_by_urlcCs||}||ddS)z7Return a global xsd.Element object with the given qname get_elementrB) _create_qname _get_instancerr?rrrrP|s zSchema.get_elementFc CsZ||}z||ddWStjy,}z|r tt|nWYd}~dSd}~ww)zReturn a global xsd.Type object with the given qname :rtype: zeep.xsd.ComplexType or zeep.xsd.AnySimpleType get_typetypeN)rQrRrNamespaceErrorloggerdebugstr)rr? fail_silentlyexcrrrrTs zSchema.get_typecC||ddS)z6Return a global xsd.Group object with the given qname. get_groupgrouprRrSrrrr]zSchema.get_groupcCr\)z9Return a global xsd.attribute object with the given qname get_attribute attributer_rSrrrrar`zSchema.get_attributecCr\)z>Return a global xsd.attributeGroup object with the given qnameget_attribute_groupattributeGroupr_rSrrrrcr`zSchema.get_attribute_groupcCs||j|<dSr1)r)rprefix namespacerrr set_ns_prefixzSchema.set_ns_prefixcCsJzz|j|WWSty|j|YWSwty$td|w)NzNo such prefix %r)rKeyErrorr ValueError)rrerrr get_ns_prefixs   zSchema.get_ns_prefixcCsX|jD] \}}||kr|Sq|jD] \}}||kr#|Sq|dkr*dS|S)Nz)http://schemas.xmlsoap.org/soap/envelope/zsoap-env)rr.r)rrfreother_namespacerrrget_shorthand_for_nsszSchema.get_shorthand_for_nscCsR|dur |dnd}|s|}|dur|}t|||}|j|||||S)z: :rtype: zeep.xsd.schema.SchemaDocument NtargetNamespace)getSchemaDocumentrr@load)rrrMbase_urltarget_namespacerfschemarrrrHs   zSchema.create_new_documentcCs&|jD]}|j|q||_dS)z%Merge an other XSD schema in this oneN)rr@rKr)rrtr8rrrmerges z Schema.mergecCs||j}|j||dS)N)rt)rPtagparse)rrelmrrr deserializes zSchema.deserializecCshttjdd}tjD]}|dd}||j|q tjD] }|}| |j |qd|_ |j ||S)NT) is_global)rprXSDxsd_builtins_typesrD register_type_default_qnamexsd_builtins_elementsr>register_elementr?r2rr@)rrtclsinstancerrrrs    zSchema._load_default_documentsc Cs||}z6d}||jD]$}t||}z||WWStjy2}z|}WYd}~qd}~ww|dur9|WdStjyOtd||j|jfw)z1Return an object from one of the SchemaDocument'sNzBUnable to resolve %s %s. No schema available for the namespace %r.)rQ_get_schema_documentsrfgetattrr LookupErrorrVtext)rr? method_namenamelast_exceptionrtmethodr[rrrrRs,    zSchema._get_instancecCst|tjr|S|ds>d|vr>|jr>|dd\}}||jvr*t|j||S||jvr8t|j||Std|t|S)zCreate an `lxml.etree.QName()` object for the given qname string. This also expands the shorthand notation. :rtype: lxml.etree.QNaame {:z&No namespace defined for the prefix %r)rrQName startswithrsplitrrj)rrre localnamerrrrQs     zSchema._create_qnamecCsJddi}d}|jD]}|dus||vrq ||d|<|d7}q |S)Nxsdz http://www.w3.org/2001/XMLSchemarzns%dr)rr:r&)rr0irfrrrrK s  zSchema._create_prefix_mapcCs:|j|s|tjvrtd||||j||S)zlReturn a list of SchemaDocument's for the given namespace. :rtype: list of SchemaDocument z'Auto importing missing known schema: %s)rhas_schema_document_for_nsrAUTO_IMPORT_NAMESPACESrWrXrOget_by_namespacerrfrZrrrrs    zSchema._get_schema_documents)NNNN)F)NN))__name__ __module__ __qualname____doc__rr$propertyr0r r7r;r r typingListr_ElementrYrrO xsd_elementsElementrPrTGroupr] AttributeraAttributeGrouprcrgrkrmrHruryrrRrQrKrrrrrr sP             r c@seZdZdZddZddZdd d Zdeje fd d Z dejdfddZ de de dej dfddZ de defddZddZd S)rzUContainer instances to store multiple SchemaDocument objects per namespace. cCs t|_dSr1)r _instancesr/rrrr.s z_SchemaContainer.__init__ccs|EdHdSr1r%r/rrr__iter__1sz_SchemaContainer.__iter__r8rprGNcCs2td|jt||j|jg}||dS)zAdd a schema documentz%Add document with tns %s to schema %sN)rWrXrfidr setdefaultrI)rr8rrrrr@4s  z_SchemaContainer.addcCst|jSr1)rrkeysr/rrrr:<rhz#_SchemaContainer.get_all_namespacescCs*||jvr|r gStd||j|S)Nz(No schema available for the namespace %r)rrrVrrrrr?s  z!_SchemaContainer.get_by_namespacerfrcCs.|j|dd}|D] }|j|kr|Sq dS)z_groupsrDr_imports _element_form_attribute_form _resolved)rrfrrrrrrris   zSchemaDocument.__init__cCsd|j|j|jfS)Nz2)r!r"r7r/rrrr$s zSchemaDocument.__repr__cCs|jSr1)r"r/rrrrfszSchemaDocument.namespacecCst|jp |jp |j Sr1)rrrDr>r/rrrr7szSchemaDocument.is_emptycCs:|durdS|j|jstdt||}||dS)zLoad the XML Schema passed in via the node attribute. :type schema: zeep.xsd.schema.Schema :type node: etree._Element NzIThe document needs to be registered in the schema before it can be loaded)rrr" RuntimeErrorr visit_schema)rrtrvisitorrrrrqs zSchemaDocument.loadcCs|td||jr dSd|_|jD] }|D]}|qqdd}||j||j||j||j ||j dS)NzResolving in schema %sTc Sstz|D]\}}|}|dusJd|||<qWdStjy9}ztd|j|j|j|jdd}~ww)Nz!resolve() should return an objectzGUnable to resolve %(item_name)s %(qname)s in %(file)s. (via %(parent)s)) item_namer?fileparent)r.rJrrrr?r)valkeyobjnewr[rrr _resolve_dicts$ z-SchemaDocument.resolve.._resolve_dict) rWrXrrr&rJrrr>rrD)rschemasrtrrrrrJs      zSchemaDocument.resolvecCs|j|g}||dS)zRegister an import for an other schema document. :type namespace: str :type schema: zeep.xsd.schema.SchemaDocument N)rrrI)rrfrtrrrrregister_importszSchemaDocument.register_importcCrr1)rrrrr is_importeds zSchemaDocument.is_importedr?valuecC||||jddS)z"Register a xsd.Type in this schemarUN)_add_componentrDrr?rrrrr}r6zSchemaDocument.register_typecCr)z%Register a xsd.Element in this schemarBN)rr>rrrrrr6zSchemaDocument.register_elementcCr)z#Register a xsd:Group in this schemar^N)rrrrrrregister_groupr6zSchemaDocument.register_groupcCr)z'Register a xsd:Attribute in this schemarbN)rrrrrrregister_attributer6z!SchemaDocument.register_attributerGNcCr)z,Register a xsd:AttributeGroup in this schemaattribute_groupN)rrrrrrregister_attribute_groupsz'SchemaDocument.register_attribute_groupcC|||jdS)zsReturn a xsd.Type object from this schema :rtype: zeep.xsd.ComplexType or zeep.xsd.AnySimpleType rU)_get_componentrDrSrrrrTszSchemaDocument.get_typecCr)z,Return a xsd.Element object from this schemarB)rr>rSrrrrPzSchemaDocument.get_elementcCr)z*Return a xsd.Group object from this schemar^)rrrSrrrr]rzSchemaDocument.get_groupcCr)z.Return a xsd.Attribute object from this schemarb)rrrSrrrrarzSchemaDocument.get_attributecCr)z3Return a xsd.AttributeGroup object from this schemard)rrrSrrrrcrz"SchemaDocument.get_attribute_groupcCs.t|tjr |j}td||||||<dS)Nzregister_%s(%r, %r))rrrrrWrX)rrrr.rrrrrs  zSchemaDocument._add_componentc CsXz||WSty+d|}tjd||d|j|j|p!dd|||jdw)Nz, zpNo %(item_name)s '%(localname)s' in namespace %(namespace)s. Available %(item_name_plural)s are: %(known_items)ssz - )ritem_name_pluralrrf known_items)r?rr)rijoinrrrrrfr!)rr?r.rrrrrr s$   zSchemaDocument._get_component)#rrrrrr$rrfr7rqrJrrrr xsd_typesTyper}rrrrrrYrrrrrTrPr]rarcrrrrrrrpas<  %   rp) loggingr collectionsrlxmlrzeeprr zeep.loaderr zeep.settingsrzeep.xsdrr rr rzeep.xsd.elementsr rzeep.xsd.typesr|zeep.xsd.visitorr getLoggerrrWr rrprrrrs&           9