Synnefo

#

# Copyright (c) 2010 Greek Research and Technology Network

#

from synnefo.api.common import method_not_allowed

from synnefo.api.faults import BadRequest, Unauthorized

from synnefo.api.util import (isoformat, isoparse, get_user, get_vm, get_image, get_image_meta,

                                get_request_dict, render_metadata, render_meta, api_method)

from synnefo.db.models import Image, ImageMetadata

from django.conf.urls.defaults import patterns

from django.http import HttpResponse

from django.template.loader import render_to_string

from django.utils import simplejson as json

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

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

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

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

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

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

)

def demux(request):

    if request.method == 'GET':

        return list_images(request)

    elif request.method == 'POST':

        return create_image(request)

    else:

        return method_not_allowed(request)

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:

        return method_not_allowed(request)

def metadata_demux(request, image_id):

    if request.method == 'GET':

        return list_metadata(request, image_id)

    elif request.method == 'POST':

        return update_metadata(request, image_id)

    else:

        return method_not_allowed(request)

def metadata_item_demux(request, image_id, key):

    if request.method == 'GET':

        return get_metadata_item(request, image_id, key)

    elif request.method == 'PUT':

        return create_metadata_item(request, image_id, key)

    elif request.method == 'DELETE':

        return delete_metadata_item(request, image_id, key)

    else:

        return method_not_allowed(request)

def image_to_dict(image, detail=True):

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

    if detail:

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

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

        d['status'] = image.state

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

        if image.sourcevm:

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

        metadata = {}

        for meta in ImageMetadata.objects.filter(image=image):

            metadata[meta.meta_key] = meta.meta_value

        if metadata:

            d['metadata'] = {'values': metadata}

    return d

def metadata_to_dict(image):

    image_meta = image.imagemetadata_set.all()

    return dict((meta.meta_key, meta.meta_value) for meta in image_meta)

@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)

    since = isoparse(request.GET.get('changes-since'))

    if since:

        avail_images = Image.objects.filter(updated__gte=since)

        if not avail_images:

            return HttpResponse(status=304)

    else:

        avail_images = Image.objects.all()

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

    if request.serialization == 'xml':

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

    else:

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

    return HttpResponse(data, 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)

    try:

        d = req['image']

        server_id = d['serverRef']

        name = d['name']

    except (KeyError, ValueError):

        raise BadRequest('Malformed request.')

    owner = get_user()

    vm = get_vm(server_id)

    image = Image.objects.create(name=name, owner=owner, sourcevm=vm)

    imagedict = image_to_dict(image)

    if request.serialization == '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)

    image = get_image(image_id)

    imagedict = image_to_dict(image)

    if request.serialization == '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)

    image = get_image(image_id)

    if image.owner != get_user():

        raise Unauthorized('Image does not belong to user.')

    image.delete()

    return HttpResponse(status=204)

@api_method('GET')

def list_metadata(request, image_id):

    # Normal Response Codes: 200, 203

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

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badRequest (400),

    #                       overLimit (413)

    image = get_image(image_id)

    metadata = metadata_to_dict(image)

    return render_metadata(request, metadata, use_values=True, status=200)

@api_method('POST')

def update_metadata(request, image_id):

    # Normal Response Code: 201

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

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       badRequest (400),

    #                       buildInProgress (409),

    #                       badMediaType(415),

    #                       overLimit (413)

    image = get_image(image_id)

    req = get_request_dict(request)

    try:

        metadata = req['metadata']

        assert isinstance(metadata, dict)

    except (KeyError, AssertionError):

        raise BadRequest('Malformed request.')

    updated = {}

    for key, val in metadata.items():

        try:

            meta = ImageMetadata.objects.get(meta_key=key, image=image)

            meta.meta_value = val

            meta.save()

            updated[key] = val

        except ImageMetadata.DoesNotExist:

            pass    # Ignore non-existent metadata

    return render_metadata(request, updated, status=201)

@api_method('GET')

def get_metadata_item(request, image_id, key):

    # Normal Response Codes: 200, 203

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

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       itemNotFound (404),

    #                       badRequest (400),

    #                       overLimit (413)

    meta = get_image_meta(image_id, key)

    return render_meta(request, meta, status=200)

@api_method('PUT')

def create_metadata_item(request, image_id, key):

    # Normal Response Code: 201

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

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       itemNotFound (404),

    #                       badRequest (400),

    #                       buildInProgress (409),

    #                       badMediaType(415),

    #                       overLimit (413)

    image = get_image(image_id)

    req = get_request_dict(request)

    try:

        metadict = req['meta']

        assert isinstance(metadict, dict)

        assert len(metadict) == 1

        assert key in metadict

    except (KeyError, AssertionError):

        raise BadRequest('Malformed request.')

    meta, created = ImageMetadata.objects.get_or_create(meta_key=key, image=image)

    meta.meta_value = metadict[key]

    meta.save()

    return render_meta(request, meta, status=201)

@api_method('DELETE')

def delete_metadata_item(request, image_id, key):

    # Normal Response Code: 204

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

    #                       serviceUnavailable (503),

    #                       unauthorized (401),

    #                       itemNotFound (404),

    #                       badRequest (400),

    #                       buildInProgress (409),

    #                       badMediaType(415),

    #                       overLimit (413),

    meta = get_image_meta(image_id, key)

    meta.delete()

    return HttpResponse(status=204)