o ˷e@s$GdddeZGdddeZdS)c@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)YadisServiceManagerzHolds the state of a list of selected Yadis services, managing storing it in a session and iterating over the services in order.cCs&||_||_t||_||_d|_dSN) starting_url yadis_urllistservices session_key_current)selfrrrrr K/var/www/ideatree/venv/lib/python3.10/site-packages/openid/yadis/manager.py__init__s   zYadisServiceManager.__init__cCs t|jS)z!How many untried services remain?)lenrr r r r __len__ zYadisServiceManager.__len__cCs|Srr rr r r __iter__szYadisServiceManager.__iter__cCs*z |jd|_W|jStytw)zReturn the next service self.current() will continue to return that service until the next call to this method.)rpopr IndexError StopIterationrr r r __next__s  zYadisServiceManager.__next__cCs|jS)zYReturn the current service. Returns None if there are no services left. rrr r r current'szYadisServiceManager.currentcCs||j|jfvSr)rr)r urlr r r forURL.szYadisServiceManager.forURLcCs |jduS)z$Has the first service been returned?Nrrr r r started1rzYadisServiceManager.startedcCs|||j<dS)z5Store this object in the session, by its session key.N)r)r sessionr r r store5szYadisServiceManager.storeN) __name__ __module__ __qualname____doc__r rrrrrrrr r r r rs  rc@sZeZdZdZdZdZdddZddZdd d Zd d Z dddZ dddZ dddZ dS) DiscoveryacState management for discovery. High-level usage pattern is to call .getNextService(discover) in order to find the next available service for this user for this session. Once a request completes, call .finish() to clean up the session state. @ivar session: a dict-like object that stores state unique to the requesting user-agent. This object must be able to store serializable objects. @ivar url: the URL that is used to make the discovery request @ivar session_key_suffix: The suffix that will be used to identify this object in the session object. auth_yadis_services_NcCs$||_||_|dur |j}||_dS)zInitialize a discovery objectN)rrDEFAULT_SUFFIXsession_key_suffix)r rrr&r r r r Os  zDiscovery.__init__cCs^|}|dur|s||s||j\}}|||}|r+t|}||j|Sd}|S)a:Return the next authentication service for the pair of user_input and session. This function handles fallback. @param discover: a callable that takes a URL and returns a list of services @type discover: str -> [service] @return: the next available service N) getManagerdestroyManagerr createManagernextrr)r discovermanagerrrservicer r r getNextServiceXs   zDiscovery.getNextServiceFcCs4|j|d}|dur|}|j|d|Sd}|S)atClean up Yadis-related services in the session and return the most-recently-attempted service from the manager, if one exists. @param force: True if the manager should be deleted regardless of whether it's a manager for self.url. @return: current service endpoint object or None if there is no current service forceN)r'rr()r r0r,r-r r r cleanupus  zDiscovery.cleanupcCs |j|jS)zsGet the session key for this starting URL and suffix @return: The session key @rtype: str )PREFIXr&rr r r getSessionKeys zDiscovery.getSessionKeycCs0|j|}|dur||js|r|SdS)a>Extract the YadisServiceManager for this object's URL and suffix from the session. @param force: True if the manager should be returned regardless of whether it's a manager for self.url. @return: The current YadisServiceManager, if it's for this URL, or else None N)rgetr3rr)r r0r,r r r r's zDiscovery.getManagercCsJ|}|rtd||jf|sdSt|j|||}||j|S)zCreate a new YadisService Manager for this starting URL and suffix, and store it in the session. @raises KeyError: When I already have a manager. @return: A new YadisServiceManager or None z$There is already a %r manager for %rN)r3r'KeyErrorrrrr)r rrkeyr,r r r r)s zDiscovery.createManagercCs(|j|ddur|}|j|=dSdS)aTDelete any YadisServiceManager with this starting URL and suffix from the session. If there is no service manager or the service manager is for a different URL, it silently does nothing. @param force: True if the manager should be deleted regardless of whether it's a manager for self.url. r/N)r'r3r)r r0r6r r r r(s  zDiscovery.destroyManagerr)F) rrr r!r%r2r r.r1r3r'r)r(r r r r r":s    r"N)objectrr"r r r r s9