コード例 #1
0
    def process_request(self, request):

        request.twitter = Session.from_request(request)

        # If is_secure is True (should always be true except for local dev),
        # we will be redirected back to https and all cookies set will be
        # secure only.
        is_secure = settings.TWITTER_COOKIE_SECURE

        ssl_url = url(request, {'scheme': 'https' if is_secure else 'http'})

        if request.REQUEST.get('twitter_delete_auth'):
            request.twitter = Session()
            return http.HttpResponseRedirect(url(request))

        elif request.twitter.authed:
            request.twitter.api = Twython(settings.TWITTER_CONSUMER_KEY,
                                          settings.TWITTER_CONSUMER_SECRET,
                                          request.twitter.key,
                                          request.twitter.secret,
                                          ssl_verify=True)
        else:
            verifier = request.GET.get('oauth_verifier')
            if verifier:
                # We are completing an OAuth login

                request_key = request.COOKIES.get(REQUEST_KEY_NAME)
                request_secret = request.COOKIES.get(REQUEST_SECRET_NAME)

                if (validate_token(request_key)
                        and validate_token(request_secret)):

                    t = Twython(settings.TWITTER_CONSUMER_KEY,
                                settings.TWITTER_CONSUMER_SECRET,
                                request_key,
                                request_secret,
                                ssl_verify=True)

                    try:
                        tokens = t.get_authorized_tokens(verifier)
                    except (TwythonError, TwythonAuthError):
                        log.warning('Twython Error with verifier token')
                        pass
                    else:
                        # Override path to drop query string.
                        ssl_url = url(
                            request, {
                                'scheme': 'https' if is_secure else 'http',
                                'path': request.path
                            })
                        response = http.HttpResponseRedirect(ssl_url)

                        request.twitter.screen_name = tokens['screen_name']

                        Session(tokens['oauth_token'],
                                tokens['oauth_token_secret']).save(
                                    request, response)

                        return response
                else:
                    # request tokens didn't validate
                    log.warning("Twitter Oauth request tokens didn't validate")

            elif request.REQUEST.get('twitter_auth_request'):
                # We are requesting Twitter auth
                t = Twython(settings.TWITTER_CONSUMER_KEY,
                            settings.TWITTER_CONSUMER_SECRET,
                            ssl_verify=True)
                try:
                    auth_props = t.get_authentication_tokens(
                        callback_url=ssl_url)
                except (TwythonError, TwythonAuthError):
                    log.warning('Twython error while getting authorization '
                                'url')
                else:
                    response = http.HttpResponseRedirect(
                        auth_props['auth_url'])
                    response.set_cookie(REQUEST_KEY_NAME,
                                        auth_props['oauth_token'],
                                        secure=is_secure)
                    response.set_cookie(REQUEST_SECRET_NAME,
                                        auth_props['oauth_token_secret'],
                                        secure=is_secure)
                    return response
コード例 #2
0
    def process_request(self, request):

        request.twitter = Session.from_request(request)

        # If is_secure is True (should always be true except for local dev),
        # we will be redirected back to https and all cookies set will be
        # secure only.
        is_secure = settings.TWITTER_COOKIE_SECURE

        ssl_url = url(request, {"scheme": "https" if is_secure else "http"})

        try:
            if request.GET.get("twitter_delete_auth"):
                request.twitter = Session()
                return http.HttpResponseRedirect(url(request))
        except IOError:
            # Django throws an IOError if there's a bad read when parsing
            # the request data. If it throws an error, this is not a good
            # request and there's certainly no twitter stuff here.
            return

        if request.twitter.authed:
            request.twitter.api = get_twitter_api(
                request.twitter.key, request.twitter.secret
            )
            return

        verifier = request.GET.get("oauth_verifier")
        if verifier:
            # We are completing an OAuth login

            request_key = request.COOKIES.get(REQUEST_KEY_NAME)
            request_secret = request.COOKIES.get(REQUEST_SECRET_NAME)

            if validate_token(request_key) and validate_token(request_secret):
                t = get_twitter_api(request_key, request_secret)

                try:
                    tokens = t.get_authorized_tokens(verifier)
                except (TwythonError, TwythonAuthError):
                    log.warning("Twython Error with verifier token")
                    pass
                else:
                    # Override path to drop query string.
                    ssl_url = url(
                        request,
                        {
                            "scheme": "https" if is_secure else "http",
                            "path": request.path,
                        },
                    )
                    response = http.HttpResponseRedirect(ssl_url)

                    request.twitter.screen_name = tokens["screen_name"]

                    Session(tokens["oauth_token"], tokens["oauth_token_secret"]).save(
                        request, response
                    )

                    return response
            else:
                # request tokens didn't validate
                log.warning("Twitter Oauth request tokens didn't validate")

        elif request.GET.get("twitter_auth_request"):
            # We are requesting Twitter auth
            t = get_twitter_api(None, None)
            try:
                auth_props = t.get_authentication_tokens(callback_url=ssl_url)
            except (TwythonError, TwythonAuthError):
                log.warning("Twython error while getting authorization " "url")
            else:
                response = http.HttpResponseRedirect(auth_props["auth_url"])
                response.set_cookie(
                    REQUEST_KEY_NAME, auth_props["oauth_token"], secure=is_secure
                )
                response.set_cookie(
                    REQUEST_SECRET_NAME,
                    auth_props["oauth_token_secret"],
                    secure=is_secure,
                )
                return response
コード例 #3
0
ファイル: middleware.py プロジェクト: Acidburn0zzz/kitsune
    def process_request(self, request):

        request.twitter = Session.from_request(request)

        # If is_secure is True (should always be true except for local dev),
        # we will be redirected back to https and all cookies set will be
        # secure only.
        is_secure = settings.TWITTER_COOKIE_SECURE

        ssl_url = url(request, {'scheme': 'https' if is_secure else 'http'})

        try:
            if request.REQUEST.get('twitter_delete_auth'):
                request.twitter = Session()
                return http.HttpResponseRedirect(url(request))
        except IOError:
            # Django throws an IOError if there's a bad read when parsing
            # the request data. If it throws an error, this is not a good
            # request and there's certainly no twitter stuff here.
            return

        if request.twitter.authed:
            request.twitter.api = Twython(settings.TWITTER_CONSUMER_KEY,
                                          settings.TWITTER_CONSUMER_SECRET,
                                          request.twitter.key,
                                          request.twitter.secret,
                                          ssl_verify=True)
            return

        verifier = request.GET.get('oauth_verifier')
        if verifier:
            # We are completing an OAuth login

            request_key = request.COOKIES.get(REQUEST_KEY_NAME)
            request_secret = request.COOKIES.get(REQUEST_SECRET_NAME)

            if (validate_token(request_key) and
                    validate_token(request_secret)):

                t = Twython(settings.TWITTER_CONSUMER_KEY,
                            settings.TWITTER_CONSUMER_SECRET,
                            request_key,
                            request_secret,
                            ssl_verify=True)

                try:
                    tokens = t.get_authorized_tokens(verifier)
                except (TwythonError, TwythonAuthError):
                    log.warning('Twython Error with verifier token')
                    pass
                else:
                    # Override path to drop query string.
                    ssl_url = url(
                        request,
                        {'scheme': 'https' if is_secure else 'http',
                         'path': request.path})
                    response = http.HttpResponseRedirect(ssl_url)

                    request.twitter.screen_name = tokens['screen_name']

                    Session(
                        tokens['oauth_token'],
                        tokens['oauth_token_secret']).save(request,
                                                           response)

                    return response
            else:
                # request tokens didn't validate
                log.warning("Twitter Oauth request tokens didn't validate")

        elif request.REQUEST.get('twitter_auth_request'):
            # We are requesting Twitter auth
            t = Twython(settings.TWITTER_CONSUMER_KEY,
                        settings.TWITTER_CONSUMER_SECRET,
                        ssl_verify=True)
            try:
                auth_props = t.get_authentication_tokens(
                    callback_url=ssl_url)
            except (TwythonError, TwythonAuthError):
                log.warning('Twython error while getting authorization '
                            'url')
            else:
                response = http.HttpResponseRedirect(
                    auth_props['auth_url'])
                response.set_cookie(
                    REQUEST_KEY_NAME,
                    auth_props['oauth_token'],
                    secure=is_secure)
                response.set_cookie(
                    REQUEST_SECRET_NAME,
                    auth_props['oauth_token_secret'],
                    secure=is_secure)
                return response