o ˷e @sfdZgdZddlZddlZddlZddlmZddlmZddlm Z m Z e Z e Z ddl mZGdd d eZGd d d eZd d ZdZdZddZddZddZedZedZedZedZedZedZedZddZdd Z d7d!d"Z!d#d$Z"ej#Gd%d&d&e$Z%e%Z&d'd(Z'd)d*Z(d+d,Z)d-d.Z*d/d0Z+d1d2Z,d3d4Z-d5d6Z.dS)8z+ ElementTree interface to an XRD document. ) nsTagmkXRDTagisXRDS parseXRDSgetCanonicalID getYadisXRDgetPriorityStrict getPriorityprioSort iterServices expandServiceexpandServicesN)datetime)strptime)importElementTreeimportSafeElementTree)xric@seZdZdZdZdS) XRDSErrorz An error with the XRDS document.N)__name__ __module__ __qualname____doc__reasonrrI/var/www/ideatree/venv/lib/python3.10/site-packages/openid/yadis/etxrd.pyr$src@seZdZdZdS) XRDSFraudzbRaised when there's an assertion in the XRDS that it does not have the authority to make. N)rrrrrrrrr+src Cszt|tr |dn|}t|}Wn ttttfyt y3}z t d}||_ |d}~wwt |}t |sAt d|S)zParse the given text as an XRDS document. @return: ElementTree containing an XRDS document @raises XRDSError: When there is a parse error or the document does not contain an XRDS. utf8zError parsing document as XMLNzNot an XRDS document) isinstancestrencodeSafeElementTreeXML SystemExit MemoryErrorAssertionError ImportError Exceptionrr ElementTreer)text bytestringelementwhyexctreerrrr1s rzxri://$xrd*($v*2.0)z xri://$xrdscCs d||fS)Nz{%s}%sr)nstrrrrS rcC tt|S)zybasestring -> basestring Create a tag name in the XRD 2.0 XML namespace suitable for using with ElementTree )r XRD_NS_2_0r/rrrrW rcCr1)zvbasestring -> basestring Create a tag name in the XRDS XML namespace suitable for using with ElementTree )rXRDS_NSr3rrr mkXRDSTag`r4r6XRDSServiceXRDTypeURIExpires CanonicalIDcCs|}|jtkS)z"Is this document an XRDS document?)getroottagroot_tag)xrd_treerootrrrrus rcCs(d}|tD]}q|durtd|S)z=Return the XRD element that should contain the Yadis servicesNzNo XRD present in tree)findallxrd_tagrrAxrdrrrr{s rcCs6|t}|dur |S|j}t|d}t|ddS)aReturn the expiration date of this XRD element, or None if no expiration was specified. @type xrd_element: ElementTree node @param default: The value to use as the expiration if no expiration was specified in the XRD. @rtype: datetime.datetime @raises ValueError: If the xrd:Expires element is present, but its contents are not formatted according to the specification. Nz%Y-%m-%dT%H:%M:%SZr )find expires_tagr(rr) xrd_elementdefaultexpires_elementexpires_string expires_timerrrgetXRDExpirations  rOc Cs|t}|zt|dtdj}Wn ty"YdSw|}|ddD]"}| ddd}t| t}||krMt d||f|}q-t |}t ||sct d||f|S)aReturn the CanonicalID from this XRDS document. @param iname: the XRI being resolved. @type iname: unicode @param xrd_tree: The XRDS output from the resolver. @type xrd_tree: ElementTree @returns: The XRI CanonicalID or None. @returntype: unicode or None r N!z%r can not come from %sz%r can not come from root %r)rCrDreverserXRIcanonicalID_tagr( IndexErrorlowerrsplitfindtextr rootAuthorityproviderIsAuthoritative) inamerAxrd_list canonicalIDchildIDrF parent_soughtparentrBrrrrs$     rc@s eZdZdZddZddZdS)_Maxz Value that compares greater than any other value. Should only be used as a singleton. Implemented for use as a priority value for when a priority is not specified. cC t||jSNr __class__selfotherrrr__lt__r0z _Max.__lt__cCrbrcrdrfrrr__eq__r0z _Max.__eq__N)rrrrrirjrrrrras racCs2|d}|durt|}|dkr|StdtS)zGet the priority of this element. Raises ValueError if the value of the priority is invalid. If no priority is specified, it returns a value that compares greater than any other value. priorityNr z-Priority values must be non-negative integers)getint ValueErrorMax)r*prio_strprio_valrrrrs rcCs"zt|WStytYSw)ztGet the priority of this element Returns Max if no priority is specified or the priority value is invalid. )rrnro)r*rrrrs   rcCst|t|td}|S)z5Sort a list of elements that have priority attributes)key)randomshufflesortedr)elements sorted_elemsrrrr s  r cCst|}t|tS)zUReturn an iterable over the Service elements in the Yadis XRD sorted by priority)rr rC service_tagrErrrr sr cCsddt|tDS)z]Given a Service element, return a list of the contents of all URI tags in priority order.cSg|]}|jqSrr().0 uri_elementrrr szsortedURIs..)r rCuri_tagservice_elementrrr sortedURIs s rcCsdd|tDS)zKGiven a Service element, return a list of the contents of all Type tagscSryrrz)r{ type_elementrrrr}szgetTypeURIs..)rCtype_tagrrrr getTypeURIssrcCs<t|}|s dg}g}|D]}t|}||||fq |S)zeTake a service element and expand it into an iterator of: ([type_uri], uri, service_element) N)rrappend)rurisexpandeduri type_urisrrrr sr cCs g}|D] }|t|q|S)aTake a sorted iterator of service elements and expand it into a sorted iterator of: ([type_uri], uri, service_element) There may be more than one item in the resulting list for each service element if there is more than one URI or type for a service, but each triple will be unique. If there is no URI or Type for a Service element, it will not appear in the result. )extendr )service_elementsrrrrrr .s r rc)/r__all__sysrs functoolsrtimeropenid.oidutilrrr'r openid.yadisrr&rrrr2r5rrr6r@rxrDrr~rIrTrrrOrtotal_orderingobjectrarorrr r rrr r rrrrsR      %