Add create_vm_pricing management command
This commit is contained in:
		
					parent
					
						
							
								588f513f2a
							
						
					
				
			
			
				commit
				
					
						d50f282057
					
				
			
		
					 1 changed files with 36 additions and 0 deletions
				
			
		|  | @ -0,0 +1,36 @@ | ||||||
|  | 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 | ||||||
|  |                 ) | ||||||
|  |             ) | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue