Skip to content

Commit

Permalink
Move common code to esxi_vm_functions,
Browse files Browse the repository at this point in the history
  • Loading branch information
@andrivet
andrivet committed Jan 11, 2018
1 parent d7c4e3c commit de57ce5
Show file tree
Hide file tree
Showing 3 changed files with 69 additions and 74 deletions.
54 changes: 16 additions & 38 deletions esxi-vm-create
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,14 +7,6 @@ import paramiko # For remote ssh
from esxi_vm_functions import *


def exec_ssh_command(message, command, verbose):
if verbose:
if message:
print(message)
print("SSH: " + command)
return ssh.exec_command(command)


# Defaults and Variable setup
ConfigData = setup_config()
NAME = ""
Expand Down Expand Up @@ -146,35 +138,20 @@ if args.UPDATE:
if NAME == "":
sys.exit(0)

#
# main()
#
LogOutput = '{'
LogOutput += '"datetime":"{}",'.format(the_current_date_time())

if NAME == "":
print("ERROR: Missing required option --name")
sys.exit(1)

try:
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(HOST, port=PORT, username=USER, password=PASSWORD, key_filename=KEY)
LogOutput = '{'
LogOutput += '"datetime":"{}",'.format(the_current_date_time())

(stdin, stdout, stderr) = exec_ssh_command("Get ESXi version", "esxcli system version get |grep Version", isVerbose)
if re.match("Version", str(stdout.readlines())) is not None:
print("Unable to determine if this is a ESXi Host: {}, port: {}, username: {}".format(HOST, PORT, USER))
sys.exit(1)
except Exception as e:
print("The Error is {}".format(e))
print("Unable to access ESXi Host: {}, port: {}, username: {}".format(HOST, PORT, USER))
sys.exit(1)
ssh = connect_to_esxi(HOST, PORT, USER, PASSWORD, KEY, isVerbose)

try:
(stdin, stdout, stderr) = \
exec_ssh_command("Get existing volumes",
"esxcli storage filesystem list |grep '/vmfs/volumes/.*true VMFS' |sort -nk7",
isVerbose)
ssh, isVerbose)
VOLUMES = {}
for line in stdout.readlines():
splitLine = line.split()
Expand All @@ -192,7 +169,7 @@ try:
(stdin, stdout, stderr) = \
exec_ssh_command("Get existing networks",
"esxcli network vswitch standard list|grep Portgroups|sed 's/^ Portgroups: //g'",
isVerbose)
ssh, isVerbose)
VMNICS = []
for line in stdout.readlines():
splitLine = re.split('[,\n]', line)
Expand Down Expand Up @@ -224,14 +201,14 @@ if ISO != "":
(stdin, stdout, stderr) = exec_ssh_command("Search ISO image",
"find /vmfs/volumes/ -type f -name {}".format(ISO) +
" -exec sh -c 'echo $1; kill $PPID' sh {} 2>/dev/null \;",
isVerbose
ssh, isVerbose
)
FoundISOPath = str(stdout.readlines()[0]).strip('\n')
if isVerbose:
print("FoundISOPath: {}".format(FoundISOPath))
ISO = str(FoundISOPath)

(stdin, stdout, stderr) = exec_ssh_command("Check ISO", "ls {}".format(ISO), isVerbose)
(stdin, stdout, stderr) = exec_ssh_command("Check ISO", "ls {}".format(ISO), ssh, isVerbose)
if stdout.readlines() and not stderr.readlines():
ISOfound = True

Expand All @@ -241,7 +218,7 @@ if ISO != "":

VMID = -1
try:
(stdin, stdout, stderr) = exec_ssh_command("Get list of VMs", "vim-cmd vmsvc/getallvms", isVerbose)
(stdin, stdout, stderr) = exec_ssh_command("Get list of VMs", "vim-cmd vmsvc/getallvms", ssh, isVerbose)
for line in stdout.readlines():
splitLine = line.split()
if NAME == splitLine[1]:
Expand Down Expand Up @@ -301,7 +278,7 @@ if ISO != "" and not ISOfound:
# Check if DSPATH/NAME already exists
try:
FullPath = DSPATH + "/" + NAME
(stdin, stdout, stderr) = exec_ssh_command("List VMs directories", "ls -d {}".format(FullPath), isVerbose)
(stdin, stdout, stderr) = exec_ssh_command("List VMs directories", "ls -d {}".format(FullPath), ssh, isVerbose)
if stdout.readlines() and not stderr.readlines():
print("ERROR: Directory {} already exists.".format(FullPath))
ErrorMessages += " Directory {} already exists.".format(FullPath)
Expand Down Expand Up @@ -393,28 +370,29 @@ else:
if not isDryRun and not CheckHasErrors:
try:
(stdin, stdout, stderr) = exec_ssh_command("Create {}.vmx file".format(NAME),
"mkdir {}".format(FullPath), isVerbose)
"mkdir {}".format(FullPath), ssh, isVerbose)
for k, v in vmx.items():
(stdin, stdout, stderr) = exec_ssh_command(None, "echo '{} = \"{}\"' >> {}.vmx".format(k, v, MyVM),
isVerbose)
ssh, isVerbose)

(stdin, stdout, stderr) = exec_ssh_command("Create {}.vmdk file".format(NAME),
"vmkfstools -c {}G -d {} {}.vmdk".format(HDISK, DISKFORMAT, MyVM),
isVerbose)
ssh, isVerbose)

(stdin, stdout, stderr) = exec_ssh_command("Register VM",
"vim-cmd solo/registervm {}.vmx".format(MyVM), isVerbose)
"vim-cmd solo/registervm {}.vmx".format(MyVM), ssh, isVerbose)
VMID = int(stdout.readlines()[0])

(stdin, stdout, stderr) = exec_ssh_command("Power ON VM", "vim-cmd vmsvc/power.on {}".format(VMID), isVerbose)
(stdin, stdout, stderr) = exec_ssh_command("Power ON VM", "vim-cmd vmsvc/power.on {}".format(VMID),
ssh, isVerbose)
if stderr.readlines():
print("Error Powering-on VM.")
Result = "Fail"

if NET != "None":
(stdin, stdout, stderr) = exec_ssh_command("Get MAC address",
"grep -i 'ethernet0.*ddress = ' {}.vmx".format(MyVM) +
" |tail -1|awk '{print $NF}'", isVerbose)
" |tail -1|awk '{print $NF}'", ssh, isVerbose)
GeneratedMAC = str(stdout.readlines()[0]).strip('\n"')

except Exception as e:
Expand Down
51 changes: 16 additions & 35 deletions esxi-vm-destroy
View file
Original file line number Diff line number Diff line change
Expand Up @@ -71,36 +71,20 @@ if args.KEY:
if args.NAME:
NAME = args.NAME

#
# main()
#
LogOutput = '{'
LogOutput += '"datetime":"' + str(the_current_date_time()) + '",'

if NAME == "":
print("ERROR: Missing required option --name")
sys.exit(1)

try:
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(HOST, port=PORT, username=USER, password=PASSWORD, key_filename=KEY)
LogOutput = '{'
LogOutput += '"datetime":"{}",'.format(the_current_date_time())

(stdin, stdout, stderr) = ssh.exec_command("esxcli system version get |grep Version")
type(stdin)
if re.match("Version", str(stdout.readlines())) is not None:
print("Unable to determine if this is a ESXi Host: {}, port: {}, username: {}".format(HOST, PORT, USER))
sys.exit(1)
except Exception as e:
print("The Error is {}".format(e))
print("Unable to access ESXi Host: {}, port: {}, username: {}".format(HOST, PORT, USER))
sys.exit(1)
ssh = connect_to_esxi(HOST, PORT, USER, PASSWORD, KEY, isVerbose)

VMID = -1
CheckHasWarnings = False

try:
(stdin, stdout, stderr) = ssh.exec_command("vim-cmd vmsvc/getallvms")
(stdin, stdout, stderr) = exec_ssh_command("Get list of VMs", "vim-cmd vmsvc/getallvms", ssh, isVerbose)
type(stdin)
for line in stdout.readlines():
splitLine = line.split()
Expand All @@ -119,10 +103,11 @@ except Exception as e:
print("The Error is {}".format(e))
sys.exit(1)

# Get List of Volumes,
try:
(stdin, stdout, stderr) = \
ssh.exec_command("esxcli storage filesystem list |grep '/vmfs/volumes/.*true VMFS' |sort -nk7")
exec_ssh_command("Get List of Volumes",
"esxcli storage filesystem list |grep '/vmfs/volumes/.*true VMFS' |sort -nk7",
ssh, isVerbose)
type(stdin)
VOLUMES = {}
for line in stdout.readlines():
Expand Down Expand Up @@ -150,31 +135,27 @@ else:
if not CheckHasErrors:
try:
for i in range(0, 10):
if isVerbose:
print("Get state of VM")
(stdin, stdout, stderr) = ssh.exec_command("vim-cmd vmsvc/power.getstate {}".format(VMID))
type(stdin)
(stdin, stdout, stderr) = exec_ssh_command("Get state of VM",
"vim-cmd vmsvc/power.getstate {}".format(VMID),
ssh, isVerbose)
lines = str(stdout.readlines()) + str(stderr.readlines())
if isVerbose:
print("power.getstate: {}".format(lines))
if re.search("Powered off", lines):
break

if isVerbose:
print("Power OFF VM")
(stdin, stdout, stderr) = ssh.exec_command("vim-cmd vmsvc/power.off {} ||echo".format(VMID))
type(stdin)
(stdin, stdout, stderr) = exec_ssh_command("Power OFF VM",
"vim-cmd vmsvc/power.off {} ||echo".format(VMID),
ssh, isVerbose)
lines = str(stdout.readlines()) + str(stderr.readlines())
if isVerbose:
print("power.off: {}".format(lines))

time.sleep(1)

# destroy VM
if isVerbose:
print("Destroy VM")
(stdin, stdout, stderr) = ssh.exec_command("vim-cmd vmsvc/destroy {}".format(VMID))
type(stdin)
(stdin, stdout, stderr) = exec_ssh_command("Destroy VM",
"vim-cmd vmsvc/destroy {}".format(VMID),
ssh, isVerbose)
lines = str(stdout.readlines()) + str(stderr.readlines())
if isVerbose:
print("destroy: {}".format(lines))
Expand Down
38 changes: 37 additions & 1 deletion esxi_vm_functions.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,8 @@
import sys
import yaml
import datetime # For current Date/Time

import re # For regex
import paramiko # For remote ssh

def setup_config():

Expand Down Expand Up @@ -94,3 +95,38 @@ def save_config(config_data):
def the_current_date_time():
i = datetime.datetime.now()
return str(i.isoformat())


def exec_ssh_command(message, command, ssh, verbose):
if verbose:
if message:
print(message)
print("SSH: " + command)
return ssh.exec_command(command)


def get_esxi_version(ssh, verbose):
try:
(stdin, stdout, stderr) = exec_ssh_command("Get ESXi version", "esxcli system version get |grep Version",
ssh, verbose)
if re.match("Version", str(stdout.readlines())) is not None:
print("Unable to determine if this is a ESXi Host: {}, port: {}, username: {}".format(HOST, PORT, USER))
sys.exit(1)

except Exception as e:
print("Unable to get ESXi version")
print("The Error is {}".format(e))
sys.exit(1)


def connect_to_esxi(host, port, user, password, key, verbose):
try:
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(host, port=port, username=user, password=password, key_filename=key)
get_esxi_version(ssh, verbose)
return ssh
except Exception as e:
print("Unable to access ESXi Host: {}, port: {}, username: {}, key: {}".format(host, port, user, key))
print("The Error is {}".format(e))
sys.exit(1)

0 comments on commit de57ce5

Please sign in to comment.