122 lines
3.4 KiB
Python
122 lines
3.4 KiB
Python
import json
|
|
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.core import serializers
|
|
from membership.models import StripeCustomer
|
|
from utils.models import BillingAddress
|
|
|
|
|
|
class RailsBetaUser(models.Model):
|
|
email = models.EmailField(unique=True)
|
|
received_date = models.DateTimeField('date received')
|
|
|
|
def __str__(self):
|
|
return "%s - %s" % (self.email, self.received_date)
|
|
|
|
|
|
class VirtualMachineType(models.Model):
|
|
|
|
HETZNER_NUG = 'hetzner_nug'
|
|
HETZNER = 'hetzner'
|
|
HETZNER_R6 = 'hetzner_raid6'
|
|
HETZNER_G = 'hetzner_glusterfs'
|
|
BERN = 'bern'
|
|
|
|
HOSTING_TYPES = (
|
|
(HETZNER_NUG, 'Hetzner No Uptime Guarantee'),
|
|
(HETZNER, 'Hetzner'),
|
|
(HETZNER_R6, 'Hetzner Raid6'),
|
|
(HETZNER_G, 'Hetzner Glusterfs'),
|
|
(BERN, 'Bern'),
|
|
)
|
|
|
|
description = models.TextField()
|
|
base_price = models.FloatField()
|
|
memory_price = models.FloatField()
|
|
core_price = models.FloatField()
|
|
disk_size_price = models.FloatField()
|
|
hosting_company = models.CharField(max_length=30, choices=HOSTING_TYPES)
|
|
|
|
def __str__(self):
|
|
return "%s" % (self.get_hosting_company_display())
|
|
|
|
@classmethod
|
|
def get_serialized_vm_types(cls):
|
|
return [vm.get_serialized_data()
|
|
for vm in cls.objects.all()]
|
|
|
|
def calculate_price(self, specifications):
|
|
price = float(specifications['cores']) * self.core_price
|
|
price += float(specifications['memory']) * self.memory_price
|
|
price += float(specifications['disk_size']) * self.disk_size_price
|
|
price += self.base_price
|
|
return price
|
|
|
|
def defeault_price(self):
|
|
price = self.base_price
|
|
price += self.core_price
|
|
price += self.memory_price
|
|
price += self.disk_size_price * 10
|
|
return price
|
|
|
|
def get_serialized_data(self):
|
|
return {
|
|
'description': self.description,
|
|
'base_price': self.base_price,
|
|
'core_price': self.core_price,
|
|
'disk_size_price': self.disk_size_price,
|
|
'memory_price': self.memory_price,
|
|
'hosting_company_name': self.get_hosting_company_display(),
|
|
'hosting_company': self.hosting_company,
|
|
'default_price': self.defeault_price(),
|
|
'id': self.id,
|
|
}
|
|
|
|
|
|
class VirtualMachinePlan(models.Model):
|
|
cores = models.IntegerField()
|
|
memory = models.IntegerField()
|
|
disk_size = models.IntegerField()
|
|
vm_type = models.ForeignKey(VirtualMachineType)
|
|
price = models.FloatField()
|
|
|
|
@classmethod
|
|
def create(cls, data, user):
|
|
instance = cls.objects.create(**data)
|
|
return instance
|
|
|
|
|
|
class HostingOrder(models.Model):
|
|
VMPlan = models.OneToOneField(VirtualMachinePlan)
|
|
customer = models.ForeignKey(StripeCustomer)
|
|
billing_address = models.ForeignKey(BillingAddress)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
approved = models.BooleanField(default=False)
|
|
stripe_charge_id = models.CharField(max_length=100, null=True)
|
|
|
|
@classmethod
|
|
def create(cls, VMPlan=None, customer=None, billing_address=None):
|
|
instance = cls.objects.create(VMPlan=VMPlan, customer=customer,
|
|
billing_address=billing_address)
|
|
return instance
|
|
|
|
def set_approved(self):
|
|
self.approved = True
|
|
self.save()
|
|
|
|
def set_stripe_charge(self, stripe_charge):
|
|
self.stripe_charge_id = stripe_charge.id
|
|
self.save()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|