API Integration
Please review carefully.
This commit is contained in:
Modulos
parent
b7e8eceb25
commit
130c00c8ee
19 changed files with 310 additions and 890 deletions
|
|
@ -1,40 +0,0 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
# Generated by Django 1.9.4 on 2017-05-09 14:36
|
||||
from __future__ import unicode_literals
|
||||
|
||||
from django.db import migrations, models
|
||||
import django.db.models.deletion
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
initial = True
|
||||
|
||||
dependencies = [
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.CreateModel(
|
||||
name='VirtualMachine',
|
||||
fields=[
|
||||
('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||
('opennebula_id', models.IntegerField()),
|
||||
],
|
||||
),
|
||||
migrations.CreateModel(
|
||||
name='VirtualMachineTemplate',
|
||||
fields=[
|
||||
('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||
('opennebula_id', models.IntegerField()),
|
||||
('base_price', models.FloatField()),
|
||||
('memory_price', models.FloatField()),
|
||||
('core_price', models.FloatField()),
|
||||
('disk_size_price', models.FloatField()),
|
||||
],
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name='virtualmachine',
|
||||
name='template',
|
||||
field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='opennebula_api.VirtualMachineTemplate'),
|
||||
),
|
||||
]
|
||||
|
|
@ -1,20 +0,0 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
# Generated by Django 1.9.4 on 2017-05-11 02:46
|
||||
from __future__ import unicode_literals
|
||||
|
||||
from django.db import migrations
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
('opennebula_api', '0001_initial'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.RenameField(
|
||||
model_name='virtualmachine',
|
||||
old_name='template',
|
||||
new_name='vm_template',
|
||||
),
|
||||
]
|
||||
|
|
@ -1,24 +0,0 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
# Generated by Django 1.9.4 on 2017-05-11 09:06
|
||||
from __future__ import unicode_literals
|
||||
|
||||
from django.conf import settings
|
||||
from django.db import migrations, models
|
||||
import django.db.models.deletion
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
|
||||
('opennebula_api', '0002_auto_20170511_0246'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.AddField(
|
||||
model_name='virtualmachine',
|
||||
name='owner',
|
||||
field=models.ForeignKey(default=1, on_delete=django.db.models.deletion.CASCADE, related_name='virtual_machines', to=settings.AUTH_USER_MODEL),
|
||||
preserve_default=False,
|
||||
),
|
||||
]
|
||||
|
|
@ -1,124 +1,11 @@
|
|||
import oca
|
||||
import socket
|
||||
|
||||
from django.db import models
|
||||
from django.conf import settings
|
||||
from django.utils.functional import cached_property
|
||||
|
||||
from membership.models import CustomUser
|
||||
|
||||
from oca.pool import WrongNameError
|
||||
|
||||
class VirtualMachineTemplate(models.Model):
|
||||
"""This class represents an opennebula template."""
|
||||
opennebula_id = models.IntegerField()
|
||||
base_price = models.FloatField()
|
||||
memory_price = models.FloatField()
|
||||
core_price = models.FloatField()
|
||||
disk_size_price = models.FloatField()
|
||||
|
||||
@cached_property
|
||||
def template(self):
|
||||
template = OpenNebulaManager()._get_template(self.opennebula_id)
|
||||
return template
|
||||
|
||||
def calculate_price(self):
|
||||
template = self.template.template
|
||||
|
||||
price = int(template.vcpu) * self.core_price
|
||||
price += int(template.memory) / 1024 * self.memory_price
|
||||
try:
|
||||
price += int(template.disk.size) / 1024 * self.disk_size_price
|
||||
except AttributeError:
|
||||
for disk in template.disks:
|
||||
price += int(disk.size) / 1024 * self.disk_size_price
|
||||
return price
|
||||
|
||||
def get_name(self):
|
||||
|
||||
template = self.template
|
||||
return template.name
|
||||
|
||||
def get_cores(self):
|
||||
|
||||
template = self.template.template
|
||||
return int(template.vcpu)
|
||||
|
||||
def get_disk_size(self):
|
||||
|
||||
template = self.template.template
|
||||
disk_size = 0
|
||||
for disk in template.disks:
|
||||
disk_size += int(disk.size)
|
||||
return disk_size / 1024
|
||||
|
||||
def get_memory(self):
|
||||
|
||||
template = self.template.template
|
||||
return int(template.memory) / 1024
|
||||
|
||||
class VirtualMachine(models.Model):
|
||||
"""This class represents an opennebula virtual machine."""
|
||||
opennebula_id = models.IntegerField()
|
||||
owner = models.ForeignKey(CustomUser, related_name='virtual_machines')
|
||||
vm_template = models.ForeignKey(VirtualMachineTemplate)
|
||||
|
||||
VM_STATE = {
|
||||
'0': 'INIT',
|
||||
'1': 'PENDING',
|
||||
'2': 'HOLD',
|
||||
'3': 'ACTIVE',
|
||||
'4': 'STOPPED',
|
||||
'5': 'SUSPENDED',
|
||||
'6': 'DONE',
|
||||
'8': 'POWEROFF',
|
||||
'9': 'UNDEPLOYED',
|
||||
'10': 'CLONING',
|
||||
'11': 'CLONING_FAILURE',
|
||||
}
|
||||
|
||||
@cached_property
|
||||
def vm(self):
|
||||
manager = OpenNebulaManager(email=self.owner.email,
|
||||
password=self.owner.password[0:20],
|
||||
create_user=True)
|
||||
vm = manager._get_vm(self.opennebula_id)
|
||||
return vm
|
||||
|
||||
def get_name(self):
|
||||
|
||||
return self.vm.name
|
||||
|
||||
def get_cores(self):
|
||||
|
||||
return self.vm_template.get_cores()
|
||||
|
||||
def get_disk_size(self):
|
||||
|
||||
return self.vm_template.get_disk_size()
|
||||
|
||||
def get_memory(self):
|
||||
|
||||
return self.vm_template.get_memory()
|
||||
|
||||
def get_id(self):
|
||||
|
||||
return self.vm.id
|
||||
|
||||
def get_ip(self):
|
||||
|
||||
try:
|
||||
return self.vm.user_template.ungleich_public_ip
|
||||
except AttributeError:
|
||||
return '-'
|
||||
|
||||
def get_state(self):
|
||||
|
||||
return self.VM_STATE.get(str(self.vm.state))
|
||||
|
||||
def get_price(self):
|
||||
return self.vm_template.calculate_price()
|
||||
|
||||
class OpenNebulaManager():
|
||||
"""This class represents an opennebula manager."""
|
||||
|
||||
|
|
@ -204,12 +91,15 @@ class OpenNebulaManager():
|
|||
raise ConnectionRefusedError
|
||||
return vm_pool
|
||||
|
||||
def get_vms(self):
|
||||
return self._get_vm_pool()
|
||||
|
||||
def _get_vm(self, vm_id):
|
||||
def get_vm(self, vm_id):
|
||||
vm_pool = self._get_vm_pool()
|
||||
return vm_pool.get_by_id(vm_id)
|
||||
return vm_pool.get_by_id(int(vm_id))
|
||||
|
||||
def create_vm(self, template_id):
|
||||
#TODO: get app with id
|
||||
def create_vm(self, template_id, app_id=None):
|
||||
vm_id = self.oneadmin_client.call(
|
||||
oca.VmTemplate.METHODS['instantiate'],
|
||||
template_id,
|
||||
|
|
@ -232,7 +122,7 @@ class OpenNebulaManager():
|
|||
self.oneadmin_client.call(
|
||||
oca.VirtualMachine.METHODS['action'],
|
||||
'terminate',
|
||||
vm_id
|
||||
int(vm_id)
|
||||
)
|
||||
|
||||
def _get_template_pool(self):
|
||||
|
|
@ -248,19 +138,31 @@ class OpenNebulaManager():
|
|||
raise ConnectionRefusedError
|
||||
return template_pool
|
||||
|
||||
def _get_template(self, template_id):
|
||||
def get_templates(self):
|
||||
public_templates = [
|
||||
template
|
||||
for template in self._get_template_pool()
|
||||
if 'public-' in template.name
|
||||
]
|
||||
return public_templates
|
||||
def get_template(self, template_id):
|
||||
template_pool = self._get_template_pool()
|
||||
return template_pool.get_by_id(template_id)
|
||||
|
||||
|
||||
|
||||
def create_template(self, name, cores, memory, disk_size):
|
||||
def create_template(self, name, cores, memory, disk_size, core_price, memory_price,
|
||||
disk_size_price, ssh='' ):
|
||||
"""Create and add a new template to opennebula.
|
||||
:param name: A string representation describing the template.
|
||||
Used as label in view.
|
||||
:param cores: Amount of virtual cpu cores for the VM.
|
||||
:param memory: Amount of RAM for the VM (MB)
|
||||
:param disk_size: Amount of disk space for VM (MB)
|
||||
:param memory: Amount of RAM for the VM (GB)
|
||||
:param disk_size: Amount of disk space for VM (GB)
|
||||
:param core_price: Price of virtual cpu for the VM per core.
|
||||
:param memory_price: Price of RAM for the VM per GB
|
||||
:param disk_size_price: Price of disk space for VM per GB
|
||||
:param ssh: User public ssh key
|
||||
"""
|
||||
template_string_formatter = """<TEMPLATE>
|
||||
<NAME>{name}</NAME>
|
||||
|
|
@ -272,6 +174,10 @@ class OpenNebulaManager():
|
|||
<SIZE>{size}</SIZE>
|
||||
<DEV_PREFIX>vd</DEV_PREFIX>
|
||||
</DISK>
|
||||
<CPU_COST>{cpu_cost}</CPU_COST>
|
||||
<MEMORY_COST>{memory_cost}</MEMORY_COST>
|
||||
<DISK_COST>{disk_cost}</DISK_COST>
|
||||
<SSH_PUBLIC_KEY>{ssh}</SSH_PUBLIC_KEY>
|
||||
</TEMPLATE>
|
||||
"""
|
||||
template_id = oca.VmTemplate.allocate(
|
||||
|
|
@ -281,7 +187,12 @@ class OpenNebulaManager():
|
|||
vcpu=cores,
|
||||
cpu=0.1*cores,
|
||||
size=1024 * disk_size,
|
||||
memory=1024 * memory
|
||||
memory=1024 * memory,
|
||||
# * 10 because we set cpu to *0.1
|
||||
cpu_cost=10*core_price,
|
||||
memory_cost=memory_price,
|
||||
disk_cost=disk_size_price,
|
||||
ssh=ssh
|
||||
)
|
||||
)
|
||||
|
||||
|
|
|
|||
|
|
@ -1,10 +0,0 @@
|
|||
from rest_framework.permissions import BasePermission
|
||||
|
||||
from .models import VirtualMachine
|
||||
|
||||
class IsOwner(BasePermission):
|
||||
|
||||
def has_object_permission(self, request, view, obj):
|
||||
if isinstance(obj, VirtualMachine):
|
||||
return obj.owner == request.user
|
||||
return obj.owner == request.user
|
||||
|
|
@ -3,98 +3,117 @@ import oca
|
|||
from rest_framework import serializers
|
||||
|
||||
from oca import OpenNebulaException
|
||||
from oca.template import VmTemplate
|
||||
|
||||
from .models import VirtualMachine, VirtualMachineTemplate, OpenNebulaManager
|
||||
from .models import OpenNebulaManager
|
||||
|
||||
class VirtualMachineTemplateSerializer(serializers.ModelSerializer):
|
||||
class VirtualMachineTemplateSerializer(serializers.Serializer):
|
||||
"""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')
|
||||
id = serializers.IntegerField(read_only=True)
|
||||
name = serializers.CharField()
|
||||
cores = serializers.IntegerField(source='template.vcpu')
|
||||
disk = serializers.IntegerField(write_only=True)
|
||||
disk_size = serializers.SerializerMethodField()
|
||||
memory = serializers.SerializerMethodField()
|
||||
core_price = serializers.FloatField(source='template.cpu_cost')
|
||||
disk_size_price = serializers.FloatField(source='template.disk_cost')
|
||||
memory_price = serializers.FloatField(source='template.memory_cost')
|
||||
price = serializers.SerializerMethodField()
|
||||
|
||||
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')
|
||||
def create(self, validated_data):
|
||||
data = validated_data
|
||||
template = data.pop('template')
|
||||
|
||||
cores = template.pop('vcpu')
|
||||
name = data.pop('name')
|
||||
disk_size = data.pop('disk')
|
||||
memory = template.pop('memory')
|
||||
core_price = template.pop('cpu_cost')
|
||||
memory_price = template.pop('memory_cost')
|
||||
disk_size_price = template.pop('disk_cost')
|
||||
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})
|
||||
disk_size=disk_size,
|
||||
core_price=core_price,
|
||||
disk_size_price=disk_size_price,
|
||||
memory_price=memory_price)
|
||||
except OpenNebulaException as err:
|
||||
raise serializers.ValidationError("OpenNebulaException occured. {0}".format(err))
|
||||
|
||||
return data
|
||||
return manager.get_template(template_id=opennebula_id)
|
||||
|
||||
def create(self, validated_data):
|
||||
return VirtualMachineTemplate.objects.create(**validated_data)
|
||||
def get_disk_size(self, obj):
|
||||
template = obj.template
|
||||
disk_size = 0
|
||||
for disk in template.disks:
|
||||
disk_size += int(disk.size)
|
||||
return disk_size / 1024
|
||||
|
||||
class TemplatePrimaryKeyRelatedField(serializers.PrimaryKeyRelatedField):
|
||||
def display_value(self, instance):
|
||||
return 'Template: {}'.format(instance.get_name())
|
||||
def get_price(self, obj):
|
||||
template = obj.template
|
||||
price = float(template.cpu) * float(template.cpu_cost)
|
||||
price += (int(template.memory)/1024 * float(template.memory_cost))
|
||||
for disk in template.disks:
|
||||
price += int(disk.size)/1024 * float(template.disk_cost)
|
||||
return price
|
||||
|
||||
class VirtualMachineSerializer(serializers.ModelSerializer):
|
||||
def get_memory(self, obj):
|
||||
return int(obj.template.memory)/1024
|
||||
|
||||
class VirtualMachineSerializer(serializers.Serializer):
|
||||
"""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')
|
||||
name = serializers.CharField(read_only=True)
|
||||
cores = serializers.IntegerField(read_only=True, source='template.vcpu')
|
||||
|
||||
owner = serializers.ReadOnlyField(source='owner.name')
|
||||
disk_size = serializers.SerializerMethodField()
|
||||
memory = serializers.SerializerMethodField()
|
||||
ip = serializers.CharField(read_only=True,
|
||||
source='user_template.ungleich_public_ip',
|
||||
default='-')
|
||||
vm_id = serializers.IntegerField(read_only=True, source='id')
|
||||
state = serializers.CharField(read_only=True, source='str_state')
|
||||
price = serializers.SerializerMethodField()
|
||||
|
||||
vm_template = VirtualMachineTemplateSerializer(read_only=True)
|
||||
|
||||
vm_template_id = TemplatePrimaryKeyRelatedField(
|
||||
queryset=VirtualMachineTemplate.objects.all(),
|
||||
source='vm_template'
|
||||
template_id = serializers.ChoiceField(
|
||||
choices=[(key.id, key.name) for key in
|
||||
OpenNebulaManager().get_templates()],
|
||||
source='template.template_id',
|
||||
write_only=True
|
||||
)
|
||||
|
||||
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', 'owner')
|
||||
read_only_fields = ('opennebula_id', )
|
||||
def create(self, validated_data):
|
||||
owner = validated_data['owner']
|
||||
template_id = validated_data['template']['template_id']
|
||||
|
||||
def validate(self, data):
|
||||
# Create the opennebula model
|
||||
|
||||
try:
|
||||
template_id = data['vm_template'].opennebula_id
|
||||
owner = self.context.get('request').user
|
||||
manager = OpenNebulaManager(email=owner.email,
|
||||
password=owner.password[0:20],
|
||||
create_user = True)
|
||||
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
|
||||
return manager.get_vm(opennebula_id)
|
||||
|
||||
def create(self, validated_data):
|
||||
return VirtualMachine.objects.create(**validated_data)
|
||||
def get_memory(self, obj):
|
||||
return int(obj.template.memory)/1024
|
||||
|
||||
def update(self, instance, validated_data):
|
||||
pass
|
||||
def get_disk_size(self, obj):
|
||||
template = obj.template
|
||||
disk_size = 0
|
||||
for disk in template.disks:
|
||||
disk_size += int(disk.size)
|
||||
return disk_size / 1024
|
||||
|
||||
def get_price(self, obj):
|
||||
template = obj.template
|
||||
price = float(template.cpu) * float(template.cpu_cost)
|
||||
price += (int(template.memory)/1024 * float(template.memory_cost))
|
||||
for disk in template.disks:
|
||||
price += int(disk.size)/1024 * float(template.disk_cost)
|
||||
return price
|
||||
|
||||
|
|
|
|||
|
|
@ -11,17 +11,20 @@ from guardian.mixins import PermissionRequiredMixin
|
|||
|
||||
from .serializers import VirtualMachineTemplateSerializer, \
|
||||
VirtualMachineSerializer
|
||||
from .models import VirtualMachineTemplate, VirtualMachine, OpenNebulaManager
|
||||
from .permissions import IsOwner
|
||||
from .models import OpenNebulaManager
|
||||
|
||||
|
||||
class TemplateCreateView(generics.ListCreateAPIView):
|
||||
"""This class handles the GET and POST requests."""
|
||||
|
||||
queryset = VirtualMachineTemplate.objects.all()
|
||||
serializer_class = VirtualMachineTemplateSerializer
|
||||
permission_classes = (permissions.IsAuthenticated, permissions.IsAdminUser)
|
||||
|
||||
def get_queryset(self):
|
||||
manager = OpenNebulaManager()
|
||||
return manager.get_templates()
|
||||
|
||||
|
||||
def perform_create(self, serializer):
|
||||
"""Save the post data when creating a new template."""
|
||||
serializer.save()
|
||||
|
|
@ -29,15 +32,24 @@ class TemplateCreateView(generics.ListCreateAPIView):
|
|||
class TemplateDetailsView(generics.RetrieveUpdateDestroyAPIView):
|
||||
"""This class handles the http GET, PUT and DELETE requests."""
|
||||
|
||||
queryset = VirtualMachineTemplate.objects.all()
|
||||
serializer_class = VirtualMachineTemplateSerializer
|
||||
permission_classes = (permissions.IsAuthenticated)
|
||||
|
||||
def get_queryset(self):
|
||||
manager = OpenNebulaManager()
|
||||
return manager.get_templates()
|
||||
|
||||
class VmCreateView(generics.ListCreateAPIView):
|
||||
"""This class handles the GET and POST requests."""
|
||||
queryset = VirtualMachine.objects.all()
|
||||
serializer_class = VirtualMachineSerializer
|
||||
permission_classes = (permissions.IsAuthenticated, IsOwner)
|
||||
permission_classes = (permissions.IsAuthenticated, )
|
||||
|
||||
def get_queryset(self):
|
||||
owner = self.request.user
|
||||
manager = OpenNebulaManager(email=owner.email,
|
||||
password=owner.password[0:20],
|
||||
create_user=True)
|
||||
return manager.get_vms()
|
||||
|
||||
def perform_create(self, serializer):
|
||||
"""Save the post data when creating a new template."""
|
||||
|
|
@ -45,16 +57,28 @@ class VmCreateView(generics.ListCreateAPIView):
|
|||
|
||||
class VmDetailsView(generics.RetrieveUpdateDestroyAPIView):
|
||||
"""This class handles the http GET, PUT and DELETE requests."""
|
||||
permission_classes = (permissions.IsAuthenticated, IsOwner)
|
||||
permission_classes = (permissions.IsAuthenticated, )
|
||||
|
||||
queryset = VirtualMachine.objects.all()
|
||||
serializer_class = VirtualMachineSerializer
|
||||
|
||||
def get_queryset(self):
|
||||
owner = self.request.user
|
||||
manager = OpenNebulaManager(email=owner.email,
|
||||
password=owner.password[0:20],
|
||||
create_user=True)
|
||||
return manager.get_vms()
|
||||
|
||||
def get_object(self):
|
||||
owner = self.request.user
|
||||
manager = OpenNebulaManager(email=owner.email,
|
||||
password=owner.password[0:20],
|
||||
create_user=True)
|
||||
return manager.get_vm(self.kwargs.get('pk'))
|
||||
|
||||
def perform_destroy(self, instance):
|
||||
owner = instance.owner
|
||||
owner = self.request.user
|
||||
manager = OpenNebulaManager(email=owner.email,
|
||||
password=owner.password[0:20],
|
||||
create_user = True)
|
||||
manager.delete_vm(instance.opennebula_id)
|
||||
instance.delete()
|
||||
manager.delete_vm(instance.id)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue