• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• asap-authentication-python-master
  • ChangeLog
  • AUTHORS
  • .mailmap
  • LICENSE
  • test-requirements.txt
  • README.rst
  • setup.py
  • atlassian_jwt_auth
    • verifier.py
    • exceptions.py
    • key.py
    • frameworks
      • common
        • decorators.py
        • asap.py
        • __init__.py
        • utils.py
        • tests
          • test_utils.py
          • test_decorators.py
          • __init__.py
        • backend.py
      • django
        • middleware.py
        • decorators.py
        • __init__.py
        • tests
          • views.py
          • urls.py
          • __init__.py
          • settings.py
          • test_django.py
        • backend.py
      • __init__.py
      • wsgi
        • middleware.py
        • __init__.py
        • tests
          • test_wsgi.py
        • backend.py
      • flask
        • decorators.py
        • __init__.py
        • tests
          • test_flask.py
          • __init__.py
        • backend.py
    • __init__.py
    • algorithms.py
    • tests
      • test_key.py
      • test_signer.py
      • test_verifier.py
      • test_signer_private_key_repo.py
      • test_public_key_provider.py
      • __init__.py
      • utils.py
      • test_private_key_provider.py
    • auth.py
    • contrib
      • requests.py
      • flask_app
        • decorators.py
        • __init__.py
      • aiohttp
        • verifier.py
        • key.py
        • __init__.py
        • auth.py
      • django
        • middleware.py
        • decorators.py
        • __init__.py
      • __init__.py
      • README.md
      • tests
        • aiohttp
          • test_verifier.py
          • test_public_key_provider.py
          • test_auth.py
          • __init__.py
        • test_requests.py
        • __init__.py
        • utils.py
      • server
        • __init__.py
    • signer.py
  • setup.cfg
  • .travis.yml
  • requirements.txt
  • .gitignore

from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.asymmetric import rsa
from cryptography.hazmat.primitives.asymmetric import ec
from cryptography.hazmat.primitives import serialization

import atlassian_jwt_auth


def get_new_rsa_private_key_in_pem_format():
    """ returns a new rsa key in pem format. """
    private_key = rsa.generate_private_key(
        key_size=2048, backend=default_backend(), public_exponent=65537)
    return private_key.private_bytes(
        encoding=serialization.Encoding.PEM,
        format=serialization.PrivateFormat.TraditionalOpenSSL,
        encryption_algorithm=serialization.NoEncryption()
    )


def get_public_key_pem_for_private_key_pem(private_key_pem):
    private_key = serialization.load_pem_private_key(
        private_key_pem,
        password=None,
        backend=default_backend()
    )
    public_key = private_key.public_key()
    return public_key.public_bytes(
        encoding=serialization.Encoding.PEM,
        format=serialization.PublicFormat.SubjectPublicKeyInfo
    )


def get_example_jwt_auth_signer(**kwargs):
    """ returns an example jwt_auth_signer instance. """
    issuer = kwargs.get('issuer', 'egissuer')
    key_id = kwargs.get('key_id', '%s/a' % issuer)
    key = kwargs.get(
        'private_key_pem', get_new_rsa_private_key_in_pem_format())
    algorithm = kwargs.get('algorithm', 'RS256')
    return atlassian_jwt_auth.create_signer(
        issuer, key_id, key, algorithm=algorithm)


def create_token(issuer, audience, key_id, private_key, subject=None):
    """" returns a token based upon the supplied parameters. """
    signer = atlassian_jwt_auth.create_signer(
        issuer, key_id, private_key, subject=subject)
    return signer.generate_jwt(audience)


class BaseJWTAlgorithmTestMixin(object):

    """ A mixin class to make testing different support for different
        jwt algorithms easier.
    """

    def get_new_private_key_in_pem_format(self):
        """ returns a new private key in pem format. """
        raise NotImplementedError("not implemented.")


class RS256KeyTestMixin(object):

    """ Private rs256 test mixin. """

    @property
    def algorithm(self):
        return 'RS256'

    def get_new_private_key_in_pem_format(self):
        return get_new_rsa_private_key_in_pem_format()


class ES256KeyTestMixin(object):

    """ Private es256 test mixin. """

    @property
    def algorithm(self):
        return 'ES256'

    def get_new_private_key_in_pem_format(self):
        private_key = ec.generate_private_key(
            ec.SECP256R1(), default_backend())
        return private_key.private_bytes(
            encoding=serialization.Encoding.PEM,
            format=serialization.PrivateFormat.TraditionalOpenSSL,
            encryption_algorithm=serialization.NoEncryption()
        )