dynamicweb/datacenterlight/models.py

from django.db import models


class BetaAccessVMType(models.Model):
    ssd = models.IntegerField()
    ram = models.IntegerField()
    cpu = models.IntegerField()
    price = models.FloatField()

    def __str__(self):
        return "ID: %s - SSD %s - RAM %s - CPU %s - Price %s " % \
               (self.id, str(self.ssd), self.ram, self.cpu, self.price)


class BetaAccess(models.Model):
    email = models.CharField(max_length=250)
    name = models.CharField(max_length=250)

    # vm = models.ForeignKey(BetaAccessVM)

    def __str__(self):
        vms = self.betaaccessvm_set.all()
        rep = "Email: %s " % self.email
        for vm in vms:
            rep += "(vm:%s - amount:%s) - " % (vm.type.id, vm.amount)
        return rep


class BetaAccessVM(models.Model):
    type = models.ForeignKey(BetaAccessVMType)
    access = models.ForeignKey(BetaAccess)
    amount = models.IntegerField()

    @classmethod
    def create(cls, data):
        VM_KEY_ID = 0
        VM_AMOUNT = 1
        ZERO = 0
        email = data.get('email')
        beta_access = BetaAccess.objects.create(email=email)
        vm_data = [(key, value) for key, value in data.items() if 'vm' in key]
        created_vms = []
        for vm in vm_data:
            if int(vm[VM_AMOUNT]) == ZERO:
                continue
            vm_id = vm[VM_KEY_ID].split('-').pop()
            vm_type = BetaAccessVMType.objects.get(id=vm_id)
            created_vms.append(cls.objects.create(access=beta_access,
                                                  amount=vm[VM_AMOUNT], type=vm_type))

        return created_vms


class VMTemplate(models.Model):
    name = models.CharField(max_length=50)
    opennebula_vm_template_id = models.IntegerField()

    @classmethod
    def create(cls, name, opennebula_vm_template_id):
        vm_template = cls(
            name=name, opennebula_vm_template_id=opennebula_vm_template_id)
        return vm_template


class StripePlan(models.Model):
    """
    A model to store Data Center Light's created Stripe plans
    """
    stripe_plan_id = models.CharField(max_length=256, null=True)

    @classmethod
    def create(cls, stripe_plan_id):
        stripe_plan = cls(stripe_plan_id=stripe_plan_id)
        return stripe_plan


class ContactUs(models.Model):
    name = models.CharField(max_length=250)
    email = models.CharField(max_length=250)
    message = models.TextField()
    field = models.DateTimeField(auto_now_add=True)