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