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