def register_infra_offering(self): try: DatabaseInfraOffering.objects.get(databaseinfra=self.infra) except DatabaseInfraOffering.DoesNotExist: DatabaseInfraOffering( offering=self.cs_offering, databaseinfra=self.infra ).save()
def do(self, workflow_dict): try: if 'environment' not in workflow_dict or 'plan' not in workflow_dict: return False cs_credentials = get_credentials_for( environment=workflow_dict['environment'], credential_type=CredentialType.CLOUDSTACK) vm_credentials = get_credentials_for( environment=workflow_dict['environment'], credential_type=CredentialType.VM) cs_provider = CloudStackProvider(credentials=cs_credentials) cs_plan_attrs = PlanAttr.objects.get(plan=workflow_dict['plan']) workflow_dict['hosts'] = [] workflow_dict['instances'] = [] workflow_dict['databaseinfraattr'] = [] workflow_dict['vms_id'] = [] bundles = list(cs_plan_attrs.bundle.all()) for index, vm_name in enumerate(workflow_dict['names']['vms']): offering = cs_plan_attrs.get_stronger_offering() if len(bundles) == 1: bundle = bundles[0] else: if index == 0: bundle = LastUsedBundle.get_next_infra_bundle( plan=workflow_dict['plan'], bundles=bundles) else: bundle = LastUsedBundle.get_next_bundle( bundle=bundle, bundles=bundles) try: DatabaseInfraOffering.objects.get( databaseinfra=workflow_dict['databaseinfra']) except ObjectDoesNotExist: LOG.info("Creating databaseInfra Offering...") dbinfra_offering = DatabaseInfraOffering() dbinfra_offering.offering = offering dbinfra_offering.databaseinfra = workflow_dict[ 'databaseinfra'] dbinfra_offering.save() LOG.debug( "Deploying new vm on cs with bundle %s and offering %s" % (bundle, offering)) vm = cs_provider.deploy_virtual_machine( offering=offering.serviceofferingid, bundle=bundle, project_id=cs_credentials.project, vmname=vm_name, affinity_group_id=cs_credentials.get_parameter_by_name( 'affinity_group_id'), ) if not vm: raise Exception( "CloudStack could not create the virtualmachine") LOG.debug("New virtualmachine: %s" % vm) workflow_dict['vms_id'].append(vm['virtualmachine'][0]['id']) host = Host() host.address = vm['virtualmachine'][0]['nic'][0]['ipaddress'] host.hostname = host.address host.cloud_portal_host = True host.save() LOG.info("Host created!") workflow_dict['hosts'].append(host) host_attr = HostAttr() host_attr.vm_id = vm['virtualmachine'][0]['id'] host_attr.vm_user = vm_credentials.user host_attr.vm_password = vm_credentials.password host_attr.host = host host_attr.save() LOG.info("Host attrs custom attributes created!") instance = Instance() instance.address = host.address instance.port = 3306 instance.hostname = host instance.databaseinfra = workflow_dict['databaseinfra'] instance.instance_type = Instance.MYSQL instance.save() LOG.info("Instance created!") workflow_dict['instances'].append(instance) if workflow_dict['qt'] == 1: LOG.info("Updating databaseinfra endpoint...") databaseinfra = workflow_dict['databaseinfra'] databaseinfra.endpoint = instance.address + \ ":%i" % (instance.port) databaseinfra.save() workflow_dict['databaseinfra'] = databaseinfra return True return True except Exception: traceback = full_stack() workflow_dict['exceptions']['error_codes'].append(DBAAS_0011) workflow_dict['exceptions']['traceback'].append(traceback) return False
def do(self, workflow_dict): try: if not 'environment' in workflow_dict and not 'plan' in workflow_dict: return False cs_credentials = get_credentials_for( environment=workflow_dict['environment'], credential_type=CredentialType.CLOUDSTACK) vm_credentials = get_credentials_for( environment=workflow_dict['environment'], credential_type=CredentialType.VM) cs_provider = CloudStackProvider(credentials=cs_credentials) cs_plan_attrs = PlanAttr.objects.get(plan=workflow_dict['plan']) workflow_dict['hosts'] = [] workflow_dict['instances'] = [] workflow_dict['databaseinfraattr'] = [] workflow_dict['vms_id'] = [] bundles = list(cs_plan_attrs.bundle.all()) for index, vm_name in enumerate(workflow_dict['names']['vms']): if bundles.__len__() == 1: bundle = bundles[0] else: bundle = LastUsedBundle.get_next_bundle( plan=workflow_dict['plan'], bundle=bundles) if workflow_dict['enginecod'] == workflow_dict[ 'MONGODB'] and index == 2: offering = cs_plan_attrs.get_weaker_offering() else: offering = cs_plan_attrs.get_stronger_offering() try: DatabaseInfraOffering.objects.get( databaseinfra=workflow_dict['databaseinfra']) except ObjectDoesNotExist, e: LOG.info("Creating databaseInfra Offering...") dbinfra_offering = DatabaseInfraOffering() dbinfra_offering.offering = offering dbinfra_offering.databaseinfra = workflow_dict[ 'databaseinfra'] dbinfra_offering.save() LOG.debug( "Deploying new vm on cs with bundle %s and offering %s" % (bundle, offering)) vm = cs_provider.deploy_virtual_machine( offering=offering.serviceofferingid, bundle=bundle, project_id=cs_credentials.project, vmname=vm_name, ) if not vm: raise Exception( "CloudStack could not create the virtualmachine") LOG.debug("New virtualmachine: %s" % vm) workflow_dict['vms_id'].append(vm['virtualmachine'][0]['id']) host = Host() host.address = vm['virtualmachine'][0]['nic'][0]['ipaddress'] host.hostname = host.address host.cloud_portal_host = True host.save() LOG.info("Host created!") workflow_dict['hosts'].append(host) host_attr = HostAttr() host_attr.vm_id = vm['virtualmachine'][0]['id'] host_attr.vm_user = vm_credentials.user host_attr.vm_password = vm_credentials.password host_attr.host = host host_attr.save() LOG.info("Host attrs custom attributes created!") instance = Instance() instance.address = host.address if workflow_dict['enginecod'] == workflow_dict['MYSQL']: instance.port = 3306 elif workflow_dict['enginecod'] == workflow_dict['MONGODB']: instance.port = 27017 elif workflow_dict['enginecod'] == workflow_dict['REDIS']: instance.port = 6379 instance.is_active = True if workflow_dict['enginecod'] == workflow_dict[ 'MONGODB'] and index == 2: instance.is_arbiter = True else: instance.is_arbiter = False instance.hostname = host instance.databaseinfra = workflow_dict['databaseinfra'] instance.save() LOG.info("Instance created!") workflow_dict['instances'].append(instance) if workflow_dict['qt'] == 1: LOG.info("Updating databaseinfra endpoint...") databaseinfra = workflow_dict['databaseinfra'] databaseinfra.endpoint = instance.address + ":%i" % ( instance.port) databaseinfra.save() workflow_dict['databaseinfra'] = databaseinfra return True return True