Synnefo

# or implied, of GRNET S.A.

from django import http

from django.utils import simplejson as json

from django.conf import settings

import base64

class PublicKeyPairResourceView(rest.UserResourceView):

    model = PublicKeyPair

    exclude_fields = ["user"]

class PublicKeyPairCollectionView(rest.UserCollectionView):

    model = PublicKeyPair

    exclude_fields = ["user"]

SSH_KEY_LENGTH = getattr(settings, 'USERDATA_SSH_KEY_LENGTH', 2048)

def generate_key_pair(request):

    """

    Response to generate private/public RSA key pair

        raise Exception("Application does not support ssh keys generation")

    if PublicKeyPair.user_limit_exceeded(request.user):

    # generate RSA key

    from Crypto import Random

    Random.atfork()

    # get PEM string

    pem = exportKey(key, 'PEM')

    data = {'private': pem, 'public': public}

    return http.HttpResponse(json.dumps(data), mimetype="application/json")

def download_private_key(request):

    """

    Return key contents

    response['Content-Disposition'] = 'attachment; filename=%s' % name

    response.write(data)

    return response