iframe-proxy | Sunbelt Computer Software

This repository was archived by the owner on Nov 22, 2024. It is now read-only.

151 lines (109 loc) · 4.17 KB

# -*- coding: utf-8 -*-
"""This module provides decorators to the rest of the library."""
from functools import wraps
from requests.models import Response
try:  # (No coverage)
    # python2
    from StringIO import StringIO  # (No coverage)
except ImportError:  # (No coverage)
    # python3
    from io import BytesIO as StringIO
class RequestsStringIO(StringIO):
    """Shim compatibility for string IO."""
    def read(self, n=-1, *args, **kwargs):
        """Ignore extra args and kwargs."""
        # StringIO is an old-style class, so can't use super
        return StringIO.read(self, n)
def requires_auth(func):
    """Decorator to note which object methods require authorization."""
    @wraps(func)
    def auth_wrapper(self, *args, **kwargs):
        if hasattr(self, "session") and self.session.has_auth():
            return func(self, *args, **kwargs)
        else:
            from .exceptions import error_for
            # Mock a 401 response
            r = generate_fake_error_response(
                '{"message": "Requires authentication"}'
            raise error_for(r)
    return auth_wrapper
def requires_basic_auth(func):
    """Specific (basic) authentication decorator.
    This is used to note which object methods require username/password
    authorization and won't work with token based authorization.
    @wraps(func)
    def auth_wrapper(self, *args, **kwargs):
        if hasattr(self, "session") and self.session.auth:
            return func(self, *args, **kwargs)
        else:
            from .exceptions import error_for
            # Mock a 401 response
            r = generate_fake_error_response(
                '{"message": "Requires username/password authentication"}'
            raise error_for(r)
    return auth_wrapper
def requires_app_credentials(func):
    """Require client_id and client_secret to be associated.
    This is used to note and enforce which methods require a client_id and
    client_secret to be used.
    @wraps(func)
    def auth_wrapper(self, *args, **kwargs):
        client_id, client_secret = self.session.retrieve_client_credentials()
        if client_id and client_secret:
            return func(self, *args, **kwargs)
        else:
            from .exceptions import error_for
            # Mock a 401 response
            r = generate_fake_error_response(
                '{"message": "Requires username/password authentication"}'
            raise error_for(r)
    return auth_wrapper
def requires_app_bearer_auth(func):
    """Require the use of application authentication.
    .. versionadded:: 1.2.0
    @wraps(func)
    def auth_wrapper(self, *args, **kwargs):
        from . import session
        if isinstance(self.session.auth, session.AppBearerTokenAuth):
            return func(self, *args, **kwargs)
        else:
            from . import exceptions
            raise exceptions.MissingAppBearerAuthentication(
                "This method requires GitHub App authentication."
    return auth_wrapper
def requires_app_installation_auth(func):
    """Require the use of App's installation authentication.
    .. versionadded:: 1.2.0
    @wraps(func)
    def auth_wrapper(self, *args, **kwargs):
        from . import session
        if isinstance(self.session.auth, session.AppInstallationTokenAuth):
            return func(self, *args, **kwargs)
        else:
            from . import exceptions
            raise exceptions.MissingAppInstallationAuthentication(
                "This method requires GitHub App authentication."
    return auth_wrapper
def generate_fake_error_response(msg, status_code=401, encoding="utf-8"):
    """Generate a fake Response from requests."""
    r = Response()
    r.status_code = status_code
    r.encoding = encoding
    r.raw = RequestsStringIO(msg.encode())
    r._content_consumed = True
    r._content = r.raw.read()
    return r
# Use mock decorators when generating documentation, so all functino signatures
# are displayed correctly
if os.getenv("GENERATING_DOCUMENTATION", None) == "github3":
    requires_auth = requires_basic_auth = lambda x: x  # noqa  # (No coverage)

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

decorators.py

Latest commit

History

decorators.py

File metadata and controls