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)