appcat price model

hub/services/models.py

@@ -356,6 +356,34 @@ class WebsiteFaq(models.Model):
         return self.question
 
 
+class Currency(models.TextChoices):
+    CHF = "CHF", "Swiss Franc"
+    EUR = "EUR", "Euro"
+    USD = "USD", "US Dollar"
+
+
+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")
@@ -363,13 +391,6 @@ class ComputePlan(models.Model):
     cpu_mem_ratio = models.FloatField(
         help_text="vCPU to Memory ratio. How much vCPU per GiB RAM is available?"
     )
-    price_chf = models.FloatField(help_text="Plan price in CHF excl. VAT")
-    price_eur = models.FloatField(
-        help_text="Plan price in EUR excl. VAT", blank=True, null=True
-    )
-    price_usd = models.FloatField(
-        help_text="Plan price in USD excl. VAT", blank=True, null=True
-    )
     active = models.BooleanField(default=True, help_text="Is the plan active?")
 
     cloud_provider = models.ForeignKey(
@@ -380,7 +401,119 @@ class ComputePlan(models.Model):
     valid_to = models.DateTimeField(blank=True, null=True)
 
     class Meta:
-        ordering = ["price_chf"]
+        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 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:
+        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,
+    )
+    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"
+    )
+
+    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")
+
+        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:
+        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}"