o ˷e@sxgdZddlmZddlmZddlmZmZmZddl m Z m Z Gddde Z Gdd d eZd d Zd d ZdS))discoverDiscoveryResultDiscoveryFailure)StringIO)fetchers)YADIS_HEADER_NAMEYADIS_CONTENT_TYPEYADIS_ACCEPT_HEADER) MetaNotFound findHTMLMetac@seZdZdZdZddZdS)rzBRaised when a YADIS protocol error occurs in the discovery processNcCst||||_dS)N) Exception__init__ http_response)selfmessagerrL/var/www/ideatree/venv/lib/python3.10/site-packages/openid/yadis/discover.pyr s  zDiscoveryFailure.__init__)__name__ __module__ __qualname____doc__ identity_urlr rrrrr s rc@s<eZdZdZdZdZdZdZdZddZ ddZ ddZ dS) rz:Contains the result of performing Yadis discovery on a URINcCs ||_dS)zgInitialize the state of the object sets all attributes to None except the request_uri N) request_uri)rrrrrr )s zDiscoveryResult.__init__cCs|jdurdS|j|jkS)z*Was the Yadis protocol's indirection used?NF)xrds_urinormalized_urirrrrusedYadisLocation0s  z!DiscoveryResult.usedYadisLocationcCs|p|jtkS)z5Is the response text supposed to be an XRDS document?)r content_typerrrrrisXRDS6szDiscoveryResult.isXRDS) rrrrrrrr response_textr rrrrrrrs rcCst|}tj|dtid}|jdvrtd|jf||j|_|j d|_ t ||_ |j rS| rSt|j }|jdvrLtd|jf|}|j|_||j d|_ |j|_|S)aDiscover services for a given URI. @param uri: The identity URI as a well-formed http or https URI. The well-formedness and the protocol are not checked, but the results of this function are undefined if those properties do not hold. @return: DiscoveryResult object @raises Exception: Any exception that can be raised by fetching a URL with the given fetcher. @raises DiscoveryFailure: When the HTTP response does not have a 200 code. Accept)headers)zEHTTP Response status from identity URL host is not 200. Got status %r content-typez>HTTP Response status from Yadis host is not 200. Got status %r)rrfetchr statusr final_urlrr!getr whereIsYadisrrrbodyr)uriresultrespexcrrrr<s4    rc Cs*|jd}|r|dddtkr|jS|jt}|s|p$d}|dd}t|dkrG|d drG|dddd }nd }t |j t rzz|j |}Wn%tyyz|j d }Wntyv|j |d }YnwYnw|j }z tt|}W|SttfyY|Sw|S) zGiven a HTTPResponse, return the location of the Yadis document. May be the URL just retrieved, another URL, or None if no suitable URL can be found. [non-blocking] @returns: str or None r$;rzcharset==zutf-8replace)r!r(splitlowerrr'rrsplitlenstrip startswith isinstancer*bytesdecode UnicodeErrorr rr )r-r yadis_locencodingcontentrrrr)hsB      r)N)__all__ioropenidropenid.yadis.constantsrrr openid.yadis.parsehtmlr r r robjectrrr)rrrrs   & ,