37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
from django.core.management.base import BaseCommand
|
|
|
|
from datacenterlight.models import VMPricing
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = '''Creates default VMPricing object'''
|
|
DEFAULT_VMPRICING_NAME = 'default'
|
|
|
|
def handle(self, *args, **options):
|
|
self.create_default_vm_pricing()
|
|
|
|
def create_default_vm_pricing(self):
|
|
obj, created = VMPricing.objects.get_or_create(
|
|
name=self.DEFAULT_VMPRICING_NAME,
|
|
defaults={
|
|
"vat_inclusive": True,
|
|
"cores_unit_price": 5,
|
|
"ram_unit_price": 2,
|
|
"ssd_unit_price": 0.6,
|
|
"hdd_unit_price": 0.01
|
|
}
|
|
)
|
|
|
|
if created:
|
|
print(
|
|
'Successfully created {} VMPricing object'.format(
|
|
self.DEFAULT_VMPRICING_NAME
|
|
)
|
|
)
|
|
else:
|
|
print(
|
|
'{} VMPricing exists already.'.format(
|
|
self.DEFAULT_VMPRICING_NAME
|
|
)
|
|
)
|