Source code for aap_client.public_keys

"""Utilities to load public keys from files"""
from cryptography.x509 import load_pem_x509_certificate as load_pem
from cryptography.x509 import load_der_x509_certificate as load_der
from cryptography.hazmat.backends import default_backend


def load_from_pem(pem_filename):
    """Opens an X509 certificate in PEM format and returns its public key"""
    with open(pem_filename, 'r') as cert_file:
        cert = load_pem(cert_file.read().encode(),
                        default_backend())
        key = cert.public_key()
    return key

def load_from_der(der_filename):
    """Opens an X509 certificate in DER format and returns its public key"""
    with open(der_filename, 'rb') as cert_file:
        cert = load_der(cert_file.read(),
                        default_backend())
        key = cert.public_key()
    return key