Question

我正在龙卷风中使用OAuth2身份验证的应用程序。登录类如下：

class IDPLogin(tornado.web.RequestHandler, IDPLoginHandler):
    async def get(self):
        if self.get_argument('code', False):
            access = await self.get_authenticated_user(
                ...
            )
            print(type(access))
            #self.set_secure_cookie('user', self.get_argument('code'))
            #self.redirect('/')
        else:
            await self.authorize_redirect(
                ...
            )

使用get_authenticated_user方法，如下所示（两个额外的回调，用于获取评估用户所需的所有详细信息）：

class IDPHubLoginHandler(tornado.auth.OAuth2Mixin):
    def __init__(self):
        self._OAUTH_AUTHORIZE_URL = "..."
        self._OAUTH_ACCESS_TOKEN_URL = "..."
        self._USERINFO_ENDPOINT = "..."

    async def get_authenticated_user(self, redirect_uri, client_id, client_secret, code):
        http = self.get_auth_http_client()
        body = urllib_parse.urlencode({
            "redirect_uri": redirect_uri,
            "code": code,
            "client_id": client_id,
            "client_secret": client_secret,
            "grant_type": "authorization_code",
        })
        fut = http.fetch(self._OAUTH_ACCESS_TOKEN_URL,
                         method="POST",
                         headers={'Content-Type': 'application/x-www-form-urlencoded'},
                         body=body)
        fut.add_done_callback(wrap(functools.partial(self._on_access_token)))

    def _on_access_token(self, future):
        """Callback function for the exchange to the access token."""
        try:
            response = future.result()
        except Exception as e:
            future.set_exception(AuthError('IDP auth error: %s' % str(e)))
            return

        args = escape.json_decode(response.body)
        # Fetch userinfo
        http = self.get_auth_http_client()
        fut = http.fetch(self._USERINFO_ENDPOINT,
                         method="GET",
                         headers={
                             'Authorization': 'Bearer ' + args['access_token'],
                             'accept': 'application/json'
                         }
        )
        fut.add_done_callback(wrap(functools.partial(self._on_userinfo)))

    def _on_userinfo(self, future):
        response = future.result()
        r_body = escape.json_decode(response.body)
        return r_body

我希望能够访问_on_userinfo回调中返回的正文，但是Login类中的 access 是'NoneType'，并且我希望评估响应以拒绝访问或向用户显示一块饼干。

呈现的代码成功获取了所有必需的输入数据，但是我努力了解如何从回调返回值并在主登录类（IDPLogin）中重用它们。我浏览了龙卷风文档，找不到答案。 Oauth2 / OpenID示例充其量只是非常简短。

尝试对将来的结果进行set_result会导致asyncio.base_futures.InvalidStateError。

Answer 1

找到了另一种方法。不确定这是否是最规范的方法，但似乎可以完成工作。

实施Oauth2 Mixin如下：

class IDPLoginHandler(tornado.auth.OAuth2Mixin):
...
    async def get_authenticated_user(self, redirect_uri, client_id, client_secret, code):
        #do fetching and return future result
    async def get_user_info(self, access_token):
        #do fetching and return future result

从主登录处理程序中使用await关键字依次调用方法：

class IDPLogin(tornado.web.RequestHandler, IDPLoginHandler):
    async def get(self):
        if self.get_argument('code', False):
            response_token = await self.get_authenticated_user(
                ...
            )
            token = escape.json_decode(response_token.body)['access_token']
            response_userinfo = await self.get_user_info(token)

龙卷风-从等待中获取返回值（多个回调）

1 个答案: