refactor models into files
This commit is contained in:
parent
c3d20fda7b
commit
f5f4ec8ac9
8 changed files with 716 additions and 687 deletions
316
hub/services/models/pricing.py
Normal file
316
hub/services/models/pricing.py
Normal file
|
|
@ -0,0 +1,316 @@
|
|||
from django.db import models
|
||||
|
||||
from .base import Currency, Term, Unit
|
||||
from .providers import CloudProvider
|
||||
from .services import Service
|
||||
|
||||
|
||||
class ComputePlanPrice(models.Model):
|
||||
compute_plan = models.ForeignKey(
|
||||
"ComputePlan", on_delete=models.CASCADE, related_name="prices"
|
||||
)
|
||||
currency = models.CharField(
|
||||
max_length=3,
|
||||
choices=Currency.choices,
|
||||
)
|
||||
amount = models.DecimalField(
|
||||
max_digits=10,
|
||||
decimal_places=2,
|
||||
help_text="Price in the specified currency, excl. VAT",
|
||||
)
|
||||
|
||||
class Meta:
|
||||
unique_together = ("compute_plan", "currency")
|
||||
ordering = ["currency"]
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.compute_plan.name} - {self.amount} {self.currency}"
|
||||
|
||||
|
||||
class ComputePlan(models.Model):
|
||||
name = models.CharField(max_length=200)
|
||||
vcpus = models.FloatField(help_text="Number of available vCPUs")
|
||||
ram = models.FloatField(help_text="Amount of RAM available")
|
||||
cpu_mem_ratio = models.FloatField(
|
||||
help_text="vCPU to Memory ratio. How much vCPU per GiB RAM is available?"
|
||||
)
|
||||
active = models.BooleanField(default=True, help_text="Is the plan active?")
|
||||
term = models.CharField(
|
||||
max_length=3,
|
||||
default=Term.MTH,
|
||||
choices=Term.choices,
|
||||
)
|
||||
|
||||
cloud_provider = models.ForeignKey(
|
||||
CloudProvider, on_delete=models.CASCADE, related_name="compute_plans"
|
||||
)
|
||||
|
||||
valid_from = models.DateTimeField(blank=True, null=True)
|
||||
valid_to = models.DateTimeField(blank=True, null=True)
|
||||
|
||||
class Meta:
|
||||
ordering = ["name"]
|
||||
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
def get_price(self, currency_code: str):
|
||||
try:
|
||||
return self.prices.get(currency=currency_code).amount
|
||||
except ComputePlanPrice.DoesNotExist:
|
||||
return None
|
||||
|
||||
|
||||
class StoragePlanPrice(models.Model):
|
||||
storage_plan = models.ForeignKey(
|
||||
"StoragePlan", on_delete=models.CASCADE, related_name="prices"
|
||||
)
|
||||
currency = models.CharField(
|
||||
max_length=3,
|
||||
choices=Currency.choices,
|
||||
)
|
||||
amount = models.DecimalField(
|
||||
max_digits=10,
|
||||
decimal_places=2,
|
||||
help_text="Price in the specified currency, excl. VAT",
|
||||
)
|
||||
|
||||
class Meta:
|
||||
unique_together = ("storage_plan", "currency")
|
||||
ordering = ["currency"]
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.storage_plan.name} - {self.amount} {self.currency}"
|
||||
|
||||
|
||||
class StoragePlan(models.Model):
|
||||
name = models.CharField(max_length=200)
|
||||
cloud_provider = models.ForeignKey(
|
||||
CloudProvider, on_delete=models.CASCADE, related_name="storage_plans"
|
||||
)
|
||||
term = models.CharField(
|
||||
max_length=3,
|
||||
default=Term.MTH,
|
||||
choices=Term.choices,
|
||||
)
|
||||
unit = models.CharField(
|
||||
max_length=3,
|
||||
default=Unit.GIB,
|
||||
choices=Unit.choices,
|
||||
)
|
||||
|
||||
valid_from = models.DateTimeField(blank=True, null=True)
|
||||
valid_to = models.DateTimeField(blank=True, null=True)
|
||||
|
||||
class Meta:
|
||||
unique_together = ("cloud_provider", "term", "unit", "valid_from", "valid_to")
|
||||
ordering = ["name"]
|
||||
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
def get_price(self, currency_code: str):
|
||||
try:
|
||||
return self.prices.get(currency=currency_code).amount
|
||||
except StoragePlanPrice.DoesNotExist:
|
||||
return None
|
||||
|
||||
|
||||
class ProgressiveDiscountModel(models.Model):
|
||||
name = models.CharField(max_length=100)
|
||||
description = models.TextField(blank=True)
|
||||
active = models.BooleanField(default=True)
|
||||
|
||||
class Meta:
|
||||
verbose_name = "Discount Model"
|
||||
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
def calculate_discount(self, base_rate, units):
|
||||
"""Calculate price using progressive percentage discounts."""
|
||||
final_price = 0
|
||||
remaining_units = units
|
||||
processed_units = 0
|
||||
|
||||
# Get all tiers sorted by threshold
|
||||
discount_tiers = self.tiers.all().order_by("threshold")
|
||||
|
||||
for i, tier in enumerate(discount_tiers):
|
||||
# Calculate how many units fall into this tier
|
||||
if i < discount_tiers.count() - 1:
|
||||
next_threshold = discount_tiers[i + 1].threshold
|
||||
tier_units = min(remaining_units, next_threshold - processed_units)
|
||||
else:
|
||||
# Last tier handles all remaining units
|
||||
tier_units = remaining_units
|
||||
|
||||
# Calculate discounted rate for this tier
|
||||
discounted_rate = base_rate * (1 - tier.discount_percent / 100)
|
||||
|
||||
# Add the price for units in this tier
|
||||
final_price += discounted_rate * tier_units
|
||||
|
||||
# Update tracking variables
|
||||
remaining_units -= tier_units
|
||||
processed_units += tier_units
|
||||
|
||||
# Exit if all units have been processed
|
||||
if remaining_units <= 0:
|
||||
break
|
||||
|
||||
return final_price
|
||||
|
||||
|
||||
class DiscountTier(models.Model):
|
||||
discount_model = models.ForeignKey(
|
||||
ProgressiveDiscountModel, on_delete=models.CASCADE, related_name="tiers"
|
||||
)
|
||||
threshold = models.PositiveIntegerField(
|
||||
help_text="Starting unit count for this tier"
|
||||
)
|
||||
discount_percent = models.DecimalField(
|
||||
max_digits=5, decimal_places=2, help_text="Percentage discount applied (0-100)"
|
||||
)
|
||||
|
||||
class Meta:
|
||||
ordering = ["threshold"]
|
||||
unique_together = ["discount_model", "threshold"]
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.discount_model.name}: {self.threshold}+ units → {self.discount_percent}% discount"
|
||||
|
||||
|
||||
class VSHNAppCatBaseFee(models.Model):
|
||||
vshn_appcat_price_config = models.ForeignKey(
|
||||
"VSHNAppCatPrice", on_delete=models.CASCADE, related_name="base_fees"
|
||||
)
|
||||
currency = models.CharField(
|
||||
max_length=3,
|
||||
choices=Currency.choices,
|
||||
)
|
||||
amount = models.DecimalField(
|
||||
max_digits=10,
|
||||
decimal_places=2,
|
||||
help_text="Base fee in the specified currency, excl. VAT",
|
||||
)
|
||||
|
||||
class Meta:
|
||||
verbose_name = "Service Base Fee"
|
||||
unique_together = ("vshn_appcat_price_config", "currency")
|
||||
ordering = ["currency"]
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.vshn_appcat_price_config.service.name} Base Fee - {self.amount} {self.currency}"
|
||||
|
||||
|
||||
class VSHNAppCatPrice(models.Model):
|
||||
class VariableUnit(models.TextChoices):
|
||||
RAM = "RAM", "Memory (RAM)"
|
||||
CPU = "CPU", "CPU (vCPU)"
|
||||
USER = "USR", "Users"
|
||||
|
||||
class ServiceLevel(models.TextChoices):
|
||||
BEST_EFFORT = "BE", "Best Effort"
|
||||
GUARANTEED = "GA", "Guaranteed Availability"
|
||||
|
||||
service = models.ForeignKey(
|
||||
Service, on_delete=models.CASCADE, related_name="vshn_appcat_price"
|
||||
)
|
||||
variable_unit = models.CharField(
|
||||
max_length=3,
|
||||
choices=VariableUnit.choices,
|
||||
default=VariableUnit.RAM,
|
||||
)
|
||||
term = models.CharField(
|
||||
max_length=3,
|
||||
default=Term.MTH,
|
||||
choices=Term.choices,
|
||||
)
|
||||
discount_model = models.ForeignKey(
|
||||
ProgressiveDiscountModel,
|
||||
on_delete=models.SET_NULL,
|
||||
null=True,
|
||||
blank=True,
|
||||
related_name="price_configs",
|
||||
)
|
||||
|
||||
ha_replica_min = models.IntegerField(
|
||||
default=1, help_text="Minimum of replicas for HA"
|
||||
)
|
||||
ha_replica_max = models.IntegerField(
|
||||
default=1, help_text="Maximum supported replicas"
|
||||
)
|
||||
|
||||
valid_from = models.DateTimeField(blank=True, null=True)
|
||||
valid_to = models.DateTimeField(blank=True, null=True)
|
||||
|
||||
class Meta:
|
||||
verbose_name = "AppCat Price"
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.service.name} - {self.get_variable_unit_display()} based pricing"
|
||||
|
||||
def get_base_fee(self, currency_code: str):
|
||||
try:
|
||||
return self.base_fees.get(currency=currency_code).amount
|
||||
except VSHNAppCatBaseFee.DoesNotExist:
|
||||
return None
|
||||
|
||||
def get_unit_rate(self, currency_code: str, service_level: str):
|
||||
try:
|
||||
return self.unit_rates.get(
|
||||
currency=currency_code, service_level=service_level
|
||||
).amount
|
||||
except VSHNAppCatUnitRate.DoesNotExist:
|
||||
return None
|
||||
|
||||
def calculate_final_price(
|
||||
self, currency_code: str, service_level: str, number_of_units: int
|
||||
):
|
||||
base_fee = self.get_base_fee(currency_code)
|
||||
unit_rate = self.get_unit_rate(currency_code, service_level)
|
||||
|
||||
if base_fee is None or unit_rate is None:
|
||||
return None
|
||||
|
||||
if number_of_units < 0:
|
||||
raise ValueError("Number of units cannot be negative")
|
||||
|
||||
# Apply discount model if available
|
||||
if self.discount_model and self.discount_model.active:
|
||||
discounted_price = self.discount_model.calculate_discount(
|
||||
unit_rate, number_of_units
|
||||
)
|
||||
total_price = base_fee + discounted_price
|
||||
else:
|
||||
total_price = base_fee + (unit_rate * number_of_units)
|
||||
|
||||
return total_price
|
||||
|
||||
|
||||
class VSHNAppCatUnitRate(models.Model):
|
||||
vshn_appcat_price_config = models.ForeignKey(
|
||||
VSHNAppCatPrice, on_delete=models.CASCADE, related_name="unit_rates"
|
||||
)
|
||||
currency = models.CharField(
|
||||
max_length=3,
|
||||
choices=Currency.choices,
|
||||
)
|
||||
service_level = models.CharField(
|
||||
max_length=2,
|
||||
choices=VSHNAppCatPrice.ServiceLevel.choices,
|
||||
)
|
||||
amount = models.DecimalField(
|
||||
max_digits=10,
|
||||
decimal_places=4,
|
||||
help_text="Price per unit in the specified currency and service level, excl. VAT",
|
||||
)
|
||||
|
||||
class Meta:
|
||||
verbose_name = "Service Unit Rate"
|
||||
unique_together = ("vshn_appcat_price_config", "currency", "service_level")
|
||||
ordering = ["currency", "service_level"]
|
||||
|
||||
def __str__(self):
|
||||
return f"{self.vshn_appcat_price_config.service.name} - {self.get_service_level_display()} Unit Rate - {self.amount} {self.currency}"
|
||||
Loading…
Add table
Add a link
Reference in a new issue