#!/usr/bin/python

from datetime import datetime

from time import sleep

import re, getpass, sys, xmlrpclib, os

from numpy import loadtxt

from collections import defaultdict

import xlsxwriter

#

# If there is an update to kernel its version will be stored in one of these variables

#

kernel5_ver = ""

kernel6_ver = ""

vmware_config = """#!/bin/bash

test -d /opt/hp/hpsmh && chmod 750 /opt/hp/hpsmh

if [ -x /usr/bin/vmware-config-tools.pl ]; then echo "test -x /usr/bin/vmware-config-tools.pl && /usr/bin/vmware-config-tools.pl -d && /sbin/shutdown -r -t10 now && sed -i '/sed/d' /etc/rc.d/rc.local" >> /etc/rc.d/rc.local; fi"""

nogpg_install = """#!/bin/bash

"""

get_runningkernel = """#!/bin/bash

/bin/uname -r

"""

get_lastreboot = """#!/bin/bash

/usr/bin/last reboot | head -n 1 | awk '{print $5" "$6" "$7" "$8}'

"""

fix_hpsmh_mode = """#!/bin/bash

test -d /opt/hp/hpsmh && chmod 750 /opt/hp/hpsmh

"""

# Python 2.6 doesn't support argparse

try:

import argparse

except ImportError:

from optparse import OptionParser

def vararg_callback(option, opt_str, value, parser):

setattr(parser.values, option.dest, value)

def parsecli():

parser.parse_args()

def connect_to_spacewalk(spacewalk_server, spacewalk_login, spacewalk_password):

return client, key

#

# {'server name': [server id, [package id, package id, ... ], [package name, package name, ...], action id, flag]}

# flag: 0 - failed, 1 - success, 2 - pending

# {'smsk01mg02': [1000010074, [13154, 13154], ['kernel-2.6.18-371.el5-x86_64', 'kernel-2.6.18-371.el5-x86_64'], 8856, 1]}

#

def prepareupdate(key, servers):

sys.exit(1)

def getdetails(key, s):

"""Get system details"""

return client.system.getDetails(key, s)

def checkforupdates(key, s):

return pids, pnames

def doupdate (key, s):

return action_id

def list_pending(key):

"Return pending actions info."

return client.schedule.listInProgressActions(key)

def list_failed_systems(key, aid):

return False

def list_completed_systems(key, aid):

return False

def list_pending_systems(key, aid):

return False

def getlastboot(key, sid):

return client.system.getDetails(key, sid)['last_boot']

def getosastatusbyactionid(key, p_action):

"""Return OSA Dispatcher status of a system based on pending action id number."""

server_id = client.schedule.listInProgressSystems(key, p_action)[0]['server_id']

return client.system.getDetails(key, server_id)['osa_status']

def getosastatus(key, server_id):

return client.system.getDetails(key, server_id)['osa_status']

def postcheck (key, s):

workbook.close()

if __name__ == '__main__':

opt = parsecli()

if opt.csv and os.path.exists(opt.csv):

try:

data = loadtxt(opt.csv, dtype='string', delimiter=';', usecols=(0,8))

except IOError:

print ("Could not open " + opt.csv)

sys.exit(-1)

else:

try:

servers_input = data[:,0]

patching_groups = data[:,1]

servers_input = servers_input[patching_groups == opt.patching_group]

except IndexError:

if data.size == 2:

if data[1] == opt.patching_group:

servers_input = data[0:1]

else:

servers_input = opt.servers_list

if len(servers_input) > 0:

# Building a path to the configuration file

conf_dir = os.path.expanduser('~/.spacecmd')

config = os.path.join(conf_dir, 'config')

# Opening and parsing the configuration file

try:

fp = open(config, "r")

except IOError:

try:

spacewalk_server = str(raw_input("Enter Server name: "))

except KeyboardInterrupt:

print ""

sys.exit(-1)

try:

spacewalk_login = str(raw_input("Username: "))

except KeyboardInterrupt:

print ""

sys.exit(-1)

try:

spacewalk_password = getpass.getpass("Password:")

except KeyboardInterrupt:

print ""

sys.exit(-1)

else:

for line in fp.readlines():

m = re.search(r'server=(.+)', line, re.I)

if m:

spacewalk_server = m.group(1)

m = re.search(r'username=(.+)', line, re.I)

if m:

spacewalk_login = m.group(1)

m = re.search(r'password=(.+)', line, re.I)

if m:

spacewalk_password = m.group(1)

fp.close()

servers_to_update = defaultdict(list)

try:

client, key = connect_to_spacewalk(spacewalk_server, spacewalk_login, spacewalk_password)

except:

print "Username or password is incorrect."

sys.exit(-1)

for server in servers_input:

# Some servers have been registered with FQDN names.

server = server.partition(".")[0]

id = client.system.searchByName(key, str(server)+"$|" + str(server) + ".example.com$")

if (id) and (getosastatus(key, id[0]['id'])) == 'online':

servers_to_update[str(server)].append(id[0]['id'])

else:

print ("OSA service is offline. Server " + str(server) + " will be skipped.")

if opt.yes:

print ("The following servers will be updated:\n")

print (servers_to_update.keys())

prepareupdate(key, servers_to_update)

else:

print ("The following server(s) will be updated:\n")

print (servers_to_update.keys())

answer = raw_input("Continue (Y/N): ")

if answer == 'y' or answer == 'Y':

prepareupdate(key, servers_to_update)

else:

print("Quiting...\n")

sys.exit(0)

client.auth.logout(key)