dynamicweb/opennebula_api/serializers.py

import oca

from rest_framework import serializers

from oca import OpenNebulaException

from .models import VirtualMachine, VirtualMachineTemplate, OpenNebulaManager

class VirtualMachineTemplateSerializer(serializers.ModelSerializer):
    """Serializer to map the virtual machine template instance into JSON format."""
    cores       = serializers.IntegerField(source='get_cores') 
    name        = serializers.CharField(source='get_name')
    disk_size   = serializers.IntegerField(source='get_disk_size')
    memory      = serializers.IntegerField(source='get_memory')

    class Meta:
        model = VirtualMachineTemplate
        fields = ('id', 'name', 'cores', 'memory', 'disk_size', 'base_price', 
                  'core_price', 'memory_price', 'disk_size_price', 'opennebula_id')
        read_only_fields = ('opennebula_id', )

    def validate(self, data):
        # Create the opennebula model
        cores   = data.pop('get_cores')
        name    = data.pop('get_name')
        disk_size = data.pop('get_disk_size')
        memory  = data.pop('get_memory')

        manager = OpenNebulaManager(create_user = False)
        
        try:
            opennebula_id = manager.create_template(name=name, cores=cores,
                                                    memory=memory,
                                                    disk_size=disk_size)
            data.update({'opennebula_id':opennebula_id})
        except OpenNebulaException as err:
            raise serializers.ValidationError("OpenNebulaException occured. {0}".format(err))
        
        return data

    def create(self, validated_data):
        return VirtualMachineTemplate.objects.create(**validated_data)

class TemplatePrimaryKeyRelatedField(serializers.PrimaryKeyRelatedField):
    def display_value(self, instance):
        return 'Template: {}'.format(instance.get_name())

class VirtualMachineSerializer(serializers.ModelSerializer):
    """Serializer to map the virtual machine instance into JSON format."""

    #TODO: Maybe we can change to template.get_cores
    cores       = serializers.IntegerField(read_only=True, source='get_cores') 
    name        = serializers.CharField(read_only=True, source='get_name')
    disk_size   = serializers.IntegerField(read_only=True, source='get_disk_size')
    memory      = serializers.IntegerField(read_only=True, source='get_memory')
    #TODO: See if we can change to IPAddressField
    ip          = serializers.CharField(read_only=True, source='get_ip')
    deploy_id   = serializers.IntegerField(read_only=True, source='get_deploy_id')
    vm_id          = serializers.IntegerField(read_only=True, source='get_vm_id')
    state       = serializers.CharField(read_only=True, source='get_state')
    price       = serializers.FloatField(read_only=True, source='get_price')

    vm_template = VirtualMachineTemplateSerializer(read_only=True)

    vm_template_id = TemplatePrimaryKeyRelatedField(
                queryset=VirtualMachineTemplate.objects.all(),
                source='vm_template'
            )

    class Meta:
        model = VirtualMachine
        fields = ('id', 'opennebula_id', 'vm_template', 'vm_template_id', 'cores', 'name',
                'disk_size', 'memory', 'ip', 'deploy_id', 'state', 'vm_id',
                'price')
        read_only_fields = ('opennebula_id', )

    def validate(self, data):
        # Create the opennebula model
        manager = OpenNebulaManager(create_user = False)
        
        try:
            template_id = data['vm_template'].opennebula_id
            opennebula_id = manager.create_vm(template_id)
            data.update({'opennebula_id':opennebula_id})
        except OpenNebulaException as err:
            raise serializers.ValidationError("OpenNebulaException occured. {0}".format(err))
        
        return data

    def create(self, validated_data):
        return VirtualMachine.objects.create(**validated_data)