)Jf  dZgdZddlZddlZddlZddlmZddlmZddlm Z m Z e Z e Z ddl mZGdd eZGd d eZd Zd ZdZdZdZdZedZedZedZedZedZedZedZdZdZ d'dZ!dZ"ej#Gdde$Z%e%Z&dZ'd Z(d!Z)d"Z*d#Z+d$Z,d%Z-d&Z.dS)(z+ ElementTree interface to an XRD document. ) nsTagmkXRDTagisXRDS parseXRDSgetCanonicalID getYadisXRDgetPriorityStrict getPriorityprioSort iterServices expandServiceexpandServicesN)datetime)strptime)importElementTreeimportSafeElementTree)xriceZdZdZdZdS) XRDSErrorz An error with the XRDS document.N)__name__ __module__ __qualname____doc__reasonU/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/openid/yadis/etxrd.pyrr$s**FFFrrceZdZdZdS) XRDSFraudzbRaised when there's an assertion in the XRDS that it does not have the authority to make. N)rrrrrrrrr+srrc t|tr|dn|}t|}t |}t |std|S#ttttf$rt$r}td}||_ |d}~wwxYw)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. utf8zNot an XRDS documentzError parsing document as XMLN) isinstancestrencodeSafeElementTreeXML ElementTreerr SystemExit MemoryErrorAssertionError ImportError Exceptionr)text bytestringelementtreewhyexcs rrr1s -7tS,A,AKT[[(((t !%%j11&&w//d|| 4233 3   ^[ A 788  sAB'C)CCzxri://$xrd*($v*2.0)z xri://$xrdscd|d|S)N{}r)nsts rrrSsrr11 rc,tt|S)zybasestring -> basestring Create a tag name in the XRD 2.0 XML namespace suitable for using with ElementTree )r XRD_NS_2_0r7s rrrWs Q  rc,tt|S)zvbasestring -> basestring Create a tag name in the XRDS XML namespace suitable for using with ElementTree )rXRDS_NSr:s r mkXRDSTagr=`s !  rXRDSServiceXRDTypeURIExpires CanonicalIDcJ|}|jtkS)z"Is this document an XRDS document?)getroottagroot_tag)xrd_treeroots rrrus     D 8x rcfd}|tD]}|td|S)z=Return the XRD element that should contain the Yadis servicesNzNo XRD present in tree)findallxrd_tagrrIxrds rrr{sD C((    {0111 Jrc|t}||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_times rgetXRDExpirationrYsP"&&{33O(- 0DEE ac*++rc|t}| tj|dt dj}n#t$rYdSwxYw|}|ddD]x}| ddd}tj| t }||krt|d||}ytj |}tj ||st|d||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 rN!z can not come from z can not come from root )rLrMreverserXRIcanonicalID_tagr- IndexErrorlowerrsplitfindtextr rootAuthorityproviderIsAuthoritative) inamerIxrd_list canonicalIDchildIDrO parent_soughtparentrJs rrrsP((H ghqk11/BB1EJKK tt!!G|  sA..q1 o6677 FLLNN * *&&IJJ J  U # #D  &tW 5 5J'''44HIII s=A.. A<;A<ceZdZdZdZdZdS)_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. c,t||jSNr" __class__selfothers r__lt__z _Max.__lt__%000rc,t||jSrorprrs r__eq__z _Max.__eq__rvrN)rrrrrurxrrrrmrms<11111111rrmc|d}|&t|}|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. priorityNrz-Priority values must be non-negative integers)getint ValueErrorMax)r/prio_strprio_vals rrrsJ{{:&&Hx== q==OLMM M JrcP t|S#t$r tcYSwxYw)ztGet the priority of this element Returns Max if no priority is specified or the priority value is invalid. )rr}r~)r/s rr r s9  )))  s %%cZtj|t|t}|S)z5Sort a list of elements that have priority attributes)key)randomshufflesortedr )elements sorted_elemss rr r s, N8( 444L rcnt|}t|tS)zUReturn an iterable over the Service elements in the Yadis XRD sorted by priority)rr rL service_tagrNs rr r s+ h  C CKK ,, - --rcddt|tDS)z]Given a Service element, return a list of the contents of all URI tags in priority order.cg|] }|j Srr-).0 uri_elements r zsortedURIs..s+        r)r rLuri_tagservice_elements r sortedURIsr s9  #O$;$;G$D$DEE   rcJd|tDS)zKGiven a Service element, return a list of the contents of all Type tagscg|] }|j Srr)r type_elements rrzgetTypeURIs..s(   *    r)rLtype_tagrs r getTypeURIsrs1  .=.E.Eh.O.O   rct|}|sdg}g}|D])}t|}||||f*|S)zeTake a service element and expand it into an iterator of: ([type_uri], uri, service_element) N)rrappend)rurisexpandeduri type_uriss rr r sb o & &D vH;;00 C9:::: OrcXg}|D]$}|t|%|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_elementsrrs rr r .s:H+88 o667777 Orro)/r__all__sysr functoolsrtimeropenid.oidutilrrr'r% openid.yadisrr,rrrr9r<rrr=rHrrMrrrSr_rrrYrtotal_orderingobjectrmr~rr r r rrr r rrrrs    CCCCCCCC!! ''))   <#     9V  hy!! (5// 8F   (5//hy!! (=))   ",,,,2"""J  1 1 1 1 16 1 1 1 dff&...    r