def delete(zone_number):
zone = Zone.query.get(zone_number)
clusters = Cluster.query.filter_by(zone=zone).all()
form = ActionForm(request.form, zone=zone)
if request.method == 'POST' and form.validate():
try:
if request.form['action'] == 'Cancel':
flash('Delete {} action cancelled'.format(zone.name),
category='info')
return redirect(
url_for('zone_bp.view', zone_number=zone.number))
elif request.form['action'] == 'Confirm':
Session.delete(zone)
Session.commit()
flash('{} has been deleted'.format(zone.name),
category='success')
return redirect(url_for('zone_bp.list'))
except Exception as e:
flash('There was an error deleting zone {}'.format(zone.name),
category='success')
return redirect(url_for('zone_bp.list'))
return render_template('zone/delete.html',
form=form,
zone=zone,
clusters=clusters)
def view(pool_id):
form = ActionForm()
pool = VirtualMachinePool.query.get(pool_id)
members = pool.get_memberships()
return render_template('vpool/view.html',
form=form,
pool=pool,
members=members)
def audit_dns(pool_id):
form = ActionForm()
pool = VirtualMachinePool.query.get(pool_id)
members = pool.get_memberships()
ddns_audit_log = pool.cluster.zone.get_ddns_api().audit_pool_dns(
pool=pool, members=members, resolve_errors=True)
return render_template('vpool/audit_dns.html',
form=form,
pool=pool,
members=members,
ddns_audit_log=ddns_audit_log)
def shrink(pool_id):
pool = form_shrink_vm_ids = None
form = ActionForm()
try:
pool = VirtualMachinePool.query.get(pool_id)
if request.method == 'POST' and request.form['action'] == 'cancel':
flash('Shrink {} cancelled'.format(pool.name), category='info')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
elif request.method == 'POST' and request.form['action'] == 'shrink':
form_shrink_vm_ids = [
int(id) for id in request.form.getlist('shrink_vm_ids')
]
members = pool.get_memberships()
shrink_members = pool.get_members_to_shrink(members,
form_shrink_vm_ids)
if shrink_members is None or len(shrink_members) == 0:
raise Exception(
"Cannot determine any members to shutdown for shrinking")
except Exception as e:
flash("There was an error determining memberships for shrinking: {}".
format(e),
category='danger')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
if request.method == 'POST' and form.validate(
) and request.form['action'] == 'shrink':
title = 'Plan Shrink => Pool {} ({} members to {})'.format(
pool.name, len(members), pool.cardinality)
description = "Pool shrink triggered that will shutdown {} VM(s): \n\n*{}".format(
len(shrink_members),
"\n*".join([m.vm.name for m in shrink_members]))
task = Task(name=title,
description="{}\n{}".format(title, description),
username=current_user.username)
Session.add(task)
Session.commit()
task_thread = TaskThread(task_id=task.id,
run_function=plan_shrink,
pool=pool,
shrink_members=shrink_members)
task_thread.start()
flash(
Markup("Started background task {}: {}".format(
task.name, task.link())))
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
return render_template('vpool/shrink.html',
form=form,
pool=pool,
shrink_members=shrink_members)
def expand(pool_id):
pool = members = expansion_names = form_expansion_names = None
form = ActionForm()
try:
pool = VirtualMachinePool.query.get(pool_id)
if request.method == 'POST' and request.form['action'] == 'cancel':
flash('Expansion of {} cancelled'.format(pool.name),
category='info')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
elif request.method == 'POST' and request.form['action'] == 'expand':
form_expansion_names = request.form.getlist('expansion_names')
members = pool.get_memberships()
expansion_names = pool.get_expansion_names(members,
form_expansion_names)
except Exception as e:
flash(
"There was an error determining new names required for expansion: {}"
.format(e),
category='danger')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
if request.method == 'POST' and form.validate(
) and request.form['action'] == 'expand':
title = 'Plan Change => Pool Expansion: {} ({} members to {})'.format(
pool.name, len(members), pool.cardinality)
description = "Pool expansion triggered that will instantiate {} new VM(s): \n\n*{}".format(
len(expansion_names), "\n*".join(expansion_names))
task = Task(name=title,
description="{}\n{}".format(title, description),
username=current_user.username)
Session.add(task)
Session.commit()
task_thread = TaskThread(task_id=task.id,
run_function=plan_expansion,
pool=pool,
expansion_names=expansion_names)
task_thread.start()
flash(
Markup("Started background task {}: {}".format(
task.name, task.link())))
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
return render_template('vpool/expand.html',
form=form,
members=members,
pool=pool,
expansion_names=expansion_names)
def health(pool_id):
form = ActionForm()
pool = VirtualMachinePool.query.get(pool_id)
members = pool.get_memberships()
diagnostics = []
for m in members:
diag = Diagnostic(user=app.config['SSH_HEALTH_CHECK_USER'],
host=m.vm.name,
ssh_identity_file=app.config['SSH_IDENTITY_FILE'],
cmd=app.config['SSH_HEALTH_CHECK_CMD'],
timeout=app.config['SSH_HEALTH_CHECK_TIMEOUT'])
diag.run()
diagnostics.append(diag)
return render_template('vpool/health.html',
form=form,
pool=pool,
members=members,
diagnostics=diagnostics)
def delete(pool_id):
pool = None
vms_by_id = {}
form = ActionForm()
try:
pool = VirtualMachinePool.query.get(pool_id)
except Exception as e:
flash("There was an error fetching pool_id={}: {}".format(pool_id, e),
category='danger')
if request.method == 'POST' and form.validate():
try:
if request.form['action'] == 'cancel':
flash('Delete {} action cancelled'.format(pool.name),
category='info')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
elif request.form['action'] == 'confirm':
redirect_url = url_for('cluster_bp.view',
zone_number=pool.cluster.zone.number,
cluster_id=pool.cluster.id)
members = pool.get_memberships()
for member in members:
Session.delete(member)
for ticket in pool.get_tickets():
logging.info("deleted ticket {}".format(ticket.ticket_key))
Session.delete(ticket)
Session.delete(pool)
Session.commit()
flash('Deleted pool {} with {} members'.format(
pool.name, len(members)),
category='success')
return redirect(
url_for('cluster_bp.view',
zone_number=pool.cluster.zone.number,
cluster_id=pool.cluster.id))
except Exception as e:
# raise e
flash('There was an error deleting pool {}: {}'.format(
pool.name, e),
category='danger')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
return render_template('vpool/delete.html',
form=form,
pool=pool,
vms_by_id=vms_by_id)
def update(pool_id):
form = ActionForm()
pool = VirtualMachinePool.query.get(pool_id)
form_update_ids = None
if request.method == 'POST' and request.form['action'] == 'cancel':
flash('Update of {} cancelled'.format(pool.name), category='info')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
elif request.method == 'POST' and request.form['action'] == 'update':
form_update_ids = request.form.getlist('update_ids')
members = pool.get_memberships()
update_ids = pool.get_update_ids(members, form_update_ids)
if request.method == 'POST' and form.validate(
) and request.form['action'] == 'update':
id_to_template = {}
for m in members:
if m.vm.id in update_ids:
id_to_template[m.vm.id] = m.current_template()
title = "Plan Change => Update {} member(s) in pool {}".format(
len(update_ids), pool.name)
description = "The attached templates will replace the VMs identified in their filename (based on VM ID)"
task = Task(name=title,
description="{}\n{}".format(title, description),
username=current_user.username)
Session.add(task)
Session.commit()
task_thread = TaskThread(task_id=task.id,
run_function=plan_update,
pool=pool,
id_to_template=id_to_template)
task_thread.start()
flash(
Markup("Started background task {}: {}".format(title,
task.link())))
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
return render_template('vpool/update.html',
form=form,
pool=pool,
members=members,
update_ids=update_ids)
def assign_to_pool(zone_number, cluster_id):
vms = []
id_to_vm = {}
selected_vm_ids = {}
pools = None
zone = None
cluster = None
memberships = {}
try:
Session()
zone = Zone.query.get(zone_number)
cluster = Cluster.query.filter_by(zone=zone, id=cluster_id).first()
one_proxy = OneProxy(zone.xmlrpc_uri,
zone.session_string,
verify_certs=False)
for membership in PoolMembership.query.join(
VirtualMachinePool).filter_by(cluster=cluster).all():
memberships[membership.vm_id] = membership
for vm in one_proxy.get_vms():
if vm.disk_cluster.id == cluster.id:
vms.append(vm)
id_to_vm[vm.id] = vm
pools = VirtualMachinePool.get_all(cluster)
except Exception as e:
# raise e
flash("Error fetching VMs in zone number {}: {}".format(
zone.number, e),
category='danger')
form = ActionForm()
active_tab = 'create_new_pool'
# Form submission handling begins
if form.validate_on_submit():
# Determine which tab needs to be active based on the action
if request.form['action'] is not None:
print('something')
active_tab = {
'create new pool': 'create_new_pool',
'add to pool': 'add_to_existing_pool'
}[request.form['action']]
# Get a list of clusters of all selected VMs--pools cannot span clusters
selected_clusters = {}
for id in request.form.getlist('chk_vm_id'):
selected_vm_ids[int(id)] = id
selected_clusters[id_to_vm[int(id)].disk_cluster.id] = True
# Error checking begins
proceed = True
if len(selected_vm_ids) == 0:
flash("No virtual machines were selected!", category='danger')
proceed = False
elif len(selected_clusters) != 1:
flash("Selected VMs must all be in the same cluster",
category='danger')
proceed = False
# Handle the appropriate action if able to proceed
if proceed and request.form['action'] == 'add to pool':
if (request.form['pool_id']
) is None or request.form['pool_id'] == '':
flash('No pool selected', category='danger')
else:
pool = VirtualMachinePool.query.get(request.form['pool_id'])
for vm_id in selected_vm_ids.keys():
Session.add(
PoolMembership(pool=pool,
vm_name=id_to_vm[vm_id].name,
vm_id=vm_id,
date_added=datetime.utcnow()))
Session.commit()
flash(Markup(
'Successfully added {} members to pool <a href="{}">{}</a>'
.format(
len(selected_vm_ids),
url_for('vpool_bp.view', pool_id=pool.id),
pool.name,
)),
category='success')
return redirect(
url_for('vpool_bp.assign_to_pool',
zone_number=zone.number,
cluster_id=cluster.id))
if proceed and request.form['action'] == 'create new pool':
try:
if request.form['new_pool_name'] is None or request.form[
'new_pool_name'] == '':
raise Exception('Pool name cannot be blank')
pool = VirtualMachinePool(name=request.form['new_pool_name'],
cluster=cluster,
zone_number=zone.number,
cardinality=len(selected_vm_ids))
Session.add(pool)
Session.flush()
for vm_id in selected_vm_ids.keys():
membership = PoolMembership(pool=pool,
vm_name=id_to_vm[vm_id].name,
vm_id=vm_id,
date_added=datetime.utcnow())
memberships[vm_id] = membership
Session.add(membership)
Session.flush()
Session.commit()
flash(Markup(
'Successfully created <a href="{}">{}</a> with {} pool members'
.format(url_for('vpool_bp.view', pool_id=pool.id),
pool.name, len(selected_vm_ids))),
category='success')
except Exception as e:
Session.rollback()
flash('Error creating your new pool: {}'.format(e),
category='danger')
return render_template('vpool/assign_to_pool.html',
form=form,
zone=zone,
cluster=cluster,
vms=vms,
memberships=memberships,
selected_vm_ids=selected_vm_ids,
pools=pools,
active_tab_name=active_tab)
def remove_done(pool_id):
form = ActionForm()
pool = one_proxy = members = None
try:
pool = VirtualMachinePool.query.get(pool_id)
members = pool.get_memberships()
except Exception as e:
flash("There was an error finshed VMs: {}".format(e),
category='danger')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
if request.method == 'POST' and form.validate():
try:
if request.form['action'] == 'cancel':
flash('Cleanup of {} cancelled'.format(pool.name),
category='info')
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
elif request.form['action'] == 'confirm':
vm_ids_to_delete = [
int(id) for id in request.form.getlist('done_vm_ids')
]
delete_members = []
Session()
for m in members:
if m.vm.id in vm_ids_to_delete:
delete_members.append(m)
delete_ticket = jira.instance.create_issue(
project=app.config['JIRA_PROJECT'],
summary='[auto-{}] Pool Cleanup: {} (deleting {} done VMs)'
.format(current_user.username, pool.name,
len(vm_ids_to_delete)),
description=
"Pool cleanup triggered that will delete {} VM(s): \n\n*{}"
.format(
len(vm_ids_to_delete), "\n*".join([
'ID {}: {} ({})'.format(m.vm.id, m.vm.name,
m.vm.ip_address)
for m in delete_members
])),
customfield_13842=jira.get_datetime_now(),
issuetype={'name': 'Task'})
one_proxy = OneProxy(pool.cluster.zone.xmlrpc_uri,
pool.cluster.zone.session_string,
verify_certs=False)
for m in delete_members:
one_proxy.action_vm(m.remove_cmd(), m.vm.id)
Session.delete(m)
Session.commit()
flash('Deleted {} done VMs to cleanup pool {}'.format(
len(delete_members), pool.name))
jira.resolve(delete_ticket)
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
except Exception as e:
flash("Error performing cleanup of pool {}: {}".format(
pool.name, e),
category='danger')
jira.defect_for_exception(
"Error during cleanup of pool {}".format(pool.name), e)
return redirect(url_for('vpool_bp.view', pool_id=pool.id))
return render_template('vpool/remove_done.html',
form=form,
pool=pool,
members=members)