Synnefo

class ServiceUnavailable(Fault):

    code = 503

#

# Copyright (c) 2010 Greek Research and Technology Network

#

from synnefo.api.util import *

from synnefo.db.models import Flavor

from django.conf.urls.defaults import *

from django.http import HttpResponse

from django.template.loader import render_to_string

urlpatterns = patterns('synnefo.api.flavors',

    (r'^(?:/|.json|.xml)?$', 'list_flavors'),

    (r'^/detail(?:.json|.xml)?$', 'list_flavors', {'detail': True}),

    (r'^/(\d+)(?:.json|.xml)?$', 'get_flavor_details'),

)

def flavor_to_dict(flavor, detail=True):

    d = {'id': flavor.id, 'name': flavor.name}

    if detail:

        d['ram'] = flavor.ram

        d['disk'] = flavor.disk

    return d

@api_method('GET')

def list_flavors(request, detail=False):

    # Normal Response Codes: 200, 203

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badRequest (400),

    #                       overLimit (413)

    all_flavors = Flavor.objects.all()

    flavors = [flavor_to_dict(flavor, detail) for flavor in all_flavors]

    if request.type == 'xml':

        mimetype = 'application/xml'

        data = render_to_string('list_flavors.xml', {'flavors': flavors, 'detail': detail})

    else:

        mimetype = 'application/json'

        data = json.dumps({'flavors': {'values': flavors}})

    return HttpResponse(data, mimetype=mimetype, status=200)

@api_method('GET')

def get_flavor_details(request, flavor_id):

    # Normal Response Codes: 200, 203

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badRequest (400),

    #                       itemNotFound (404),

    #                       overLimit (413)

    try:

        falvor_id = int(flavor_id)

        flavor = flavor_to_dict(Flavor.objects.get(id=flavor_id))

    except Flavor.DoesNotExist:

        raise ItemNotFound

    if request.type == 'xml':

        data = render_to_string('flavor.xml', {'flavor': flavor})

    else:

        data = json.dumps({'flavor': flavor})

    return HttpResponse(data, status=200)

#

# Copyright (c) 2010 Greek Research and Technology Network

#

from synnefo.api.util import *

from synnefo.db.models import Image

from django.conf.urls.defaults import *

from django.http import HttpResponse

from django.template.loader import render_to_string

urlpatterns = patterns('synnefo.api.images',

    (r'^(?:/|.json|.xml)?$', 'demux'),

    (r'^/detail(?:.json|.xml)?$', 'list_images', {'detail': True}),

    (r'^/(\d+)(?:.json|.xml)?$', 'image_demux'),

)

def demux(request):

    if request.method == 'GET':

        return list_images(request)

    elif request.method == 'POST':

        return create_image(request)

    else:

        fault = BadRequest()

        return render_fault(request, fault)

def image_demux(request, image_id):

    if request.method == 'GET':

        return get_image_details(request, image_id)

    elif request.method == 'DELETE':

        return delete_image(request, image_id)

    else:

        fault = BadRequest()

        return render_fault(request, fault)

def image_to_dict(image, detail=True):

    d = {'id': image.id, 'name': image.name}

    if detail:

        d['updated'] = image.updated.isoformat()

        d['created'] = image.created.isoformat()

        d['status'] = image.state

        d['progress'] = 100 if image.state == 'ACTIVE' else 0

        d['description'] = image.description

        if image.sourcevm:

            d['serverRef'] = image.sourcevm.id

    return d

@api_method('GET')

def list_images(request, detail=False):

    # Normal Response Codes: 200, 203

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badRequest (400),

    #                       overLimit (413)

    all_images = Image.objects.all()

    images = [image_to_dict(image, detail) for image in all_images]

    if request.type == 'xml':

        mimetype = 'application/xml'

        data = render_to_string('list_images.xml', {'images': images, 'detail': detail})

    else:

        mimetype = 'application/json'

        data = json.dumps({'images': {'values': images}})

    return HttpResponse(data, mimetype=mimetype, status=200)

@api_method('POST')

def create_image(request):

    # Normal Response Code: 202

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badMediaType(415),

    #                       itemNotFound (404),

    #                       badRequest (400),

    #                       serverCapacityUnavailable (503),

    #                       buildInProgress (409),

    #                       resizeNotAllowed (403),

    #                       backupOrResizeInProgress (409),

    #                       overLimit (413)

    req = get_request_dict(request)

    owner = get_user()

    try:

        d = req['image']

        server_id = int(d['serverRef'])

        vm = VirtualMachine.objects.get(id=server_id)

        image = Image.objects.create(name=d['name'], size=0, owner=owner, sourcevm=vm)

        image.save()

    except KeyError:

        raise BadRequest

    except ValueError:

        raise BadRequest

    except VirtualMachine.DoesNotExist:

        raise ItemNotFound

    imagedict = image_to_dict(image)

    if request.type == 'xml':

        data = render_to_string('image.xml', {'image': imagedict})

    else:

        data = json.dumps({'image': imagedict})

    return HttpResponse(data, status=202)

@api_method('GET')

def get_image_details(request, image_id):

    # Normal Response Codes: 200, 203

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badRequest (400),

    #                       itemNotFound (404),

    #                       overLimit (413)

    try:

        image_id = int(image_id)

        imagedict = image_to_dict(Image.objects.get(id=image_id))

    except Image.DoesNotExist:

        raise ItemNotFound

    if request.type == 'xml':

        data = render_to_string('image.xml', {'image': imagedict})

    else:

        data = json.dumps({'image': imagedict})

    return HttpResponse(data, status=200)

@api_method('DELETE')

def delete_image(request, image_id):

    # Normal Response Code: 204

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       itemNotFound (404),

    #                       overLimit (413)

    try:

        image_id = int(image_id)

        image = Image.objects.get(id=image_id)

    except Image.DoesNotExist:

        raise ItemNotFound

    if image.owner != get_user():

        raise Unauthorized()

    image.delete()

    return HttpResponse(status=204)

        fault = BadRequest()

        return render_fault(request, fault)

        fault = BadRequest()

        return render_fault(request, fault)

def server_to_dict(server, detail=False):

    d = dict(id=server.id, name=server.name)

        d['status'] = server.rsapi_state

        d['progress'] = 100 if server.rsapi_state == 'ACTIVE' else 0

        d['hostId'] = server.hostid

        d['updated'] = server.updated.isoformat()

        d['created'] = server.created.isoformat()

        d['flavorId'] = server.flavor.id            # XXX Should use flavorRef instead?

        d['imageId'] = server.sourceimage.id        # XXX Should use imageRef instead?

        d['description'] = server.description       # XXX Not in OpenStack docs

        server_meta = server.virtualmachinemetadata_set.all()

        metadata = dict((meta.meta_key, meta.meta_value) for meta in server_meta)

        public_addrs = [dict(version=4, addr=server.ipfour), dict(version=6, addr=server.ipsix)]

def render_server(request, serverdict, status=200):

        data = render_to_string('server.xml', dict(server=serverdict, is_root=True))

        data = json.dumps({'server': serverdict})

    return HttpResponse(data, status=status)

@api_method('GET')

    user_servers = VirtualMachine.objects.filter(owner=owner, deleted=False)

    servers = [server_to_dict(server, detail) for server in user_servers]

        data = json.dumps({'servers': {'values': servers}})

    return HttpResponse(data, status=200)

@api_method('POST')

    server = VirtualMachine.objects.create(

        name = server.backend_id

    server.save()

    serverdict = server_to_dict(server, detail=True)

    serverdict['status'] = 'BUILD'

    serverdict['adminPass'] = random_password()

    return render_server(request, serverdict, status=202)

@api_method('GET')

    # Normal Response Codes: 200, 203

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badRequest (400),

    #                       itemNotFound (404),

    #                       overLimit (413)

        server_id = int(server_id)

        server = VirtualMachine.objects.get(id=server_id)

        raise ItemNotFound

    serverdict = server_to_dict(server, detail=True)

    return render_server(request, serverdict)

@api_method('PUT')

def update_server_name(request, server_id):

    # Normal Response Code: 204

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badRequest (400),

    #                       badMediaType(415),

    #                       itemNotFound (404),

    #                       buildInProgress (409),

    #                       overLimit (413)

        server_id = int(server_id)

        server = VirtualMachine.objects.get(id=server_id)

        raise ItemNotFound

    server.name = name

    server.save()

@api_method('DELETE')

    # Normal Response Codes: 204

    # Error Response Codes: computeFault (400, 500),

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       itemNotFound (404),

    #                       unauthorized (401),

    #                       buildInProgress (409),

    #                       overLimit (413)

        server_id = int(server_id)

        server = VirtualMachine.objects.get(id=server_id)

        raise ItemNotFound

    server.start_action('DESTROY')

    rapi.DeleteInstance(server.backend_id)

<?xml version="1.0" encoding="UTF-8"?>

<flavor xmlns="http://docs.openstack.org/compute/api/v1.1" xmlns:atom="http://www.w3.org/2005/Atom" id="{{ flavor.id }}" name="{{ flavor.name }}" ram="{{ flavor.ram }}" disk="{{ flavor.disk }}">

</flavor>

<?xml version="1.0" encoding="UTF-8"?>

<image xmlns="http://docs.openstack.org/compute/api/v1.1" xmlns:atom="http://www.w3.org/2005/Atom" id="{{ image.id }}" name="{{ image.name }}" serverRef="{{ image.serverRef }}" updated="{{ image.updated }}" created="{{ image.created }}" status="{{ image.status }}" progress="{{ image.progress }}"> </image>

{% spaceless %}

<?xml version="1.0" encoding="UTF-8"?>

<flavors xmlns="http://docs.openstack.org/compute/api/v1.1" xmlns:atom="http://www.w3.org/2005/Atom">

  {% for flavor in flavors %}  

  <flavor id="{{ flavor.id}}" name="{{ flavor.name }}"{% if detail %} ram="{{ flavor.ram }}" disk="{{ flavor.disk }}"{% endif %}>

  </flavor>

  {% endfor %}

</flavors>

{% endspaceless %}

{% spaceless %}

<?xml version="1.0" encoding="UTF-8"?>

<images xmlns="http://docs.openstack.org/compute/api/v1.1" xmlns:atom="http://www.w3.org/2005/Atom">

  {% for image in images %}

  <image id="{{ image.id }}" name="{{ image.name }}"{% if detail %} updated="{{ image.updated }}" created="{{ image.created }}" status="{{ image.status }}"{% endif %}>

  </image>

  {% endfor %}

</images>

{% endspaceless %}

    (r'^flavors', include('synnefo.api.flavors')),

    (r'^images', include('synnefo.api.images')),

    # XXX Quick and dirty

def render_fault(request, fault):

def api_method(http_method):

    def decorator(func):

        @wraps(func)

        def wrapper(request, *args, **kwargs):

            try:

                if request.path.endswith('.json'):

                    type = 'json'

                elif request.path.endswith('.xml'):

                    type = 'xml'

                elif request.META.get('HTTP_ACCEPT', None) == 'application/xml':

                    type = 'xml'

                else:

                    type = 'json'

                request.type = type

                if request.method != http_method:

                    raise BadRequest()

                resp = func(request, *args, **kwargs)

                resp['Content-Type'] = 'application/xml' if type == 'xml' else 'application/json'

                return resp

            except Fault, fault:

                return render_fault(request, fault)

            except Exception, e:

                log.exception('Unexpected error: %s' % e)

                fault = ServiceUnavailable()

                return render_fault(request, fault)

        return wrapper

    return decorator