19 lines
632 B
Python
19 lines
632 B
Python
|
from core.exception import PriceNotFound
|
||
|
|
from core.models import VolumePrice, InvoiceVolume, PriceMixin
|
||
|
|
from core.component.base.invoice_handler import InvoiceHandler
|
||
|
|
|
||
|
|
|
||
|
|
class VolumeInvoiceHandler(InvoiceHandler):
|
||
|
|
INVOICE_CLASS = InvoiceVolume
|
||
|
|
KEY_FIELD = "volume_id"
|
||
|
|
PRICE_DEPENDENCY_FIELDS = ['volume_type_id', 'space_allocation_gb']
|
||
|
|
INFORMATIVE_FIELDS = ['volume_name']
|
||
|
|
|
||
|
|
def get_price(self, payload) -> PriceMixin:
|
||
|
|
price = VolumePrice.objects.filter(volume_type_id=payload['volume_type_id']).first()
|
||
|
|
|
||
|
|
if price is None:
|
||
|
|
raise PriceNotFound(identifier='volume')
|
||
|
|
|
||
|
|
return price
|