Example #1
0
def setup_template_task(template_id, name, user, ssh_key, cores, memory):
    with app.app_context():
        job = get_current_job()
        proxmox = connect_proxmox()
        starrs = connect_starrs()
        db = connect_db()
        print("[{}] Retrieving template info for template {}.".format(
            name, template_id))
        template = get_template(db, template_id)
        print("[{}] Cloning template {}.".format(name, template_id))
        job.meta['status'] = 'cloning template'
        job.save_meta()
        vmid, mac = clone_vm(proxmox, template_id, name, user)
        print("[{}] Waiting until Proxmox is done provisioning.".format(name))
        job.meta['status'] = 'waiting for Proxmox'
        job.save_meta()
        timeout = 20
        retry = 0
        while retry < timeout:
            if not VM(vmid).is_provisioned():
                retry += 1
                time.sleep(3)
                continue
            break
        if retry == timeout:
            print("[{}] Failed to provision, deleting.".format(name))
            job.meta['status'] = 'failed to provision'
            job.save_meta()
            delete_vm_task(vmid)
            return
        print("[{}] Registering in STARRS.".format(name))
        job.meta['status'] = 'registering in STARRS'
        job.save_meta()
        ip = get_next_ip(starrs, app.config['STARRS_IP_RANGE'])
        register_starrs(starrs, name, app.config['STARRS_USER'], mac, ip)
        get_vm_expire(db, vmid, app.config['VM_EXPIRE_MONTHS'])
        print("[{}] Setting CPU and memory.".format(name))
        job.meta['status'] = 'setting CPU and memory'
        job.save_meta()
        vm = VM(vmid)
        vm.set_cpu(cores)
        vm.set_mem(memory)
        print("[{}] Applying cloud-init config.".format(name))
        job.meta['status'] = 'applying cloud-init'
        job.save_meta()
        vm.set_ci_user(user)
        vm.set_ci_ssh_key(ssh_key)
        vm.set_ci_network()
        print("[{}] Waiting for STARRS to propogate before starting VM.".
              format(name))
        job.meta['status'] = 'waiting for STARRS'
        job.save_meta()
        time.sleep(90)
        print("[{}] Starting VM.".format(name))
        job.meta['status'] = 'starting VM'
        job.save_meta()
        vm.start()
        print("[{}] Template successfully provisioned.".format(name))
        job.meta['status'] = 'completed'
        job.save_meta()
Example #2
0
def setup_template_task(template_id, name, user, ssh_key, cores, memory):
    with app.app_context():
        job = get_current_job()
        proxmox = connect_proxmox()
        if app.config['USE_STARRS']:
            starrs = connect_starrs()
        db = connect_db()
        logging.info('[{}] Retrieving template info for template {}.'.format(
            name, template_id))
        get_template(db, template_id)
        logging.info('[{}] Cloning template {}.'.format(name, template_id))
        set_job_status(job, 'cloning template')
        vmid = clone_vm(proxmox, template_id, name, user)
        logging.info(
            '[{}] Waiting until Proxmox is done provisioning.'.format(name))
        set_job_status(job, 'waiting for Proxmox')
        timeout = 25
        retry = 0
        while retry < timeout:
            if not VM(vmid).is_provisioned():
                retry += 1
                time.sleep(12)
                continue
            break
        if retry == timeout:
            logging.info('[{}] Failed to provision, deleting.'.format(name))
            set_job_status(job, 'failed to provision')
            delete_vm_task(vmid)
            return

        vm = VM(vmid)
        if app.config['USE_STARRS']:
            logging.info('[{}] Registering in STARRS.'.format(name))
            set_job_status(job, 'registering in STARRS')
            ip = get_next_ip(starrs, app.config['STARRS_IP_RANGE'])
            register_starrs(starrs, name, app.config['STARRS_USER'],
                            vm.get_mac(), ip)
        get_vm_expire(db, vmid, app.config['VM_EXPIRE_MONTHS'])
        logging.info('[{}] Setting CPU and memory.'.format(name))
        set_job_status(job, 'setting CPU and memory')
        vm.set_cpu(cores)
        vm.set_mem(memory)
        logging.info('[{}] Applying cloud-init config.'.format(name))
        set_job_status(job, 'applying cloud-init')
        vm.set_ci_user(user)
        vm.set_ci_ssh_key(ssh_key)
        vm.set_ci_network()

        if app.config['USE_STARRS']:
            logging.info(
                '[{}] Waiting for STARRS to propogate before starting VM.'.
                format(name))
            set_job_status(job, 'waiting for STARRS')
        job.save_meta()
        time.sleep(90)
        logging.info('[{}] Starting VM.'.format(name))
        set_job_status(job, 'starting VM')
        job.save_meta()
        vm.start()
        logging.info('[{}] Template successfully provisioned.'.format(name))
        set_job_status(job, 'completed')
        job.save_meta()