2018-09-20 03:25:58 -04:00
#!/usr/bin/env python3
# vm.py - PVC client function library, VM fuctions
# Part of the Parallel Virtual Cluster (PVC) system
#
# Copyright (C) 2018 Joshua M. Boniface <joshua@boniface.me>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
#
###############################################################################
import os
import socket
import time
import uuid
import re
import subprocess
import difflib
import colorama
import click
import lxml . objectify
import configparser
import kazoo . client
2018-10-20 15:28:25 -04:00
import client_lib . ansiprint as ansiprint
2018-09-25 02:20:32 -04:00
import client_lib . zkhandler as zkhandler
2018-09-20 03:42:40 -04:00
import client_lib . common as common
2018-09-20 03:25:58 -04:00
#
# XML information parsing functions
#
def getInformationFromXML ( zk_conn , uuid , long_output ) :
# Obtain the contents of the XML from Zookeeper
try :
2018-10-27 15:27:08 -04:00
dstate = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( uuid ) )
dnode = zkhandler . readdata ( zk_conn , ' /domains/ {} /node ' . format ( uuid ) )
dlastnode = zkhandler . readdata ( zk_conn , ' /domains/ {} /lastnode ' . format ( uuid ) )
2018-09-20 03:25:58 -04:00
except :
return None
2018-10-14 02:01:35 -04:00
if dlastnode == ' ' :
dlastnode = ' N/A '
2018-09-20 03:25:58 -04:00
2018-10-14 02:01:35 -04:00
parsed_xml = common . getDomainXML ( zk_conn , uuid )
duuid , dname , ddescription , dmemory , dvcpu , dvcputopo = common . getDomainMainDetails ( parsed_xml )
dnets = common . getDomainNetworks ( parsed_xml )
2018-09-20 03:25:58 -04:00
if long_output == True :
dtype , darch , dmachine , dconsole , demulator = common . getDomainExtraDetails ( parsed_xml )
dfeatures = common . getDomainCPUFeatures ( parsed_xml )
ddisks = common . getDomainDisks ( parsed_xml )
dcontrollers = common . getDomainControllers ( parsed_xml )
# Format a nice output; do this line-by-line then concat the elements at the end
ainformation = [ ]
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {} Virtual machine information: {} ' . format ( ansiprint . bold ( ) , ansiprint . end ( ) ) )
2018-09-20 03:25:58 -04:00
ainformation . append ( ' ' )
# Basic information
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {} UUID: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , duuid ) )
ainformation . append ( ' {} Name: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dname ) )
ainformation . append ( ' {} Description: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , ddescription ) )
2018-11-01 23:24:38 -04:00
ainformation . append ( ' {} Memory (M): {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dmemory ) )
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {} vCPUs: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dvcpu ) )
ainformation . append ( ' {} Topology (S/C/T): {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dvcputopo ) )
2018-09-20 03:25:58 -04:00
if long_output == True :
# Virtualization information
ainformation . append ( ' ' )
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {} Emulator: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , demulator ) )
ainformation . append ( ' {} Type: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dtype ) )
ainformation . append ( ' {} Arch: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , darch ) )
ainformation . append ( ' {} Machine: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dmachine ) )
ainformation . append ( ' {} Features: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , ' ' . join ( dfeatures ) ) )
2018-09-20 03:25:58 -04:00
# PVC cluster information
ainformation . append ( ' ' )
dstate_colour = {
2018-10-20 15:28:25 -04:00
' start ' : ansiprint . green ( ) ,
' restart ' : ansiprint . yellow ( ) ,
' shutdown ' : ansiprint . yellow ( ) ,
' stop ' : ansiprint . red ( ) ,
' failed ' : ansiprint . red ( ) ,
' migrate ' : ansiprint . blue ( ) ,
' unmigrate ' : ansiprint . blue ( )
2018-09-20 03:25:58 -04:00
}
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {} State: {} {} {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dstate_colour [ dstate ] , dstate , ansiprint . end ( ) ) )
ainformation . append ( ' {} Current Node: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dnode ) )
ainformation . append ( ' {} Previous Node: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , dlastnode ) )
2018-10-14 02:01:35 -04:00
# Network list
net_list = [ ]
for net in dnets :
# Split out just the numerical (VNI) part of the brXXXX name
2018-11-18 17:48:14 -05:00
net_vnis = re . findall ( r ' \ d+ ' , net [ ' source ' ] )
if net_vnis :
net_vni = net_vnis [ 0 ]
else :
net_vni = re . sub ( ' br ' , ' ' , net [ ' source ' ] )
2018-10-20 15:27:07 -04:00
net_exists = zkhandler . exists ( zk_conn , ' /networks/ {} ' . format ( net_vni ) )
2018-11-18 17:48:14 -05:00
if not net_exists and net_vni != ' cluster ' :
2018-10-20 15:28:25 -04:00
net_list . append ( ansiprint . red ( ) + net_vni + ansiprint . end ( ) + ' [invalid] ' )
2018-10-20 15:27:07 -04:00
else :
net_list . append ( net_vni )
2018-10-14 02:01:35 -04:00
ainformation . append ( ' ' )
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {} Networks: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , ' , ' . join ( net_list ) ) )
2018-09-20 03:25:58 -04:00
if long_output == True :
# Disk list
ainformation . append ( ' ' )
name_length = 0
for disk in ddisks :
_name_length = len ( disk [ ' name ' ] ) + 1
if _name_length > name_length :
name_length = _name_length
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {0} Disks: {1} {2} ID Type { 3: < {width} } Dev Bus {4} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , ansiprint . bold ( ) , ' Name ' , ansiprint . end ( ) , width = name_length ) )
2018-09-20 03:25:58 -04:00
for disk in ddisks :
ainformation . append ( ' {0: <3} {1: <5} { 2: < {width} } {3: <4} {4: <5} ' . format ( ddisks . index ( disk ) , disk [ ' type ' ] , disk [ ' name ' ] , disk [ ' dev ' ] , disk [ ' bus ' ] , width = name_length ) )
ainformation . append ( ' ' )
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {} Interfaces: {} {} ID Type Source Model MAC {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , ansiprint . bold ( ) , ansiprint . end ( ) ) )
2018-09-20 03:25:58 -04:00
for net in dnets :
ainformation . append ( ' {0: <3} {1: <8} {2: <10} {3: <8} {4} ' . format ( dnets . index ( net ) , net [ ' type ' ] , net [ ' source ' ] , net [ ' model ' ] , net [ ' mac ' ] ) )
# Controller list
ainformation . append ( ' ' )
2018-10-20 15:28:25 -04:00
ainformation . append ( ' {} Controllers: {} {} ID Type Model {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , ansiprint . bold ( ) , ansiprint . end ( ) ) )
2018-09-20 03:25:58 -04:00
for controller in dcontrollers :
ainformation . append ( ' {0: <3} {1: <14} {2: <8} ' . format ( dcontrollers . index ( controller ) , controller [ ' type ' ] , controller [ ' model ' ] ) )
# Join it all together
information = ' \n ' . join ( ainformation )
return information
#
# Cluster search functions
#
def getClusterDomainList ( zk_conn ) :
# Get a list of UUIDs by listing the children of /domains
2018-10-27 15:27:08 -04:00
uuid_list = zkhandler . listchildren ( zk_conn , ' /domains ' )
2018-09-20 03:25:58 -04:00
name_list = [ ]
# For each UUID, get the corresponding name from the data
for uuid in uuid_list :
2018-10-27 15:27:08 -04:00
name_list . append ( zkhandler . readdata ( zk_conn , ' /domains/ %s ' % uuid ) )
2018-09-20 03:25:58 -04:00
return uuid_list , name_list
def searchClusterByUUID ( zk_conn , uuid ) :
try :
# Get the lists
uuid_list , name_list = getClusterDomainList ( zk_conn )
# We're looking for UUID, so find that element ID
index = uuid_list . index ( uuid )
# Get the name_list element at that index
name = name_list [ index ]
except ValueError :
# We didn't find anything
return None
return name
def searchClusterByName ( zk_conn , name ) :
try :
# Get the lists
uuid_list , name_list = getClusterDomainList ( zk_conn )
# We're looking for name, so find that element ID
index = name_list . index ( name )
# Get the uuid_list element at that index
uuid = uuid_list [ index ]
except ValueError :
# We didn't find anything
return None
return uuid
def getDomainUUID ( zk_conn , domain ) :
# Validate and obtain alternate passed value
if common . validateUUID ( domain ) :
dom_name = searchClusterByUUID ( zk_conn , domain )
dom_uuid = searchClusterByName ( zk_conn , dom_name )
else :
dom_uuid = searchClusterByName ( zk_conn , domain )
dom_name = searchClusterByUUID ( zk_conn , dom_uuid )
return dom_uuid
2018-09-25 02:26:37 -04:00
def getDomainName ( zk_conn , domain ) :
# Validate and obtain alternate passed value
if common . validateUUID ( domain ) :
dom_name = searchClusterByUUID ( zk_conn , domain )
dom_uuid = searchClusterByName ( zk_conn , dom_name )
else :
dom_uuid = searchClusterByName ( zk_conn , domain )
dom_name = searchClusterByUUID ( zk_conn , dom_uuid )
return dom_name
2018-09-20 03:25:58 -04:00
#
# Direct functions
#
2018-10-14 02:01:35 -04:00
def define_vm ( zk_conn , config_data , target_node , selector ) :
2018-09-20 03:25:58 -04:00
# Parse the XML data
2018-09-20 03:46:05 -04:00
parsed_xml = lxml . objectify . fromstring ( config_data )
2018-09-20 03:25:58 -04:00
dom_uuid = parsed_xml . uuid . text
dom_name = parsed_xml . name . text
click . echo ( ' Adding new VM with Name " {} " and UUID " {} " to database. ' . format ( dom_name , dom_uuid ) )
2018-10-14 02:01:35 -04:00
if target_node == None :
target_node = common . findTargetNode ( zk_conn , selector , dom_uuid )
2018-09-20 03:25:58 -04:00
# Verify node is valid
2018-10-14 02:01:35 -04:00
common . verifyNode ( zk_conn , target_node )
2018-09-20 03:25:58 -04:00
# Add the new domain to Zookeeper
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , {
' /domains/ {} ' . format ( dom_uuid ) : dom_name ,
' /domains/ {} /state ' . format ( dom_uuid ) : ' stop ' ,
' /domains/ {} /node ' . format ( dom_uuid ) : target_node ,
' /domains/ {} /lastnode ' . format ( dom_uuid ) : ' ' ,
' /domains/ {} /failedreason ' . format ( dom_uuid ) : ' ' ,
' /domains/ {} /xml ' . format ( dom_uuid ) : config_data
} )
2018-09-20 03:25:58 -04:00
return True , ' '
2018-09-25 02:32:08 -04:00
def modify_vm ( zk_conn , domain , restart , new_vm_config ) :
2018-09-20 03:25:58 -04:00
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
2018-09-25 02:32:08 -04:00
dom_name = getDomainName ( zk_conn , domain )
2018-09-20 03:25:58 -04:00
# Add the modified config to Zookeeper
2018-10-27 15:27:08 -04:00
zk_data = {
' /domains/ {} ' . format ( dom_uuid ) : dom_name ,
' /domains/ {} /xml ' . format ( dom_uuid ) : new_vm_config
}
2018-09-20 03:25:58 -04:00
if restart == True :
2018-10-27 15:27:08 -04:00
zk_data . update ( { ' /domains/ {} /state ' . format ( dom_uuid ) : ' restart ' } )
zkhandler . writedata ( zk_conn , zk_data )
2018-09-20 03:25:58 -04:00
return True , ' '
2019-03-12 21:09:54 -04:00
def dump_vm ( zk_conn , domain ) :
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Gram the domain XML and dump it to stdout
vm_xml = zkhandler . readdata ( zk_conn , ' /domains/ {} /xml ' . format ( dom_uuid ) )
click . echo ( vm_xml )
return True , ' '
2018-09-20 03:25:58 -04:00
def undefine_vm ( zk_conn , domain ) :
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Shut down the VM
try :
2018-10-27 15:27:08 -04:00
current_vm_state = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
if current_vm_state != ' stop ' :
click . echo ( ' Forcibly stopping VM " {} " . ' . format ( dom_uuid ) )
# Set the domain into stop mode
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , { ' /domains/ {} /state ' . format ( dom_uuid ) : ' stop ' } )
2018-09-20 03:25:58 -04:00
2018-10-14 02:01:35 -04:00
# Wait for 3 seconds to allow state to flow to all nodes
2018-09-20 03:25:58 -04:00
click . echo ( ' Waiting for cluster to update. ' )
time . sleep ( 1 )
except :
pass
# Gracefully terminate the class instances
try :
click . echo ( ' Deleting VM " {} " from nodes. ' . format ( dom_uuid ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , { ' /domains/ {} /state ' . format ( dom_uuid ) : ' delete ' } )
2018-09-20 03:25:58 -04:00
time . sleep ( 5 )
except :
pass
# Delete the configurations
try :
click . echo ( ' Undefining VM " {} " . ' . format ( dom_uuid ) )
2019-03-15 01:13:03 -04:00
zkhandler . deletekey ( zk_conn , ' /domains/ {} ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
except :
pass
return True , ' '
def start_vm ( zk_conn , domain ) :
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Set the VM to start
click . echo ( ' Starting VM " {} " . ' . format ( dom_uuid ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , { ' /domains/ {} /state ' . format ( dom_uuid ) : ' start ' } )
2018-09-20 03:25:58 -04:00
return True , ' '
def restart_vm ( zk_conn , domain ) :
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Get state and verify we're OK to proceed
2018-10-27 15:27:08 -04:00
current_state = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
if current_state != ' start ' :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: VM " {} " is not in " start " state! ' . format ( dom_uuid )
# Set the VM to start
click . echo ( ' Restarting VM " {} " . ' . format ( dom_uuid ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , { ' /domains/ {} /state ' . format ( dom_uuid ) : ' restart ' } )
2018-09-20 03:25:58 -04:00
return True , ' '
def shutdown_vm ( zk_conn , domain ) :
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Get state and verify we're OK to proceed
2018-10-27 15:27:08 -04:00
current_state = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
if current_state != ' start ' :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: VM " {} " is not in " start " state! ' . format ( dom_uuid )
# Set the VM to shutdown
click . echo ( ' Shutting down VM " {} " . ' . format ( dom_uuid ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , { ' /domains/ {} /state ' . format ( dom_uuid ) : ' shutdown ' } )
2018-09-20 03:25:58 -04:00
return True , ' '
def stop_vm ( zk_conn , domain ) :
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Get state and verify we're OK to proceed
2018-10-27 15:27:08 -04:00
current_state = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
# Set the VM to start
click . echo ( ' Forcibly stopping VM " {} " . ' . format ( dom_uuid ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , { ' /domains/ {} /state ' . format ( dom_uuid ) : ' stop ' } )
2018-09-20 03:25:58 -04:00
return True , ' '
2018-10-14 02:01:35 -04:00
def move_vm ( zk_conn , domain , target_node , selector ) :
2018-09-20 03:25:58 -04:00
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
2018-10-27 15:27:08 -04:00
current_node = zkhandler . readdata ( zk_conn , ' /domains/ {} /node ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
2018-10-14 02:01:35 -04:00
if target_node == None :
target_node = common . findTargetNode ( zk_conn , selector , dom_uuid )
2018-09-20 03:25:58 -04:00
else :
2018-10-14 02:01:35 -04:00
if target_node == current_node :
2018-09-20 03:25:58 -04:00
common . stopZKConnection ( zk_conn )
2018-10-14 02:01:35 -04:00
return False , ' ERROR: VM " {} " is already running on node " {} " . ' . format ( dom_uuid , current_node )
2018-09-20 03:25:58 -04:00
# Verify node is valid
2018-10-14 02:01:35 -04:00
common . verifyNode ( zk_conn , target_node )
2018-09-20 03:25:58 -04:00
2018-10-27 15:27:08 -04:00
current_vm_state = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
if current_vm_state == ' start ' :
2018-10-14 02:01:35 -04:00
click . echo ( ' Permanently migrating VM " {} " to node " {} " . ' . format ( dom_uuid , target_node ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , {
' /domains/ {} /state ' . format ( dom_uuid ) : ' migrate ' ,
' /domains/ {} /node ' . format ( dom_uuid ) : target_node ,
' /domains/ {} /lastnode ' . format ( dom_uuid ) : ' '
} )
2018-09-20 03:25:58 -04:00
else :
2018-10-14 02:01:35 -04:00
click . echo ( ' Permanently moving VM " {} " to node " {} " . ' . format ( dom_uuid , target_node ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , {
' /domains/ {} /node ' . format ( dom_uuid ) : target_node ,
' /domains/ {} /lastnode ' . format ( dom_uuid ) : ' '
} )
2018-09-20 03:25:58 -04:00
return True , ' '
2018-10-14 02:01:35 -04:00
def migrate_vm ( zk_conn , domain , target_node , selector , force_migrate ) :
2018-09-20 03:25:58 -04:00
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Get state and verify we're OK to proceed
2018-10-27 15:27:08 -04:00
current_state = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
if current_state != ' start ' :
target_state = ' start '
else :
target_state = ' migrate '
2018-10-27 15:27:08 -04:00
current_node = zkhandler . readdata ( zk_conn , ' /domains/ {} /node ' . format ( dom_uuid ) )
last_node = zkhandler . readdata ( zk_conn , ' /domains/ {} /lastnode ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
2018-10-14 02:01:35 -04:00
if last_node != ' ' and force_migrate != True :
2018-09-20 03:25:58 -04:00
click . echo ( ' ERROR: VM " {} " has been previously migrated. ' . format ( dom_uuid ) )
2018-10-14 02:01:35 -04:00
click . echo ( ' > Last node: {} ' . format ( last_node ) )
click . echo ( ' > Current node: {} ' . format ( current_node ) )
click . echo ( ' Run `vm unmigrate` to restore the VM to its previous node, or use `--force` to override this check. ' )
2018-09-20 03:25:58 -04:00
common . stopZKConnection ( zk_conn )
return False , ' '
2018-10-14 02:01:35 -04:00
if target_node == None :
2018-10-21 22:10:05 -04:00
target_node = common . findTargetNode ( zk_conn , selector , dom_uuid )
2018-09-20 03:25:58 -04:00
else :
2018-10-14 02:01:35 -04:00
if target_node == current_node :
2018-09-20 03:25:58 -04:00
common . stopZKConnection ( zk_conn )
2018-10-14 02:01:35 -04:00
return False , ' ERROR: VM " {} " is already running on node " {} " . ' . format ( dom_uuid , current_node )
2018-09-20 03:25:58 -04:00
# Verify node is valid
2018-10-14 02:01:35 -04:00
common . verifyNode ( zk_conn , target_node )
2018-09-20 03:25:58 -04:00
2018-10-14 02:01:35 -04:00
click . echo ( ' Migrating VM " {} " to node " {} " . ' . format ( dom_uuid , target_node ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , {
' /domains/ {} /state ' . format ( dom_uuid ) : ' migrate ' ,
' /domains/ {} /node ' . format ( dom_uuid ) : target_node ,
' /domains/ {} /lastnode ' . format ( dom_uuid ) : current_node
} )
2018-09-20 03:25:58 -04:00
return True , ' '
def unmigrate_vm ( zk_conn , domain ) :
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Get state and verify we're OK to proceed
2018-10-27 15:27:08 -04:00
current_state = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
if current_state != ' start ' :
target_state = ' start '
else :
target_state = ' migrate '
2018-10-27 15:27:08 -04:00
target_node = zkhandler . readdata ( zk_conn , ' /domains/ {} /lastnode ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
2018-10-14 02:01:35 -04:00
if target_node == ' ' :
2018-09-20 03:25:58 -04:00
common . stopZKConnection ( zk_conn )
return False , ' ERROR: VM " {} " has not been previously migrated. ' . format ( dom_uuid )
2018-10-14 02:01:35 -04:00
click . echo ( ' Unmigrating VM " {} " back to node " {} " . ' . format ( dom_uuid , target_node ) )
2018-10-27 15:27:08 -04:00
zkhandler . writedata ( zk_conn , {
' /domains/ {} /state ' . format ( dom_uuid ) : target_state ,
' /domains/ {} /node ' . format ( dom_uuid ) : target_node ,
' /domains/ {} /lastnode ' . format ( dom_uuid ) : ' '
} )
2018-09-20 03:25:58 -04:00
return True , ' '
def get_info ( zk_conn , domain , long_output ) :
# Validate and obtain alternate passed value
dom_uuid = getDomainUUID ( zk_conn , domain )
if dom_uuid == None :
common . stopZKConnection ( zk_conn )
return False , ' ERROR: Could not find VM " {} " in the cluster! ' . format ( domain )
# Gather information from XML config and print it
information = getInformationFromXML ( zk_conn , dom_uuid , long_output )
click . echo ( information )
# Get a failure reason if applicable
2018-10-27 15:27:08 -04:00
failedreason = zkhandler . readdata ( zk_conn , ' /domains/ {} /failedreason ' . format ( dom_uuid ) )
2018-09-20 03:25:58 -04:00
if failedreason != ' ' :
click . echo ( ' ' )
2018-10-20 15:28:25 -04:00
click . echo ( ' {} Failure reason: {} {} ' . format ( ansiprint . purple ( ) , ansiprint . end ( ) , failedreason ) )
2018-09-20 03:25:58 -04:00
click . echo ( ' ' )
return True , ' '
2019-03-12 21:30:01 -04:00
def get_list ( zk_conn , node , limit , raw ) :
2018-10-14 02:01:35 -04:00
if node != None :
2018-09-20 03:25:58 -04:00
# Verify node is valid
2018-10-14 02:01:35 -04:00
common . verifyNode ( zk_conn , node )
2018-09-20 03:25:58 -04:00
2018-10-27 15:27:08 -04:00
full_vm_list = zkhandler . listchildren ( zk_conn , ' /domains ' )
2018-09-20 03:25:58 -04:00
vm_list = [ ]
vm_list_output = [ ]
2018-10-14 02:01:35 -04:00
vm_node = { }
2018-09-20 03:25:58 -04:00
vm_state = { }
vm_migrated = { }
vm_uuid = { }
vm_name = { }
vm_description = { }
vm_memory = { }
vm_vcpu = { }
2018-10-14 02:01:35 -04:00
vm_nets = { }
2018-09-20 03:25:58 -04:00
2019-03-12 23:52:59 -04:00
# Set our limit to a sensible regex
if limit != None :
try :
# Implcitly assume fuzzy limits
if re . match ( ' \ ^.* ' , limit ) == None :
limit = ' .* ' + limit
if re . match ( ' .* \ $ ' , limit ) == None :
limit = limit + ' .* '
except Exception as e :
return False , ' Regex Error: {} ' . format ( e )
2018-09-20 03:25:58 -04:00
# If we're limited, remove other nodes' VMs
2018-09-25 02:20:32 -04:00
for vm in full_vm_list :
2018-09-20 03:25:58 -04:00
# Check we don't match the limit
2018-09-25 02:20:32 -04:00
name = zkhandler . readdata ( zk_conn , ' /domains/ {} ' . format ( vm ) )
2019-03-12 21:39:17 -04:00
vm_node [ vm ] = zkhandler . readdata ( zk_conn , ' /domains/ {} /node ' . format ( vm ) )
2019-03-12 23:52:59 -04:00
# Handle limiting
2018-09-20 03:25:58 -04:00
if limit != None :
try :
2018-09-25 02:20:32 -04:00
if re . match ( limit , vm ) != None :
2018-10-14 02:01:35 -04:00
if node == None :
2018-09-25 02:20:32 -04:00
vm_list . append ( vm )
else :
2018-10-14 02:01:35 -04:00
if vm_node [ vm ] == node :
2018-09-25 02:20:32 -04:00
vm_list . append ( vm )
if re . match ( limit , name ) != None :
2018-10-14 02:01:35 -04:00
if node == None :
2018-09-25 02:20:32 -04:00
vm_list . append ( vm )
else :
2018-10-14 02:01:35 -04:00
if vm_node [ vm ] == node :
2018-09-25 02:20:32 -04:00
vm_list . append ( vm )
2018-09-20 03:25:58 -04:00
except Exception as e :
2018-09-25 02:20:32 -04:00
return False , ' Regex Error: {} ' . format ( e )
2018-09-20 03:25:58 -04:00
else :
2018-10-14 02:01:35 -04:00
# Check node to avoid unneeded ZK calls
if node == None :
2018-09-20 03:25:58 -04:00
vm_list . append ( vm )
2018-09-25 02:20:32 -04:00
else :
2018-10-14 02:01:35 -04:00
if vm_node [ vm ] == node :
2018-09-25 02:20:32 -04:00
vm_list . append ( vm )
2018-09-20 03:25:58 -04:00
# Gather information for printing
for vm in vm_list :
2018-10-27 15:27:08 -04:00
vm_state [ vm ] = zkhandler . readdata ( zk_conn , ' /domains/ {} /state ' . format ( vm ) )
vm_lastnode = zkhandler . readdata ( zk_conn , ' /domains/ {} /lastnode ' . format ( vm ) )
2018-10-14 02:01:35 -04:00
if vm_lastnode != ' ' :
vm_migrated [ vm ] = ' from {} ' . format ( vm_lastnode )
2018-09-20 03:25:58 -04:00
else :
vm_migrated [ vm ] = ' no '
try :
vm_xml = common . getDomainXML ( zk_conn , vm )
vm_uuid [ vm ] , vm_name [ vm ] , vm_description [ vm ] , vm_memory [ vm ] , vm_vcpu [ vm ] , vm_vcputopo = common . getDomainMainDetails ( vm_xml )
2018-10-14 02:01:35 -04:00
dnets = common . getDomainNetworks ( vm_xml )
2018-10-20 15:27:07 -04:00
vm_nets [ vm ] = [ ]
2018-10-14 02:01:35 -04:00
for net in dnets :
# Split out just the numerical (VNI) part of the brXXXX name
2018-11-18 17:48:14 -05:00
net_vnis = re . findall ( r ' \ d+ ' , net [ ' source ' ] )
if net_vnis :
net_vni = net_vnis [ 0 ]
else :
net_vni = re . sub ( ' br ' , ' ' , net [ ' source ' ] )
2018-10-20 15:27:07 -04:00
vm_nets [ vm ] . append ( net_vni )
2018-09-20 03:25:58 -04:00
except AttributeError :
click . echo ( ' Error: Domain {} does not exist. ' . format ( domain ) )
2019-03-12 21:40:52 -04:00
if raw :
2019-03-12 21:46:09 -04:00
for vm in sorted ( vm_name . values ( ) ) :
click . echo ( vm )
2019-03-12 21:40:52 -04:00
return True , ' '
2018-09-20 03:25:58 -04:00
# Determine optimal column widths
2018-10-14 02:01:35 -04:00
# Dynamic columns: node_name, node, migrated
2018-11-01 23:24:38 -04:00
vm_name_length = 5
2018-11-02 00:42:44 -04:00
vm_uuid_length = 37
vm_state_length = 6
2018-10-14 02:01:35 -04:00
vm_nets_length = 9
2018-11-02 00:42:44 -04:00
vm_ram_length = 8
vm_vcpu_length = 6
vm_node_length = 8
2018-10-14 02:01:35 -04:00
vm_migrated_length = 10
2018-09-20 03:25:58 -04:00
for vm in vm_list :
# vm_name column
_vm_name_length = len ( vm_name [ vm ] ) + 1
if _vm_name_length > vm_name_length :
vm_name_length = _vm_name_length
2018-11-02 00:42:44 -04:00
# vm_state column
_vm_state_length = len ( vm_state [ vm ] ) + 1
if _vm_state_length > vm_state_length :
vm_state_length = _vm_state_length
2018-10-14 02:01:35 -04:00
# vm_nets column
2018-10-20 15:27:07 -04:00
_vm_nets_length = len ( ' , ' . join ( vm_nets [ vm ] ) ) + 1
2018-10-14 02:01:35 -04:00
if _vm_nets_length > vm_nets_length :
vm_nets_length = _vm_nets_length
2018-11-02 00:42:44 -04:00
# vm_node column
_vm_node_length = len ( vm_node [ vm ] ) + 1
if _vm_node_length > vm_node_length :
vm_node_length = _vm_node_length
2018-09-20 03:25:58 -04:00
# vm_migrated column
_vm_migrated_length = len ( vm_migrated [ vm ] ) + 1
if _vm_migrated_length > vm_migrated_length :
vm_migrated_length = _vm_migrated_length
# Format the string (header)
vm_list_output . append (
2018-11-02 00:42:44 -04:00
' {bold} { vm_name: < {vm_name_length} } { vm_uuid: < {vm_uuid_length} } \
{ vm_state_colour } { vm_state : < { vm_state_length } } { end_colour } \
2018-10-14 02:01:35 -04:00
{ vm_networks : < { vm_nets_length } } \
2018-11-02 00:42:44 -04:00
{ vm_memory : < { vm_ram_length } } { vm_vcpu : < { vm_vcpu_length } } \
2018-10-14 02:01:35 -04:00
{ vm_node : < { vm_node_length } } \
2018-09-20 03:25:58 -04:00
{ vm_migrated : < { vm_migrated_length } } { end_bold } ' .format(
vm_name_length = vm_name_length ,
2018-11-02 00:42:44 -04:00
vm_uuid_length = vm_uuid_length ,
vm_state_length = vm_state_length ,
2018-10-14 02:01:35 -04:00
vm_nets_length = vm_nets_length ,
2018-11-02 00:42:44 -04:00
vm_ram_length = vm_ram_length ,
vm_vcpu_length = vm_vcpu_length ,
vm_node_length = vm_node_length ,
2018-09-20 03:25:58 -04:00
vm_migrated_length = vm_migrated_length ,
2018-10-20 15:28:25 -04:00
bold = ansiprint . bold ( ) ,
end_bold = ansiprint . end ( ) ,
2018-09-20 03:25:58 -04:00
vm_state_colour = ' ' ,
end_colour = ' ' ,
vm_name = ' Name ' ,
vm_uuid = ' UUID ' ,
vm_state = ' State ' ,
2018-10-14 02:01:35 -04:00
vm_networks = ' Networks ' ,
2018-11-01 23:24:38 -04:00
vm_memory = ' RAM (M) ' ,
2018-09-20 03:25:58 -04:00
vm_vcpu = ' vCPUs ' ,
2018-10-14 02:01:35 -04:00
vm_node = ' Node ' ,
2018-09-20 03:25:58 -04:00
vm_migrated = ' Migrated '
)
)
# Format the string (elements)
for vm in vm_list :
if vm_state [ vm ] == ' start ' :
2018-10-20 15:28:25 -04:00
vm_state_colour = ansiprint . green ( )
2018-09-20 03:25:58 -04:00
elif vm_state [ vm ] == ' restart ' :
2018-10-20 15:28:25 -04:00
vm_state_colour = ansiprint . yellow ( )
2018-09-20 03:25:58 -04:00
elif vm_state [ vm ] == ' shutdown ' :
2018-10-20 15:28:25 -04:00
vm_state_colour = ansiprint . yellow ( )
2018-09-20 03:25:58 -04:00
elif vm_state [ vm ] == ' stop ' :
2018-10-20 15:28:25 -04:00
vm_state_colour = ansiprint . red ( )
2018-09-20 03:25:58 -04:00
elif vm_state [ vm ] == ' failed ' :
2018-10-20 15:28:25 -04:00
vm_state_colour = ansiprint . red ( )
2018-09-20 03:25:58 -04:00
else :
2018-10-20 15:28:25 -04:00
vm_state_colour = ansiprint . blue ( )
2018-09-20 03:25:58 -04:00
2018-10-20 15:27:07 -04:00
# Handle colouring for an invalid network config
net_list = [ ]
2018-10-20 15:28:25 -04:00
vm_nets_colour = ansiprint . end ( )
2018-10-20 15:27:07 -04:00
for net in vm_nets [ vm ] :
net_exists = zkhandler . exists ( zk_conn , ' /networks/ {} ' . format ( net ) )
net_list . append ( net )
2018-11-18 17:48:14 -05:00
if not net_exists and net != ' cluster ' :
2018-10-20 15:28:25 -04:00
vm_nets_colour = ansiprint . red ( )
2018-10-20 15:27:07 -04:00
vm_nets [ vm ] = ' , ' . join ( net_list )
2018-09-20 03:25:58 -04:00
vm_list_output . append (
2018-11-02 00:42:44 -04:00
' {bold} { vm_name: < {vm_name_length} } { vm_uuid: < {vm_uuid_length} } \
{ vm_state_colour } { vm_state : < { vm_state_length } } { end_colour } \
2018-10-20 15:27:07 -04:00
{ vm_nets_colour } { vm_networks : < { vm_nets_length } } { end_colour } \
2018-11-02 00:42:44 -04:00
{ vm_memory : < { vm_ram_length } } { vm_vcpu : < { vm_vcpu_length } } \
2018-10-14 02:01:35 -04:00
{ vm_node : < { vm_node_length } } \
2018-09-20 03:25:58 -04:00
{ vm_migrated : < { vm_migrated_length } } { end_bold } ' .format(
vm_name_length = vm_name_length ,
2018-11-02 00:42:44 -04:00
vm_uuid_length = vm_uuid_length ,
vm_state_length = vm_state_length ,
2018-10-14 02:01:35 -04:00
vm_nets_length = vm_nets_length ,
2018-11-02 00:42:44 -04:00
vm_ram_length = vm_ram_length ,
vm_vcpu_length = vm_vcpu_length ,
vm_node_length = vm_node_length ,
2018-09-20 03:25:58 -04:00
vm_migrated_length = vm_migrated_length ,
bold = ' ' ,
end_bold = ' ' ,
vm_state_colour = vm_state_colour ,
2018-10-20 15:27:07 -04:00
vm_nets_colour = vm_nets_colour ,
2018-10-20 15:28:25 -04:00
end_colour = ansiprint . end ( ) ,
2018-09-20 03:25:58 -04:00
vm_name = vm_name [ vm ] ,
vm_uuid = vm_uuid [ vm ] ,
vm_state = vm_state [ vm ] ,
2018-10-14 02:01:35 -04:00
vm_networks = vm_nets [ vm ] ,
2018-09-20 03:25:58 -04:00
vm_memory = vm_memory [ vm ] ,
vm_vcpu = vm_vcpu [ vm ] ,
2018-10-14 02:01:35 -04:00
vm_node = vm_node [ vm ] ,
2018-09-20 03:25:58 -04:00
vm_migrated = vm_migrated [ vm ]
)
)
click . echo ( ' \n ' . join ( sorted ( vm_list_output ) ) )
return True , ' '