U H4frG@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:eZdZdZd;ddZddZeddZed d Zed d Z ed dZ eddZ eddZ e jejeddddZeddddZejdddZdd9d:Z(dS)?Schemaz-A schema is a collection of schema documents.NcCs`|pt|_||_t|_i|_i|_|t|t sL|dk rF|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 transportlocationrnodesr3/tmp/pip-unpacked-wheel-04z2hmy8/zeep/xsd/schema.py__init__s  zSchema.__init__cCs|j}|rd|j|jfSdS)Nzz ) root_document _location_target_namespace)rZmain_docrrr__repr__/szSchema.__repr__cs2i|jfdd|jDS)Ncs"i|]\}}|kr||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 AszSchema.root_documentcCstdd|jDS)zABoolean to indicate if this schema contains any types or elementscss|] }|jVqdSr0)is_empty)r&documentrrrr3Hsz"Schema.is_empty..)allrr.rrrr5EszSchema.is_emptycCs |jSr0)rget_all_namespacesr.rrr namespacesJszSchema.namespacesccsBt}|jD]0}|jD] }|j|kr|V||jqq dS)zYYield all globla xsd.Type objects :rtype: Iterable of zeep.xsd.Element N)setr _elementsr%qnameadd)rseenr6elementrrrr Ns   zSchema.elementsccsBt}|jD]0}|jD] }|j|kr|V||jqq dS)z]Yield all global xsd.Type objects :rtype: Iterable of zeep.xsd.ComplexType N)r:r_typesr%r<r=)rr>r6type_rrrr \s   z Schema.types) schema_nodesrreturncCsDg}|D]}|||}||q|D] }|q(||_dSr0)create_new_documentappendresolve_create_prefix_mapr)rrBrZ resolve_queuerr6rrrrjs   zSchema.add_documents)urlrCcCs,t||j|jd}|j||d}|dS)N)r)rH)rrrrDrF)rrHZ schema_noder6rrradd_document_by_urlwszSchema.add_document_by_urlrCcCs||}||ddS)z7Return a global xsd.Element object with the given qname get_elementr?) _create_qname _get_instancerr<rrrrK|s zSchema.get_elementFc CsZ||}z||ddWStjk rT}z|rBtt|nW5d}~XYnXdS)zReturn a global xsd.Type object with the given qname :rtype: zeep.xsd.ComplexType or zeep.xsd.AnySimpleType get_typetypeN)rLrMrNamespaceErrorloggerdebugstr)rr< fail_silentlyexcrrrrOs zSchema.get_typecCs||ddS)z6Return a global xsd.Group object with the given qname. get_groupgrouprMrNrrrrWszSchema.get_groupcCs||ddS)z9Return a global xsd.attribute object with the given qname get_attribute attributerYrNrrrrZszSchema.get_attributecCs||ddS)z>Return a global xsd.attributeGroup object with the given qnameget_attribute_groupattributeGrouprYrNrrrr\szSchema.get_attribute_groupcCs||j|<dSr0)r)rprefix namespacerrr set_ns_prefixszSchema.set_ns_prefixcCsZz4z|j|WWStk r0|j|YWSXWn tk rTtd|YnXdS)NzNo such prefix %r)rKeyErrorr ValueError)rr^rrr get_ns_prefixszSchema.get_ns_prefixcCsX|jD]\}}||kr |Sq |jD]\}}||kr.|Sq.|dkrTdS|S)Nz)http://schemas.xmlsoap.org/soap/envelope/zsoap-env)rr-r)rr_r^Zother_namespacerrrget_shorthand_for_nss  zSchema.get_shorthand_for_nscCsR|dk r|dnd}|s|}|dkr*|}t|||}|j|||||S)z: :rtype: zeep.xsd.schema.SchemaDocument NZtargetNamespace)getSchemaDocumentrr=load)rrrHbase_urlZtarget_namespacer_schemarrrrDs   zSchema.create_new_documentcCs&|jD]}|j|q||_dS)z%Merge an other XSD schema in this oneN)rr=rGr)rrir6rrrmerges z Schema.mergecCs||j}|j||dS)N)ri)rKtagparse)rrelmrrr deserializes zSchema.deserializecCshttjdd}tjD]}|dd}||j|qtjD]}|}| |j |q8d|_ |j ||S)NT) is_global)rfrZXSDxsd_builtins_typesr@ register_typeZ_default_qnamexsd_builtins_elementsr;register_elementr<r1rr=)rriclsinstancerrrrs    zSchema._load_default_documentsc Cs||}zrd}||jD]P}t||}z||WWStjk rj}z|}WYqW5d}~XYqXq|dk rz|Wn.tjk rtd||j|jfYnXdS)z1Return an object from one of the SchemaDocument'sNzBUnable to resolve %s %s. No schema available for the namespace %r.)rL_get_schema_documentsr_getattrr LookupErrorrQtext)rr< method_namenameZlast_exceptionrimethodrVrrrrMs$   zSchema._get_instancecCst|tjr|S|ds~d|kr~|jr~|dd\}}||jkrTt|j||S||jkrpt|j||Std|n t|SdS)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 %rN)rrQName startswithrsplitrrb)rr{r^ localnamerrrrLs   zSchema._create_qnamecCsJddi}d}|jD].}|dks||kr0q||d|<|d7}q|S)NZxsdz http://www.w3.org/2001/XMLSchemarzns%dr)rr8r%)rr/ir_rrrrG s  zSchema._create_prefix_mapcCs:|j|s,|tjkr,td||||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_nsrZAUTO_IMPORT_NAMESPACESrRrSrIget_by_namespacerr_rUrrrrvs   zSchema._get_schema_documents)NNNN)F)NN)F))__name__ __module__ __qualname____doc__rr#propertyr/r r5r9r r typingListrZ_ElementrTrrI xsd_elementsElementrKrOGrouprW AttributerZAttributeGroupr\r`rcrdrDrjrnrrMrLrGrvrrrrr sH              r c@seZdZdZddZddZdddd d Zeje d d d Z ejdd ddZ e e ej ddddZ e edddZddZdS)rzUContainer instances to store multiple SchemaDocument objects per namespace. cCs t|_dSr0)r _instancesr.rrrr.sz_SchemaContainer.__init__ccs|D] }|VqdSr0r$)rr6rrr__iter__1s z_SchemaContainer.__iter__rfN)r6rCcCs2td|jt||j|jg}||dS)zAdd a schema documentz%Add document with tns %s to schema %sN)rRrSr_idr setdefaultrE)rr6rrrrr=5sz_SchemaContainer.addrJcCst|jSr0)rrkeysr.rrrr8=sz#_SchemaContainer.get_all_namespacescCs*||jkr |rgStd||j|S)Nz(No schema available for the namespace %r)rrrQrrrrr@s z!_SchemaContainer.get_by_namespace)r_rrCcCs.|j|dd}|D]}|j|kr|SqdS)z)r!r"r5r.rrrr#s zSchemaDocument.__repr__cCs|jSr0)r"r.rrrr_szSchemaDocument.namespacecCst|jp|jp|j Sr0)rrr@r;r.rrrr5szSchemaDocument.is_emptycCs:|dkr dS|j|js"tdt||}||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 Z visit_schema)rrirZvisitorrrrrgs zSchemaDocument.loadcCs|td||jrdSd|_|jD]}|D] }|q.q&dd}||j||j||j||j ||j dS)NzResolving in schema %sTc Ssz6|D](\}}|}|dk s*td|||<q WnDtjk rz}z$td|j|j|j|jdW5d}~XYnXdS)Nz!resolve() should return an objectzGUnable to resolve %(item_name)s %(qname)s in %(file)s. (via %(parent)s)) item_namer<fileparent)r-rFAssertionErrorrrxrr<r)valkeyobjnewrVrrr _resolve_dictsz-SchemaDocument.resolve.._resolve_dict) rRrSrrr%rFrrr;rr@)rschemasrirrrrrFs      zSchemaDocument.resolvecCs|j|g}||dS)zRegister an import for an other schema document. :type namespace: str :type schema: zeep.xsd.schema.SchemaDocument N)rrrE)rr_rirrrrregister_importszSchemaDocument.register_importcCs ||jkSr0)rrrrr is_importedszSchemaDocument.is_imported)r<valuecCs||||jddS)z"Register a xsd.Type in this schemarPN)_add_componentr@rr<rrrrrqszSchemaDocument.register_typecCs||||jddS)z%Register a xsd.Element in this schemar?N)rr;rrrrrsszSchemaDocument.register_elementcCs||||jddS)z#Register a xsd:Group in this schemarXN)rrrrrrregister_groupszSchemaDocument.register_groupcCs||||jddS)z'Register a xsd:Attribute in this schemar[N)rrrrrrregister_attributesz!SchemaDocument.register_attributeN)r<rrCcCs||||jddS)z,Register a xsd:AttributeGroup in this schemaZattribute_groupN)rrrrrrregister_attribute_groupsz'SchemaDocument.register_attribute_group)r<cCs|||jdS)zsReturn a xsd.Type object from this schema :rtype: zeep.xsd.ComplexType or zeep.xsd.AnySimpleType rP)_get_componentr@rNrrrrOszSchemaDocument.get_typerJcCs|||jdS)z,Return a xsd.Element object from this schemar?)rr;rNrrrrKszSchemaDocument.get_elementcCs|||jdS)z*Return a xsd.Group object from this schemarX)rrrNrrrrWszSchemaDocument.get_groupcCs|||jdS)z.Return a xsd.Attribute object from this schemar[)rrrNrrrrZszSchemaDocument.get_attributecCs|||jdS)z3Return a xsd.AttributeGroup object from this schemar])rrrNrrrr\sz"SchemaDocument.get_attribute_groupcCs.t|tjr|j}td||||||<dS)Nzregister_%s(%r, %r))rrrryrRrS)rr{rr-rrrrrs zSchemaDocument._add_componentc Csbz ||WStk r\d|}tjd||d|j|j|pDdd|||jdYnXdS)Nz, zpNo %(item_name)s '%(localname)s' in namespace %(namespace)s. Available %(item_name_plural)s are: %(known_items)ssz - )rZitem_name_pluralrr_ known_items)r<rr)rajoinrrrxrr_r!)rr<r-rrrrrr s"  zSchemaDocument._get_component)#rrrrrr#rr_r5rgrFrrrr xsd_typesTyperqrrrsrrrTrrrrrOrKrWrZr\rrrrrrrfcs4  %  rf) loggingr collectionsrlxmlrZzeeprrZ zeep.loaderrZ zeep.settingsrZzeep.xsdrr rr rZzeep.xsd.elementsr rrZzeep.xsd.typesrpZzeep.xsd.visitorr getLoggerrrRr rrfrrrrs$           ;