File: instagram.py

package info (click to toggle)
python-requests-oauthlib 1.3.1-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 496 kB
  • sloc: python: 2,000; makefile: 159
file content (26 lines) | stat: -rw-r--r-- 948 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
try:
    from urlparse import urlparse, parse_qs
except ImportError:
    from urllib.parse import urlparse, parse_qs

from oauthlib.common import add_params_to_uri


def instagram_compliance_fix(session):
    def _non_compliant_param_name(url, headers, data):
        # If the user has already specified the token in the URL
        # then there's nothing to do.
        # If the specified token is different from ``session.access_token``,
        # we assume the user intends to override the access token.
        url_query = dict(parse_qs(urlparse(url).query))
        token = url_query.get("access_token")
        if token:
            # Nothing to do, just return.
            return url, headers, data

        token = [("access_token", session.access_token)]
        url = add_params_to_uri(url, token)
        return url, headers, data

    session.register_compliance_hook("protected_request", _non_compliant_param_name)
    return session