#!/usr/bin/python
import sys
import cgi
import os
import socket
import time
try:
import cgitb
cgitb.enable()
except ImportError:
sys.stderr = sys.stdout
def cgiprint(inline=''):
sys.stdout.write(inline)
sys.stdout.write('\r\n')
sys.stdout.flush()
contentheader = 'Content-Type: text/html'
thepage = '''
%s
%s
'''
h1 = '%s
'
def getform(theform, valuelist, notpresent='', nolist=False):
"""
This function, given a CGI form as a
FieldStorage instance, extracts the
data from it, based on valuelist
passed in. Any non-present values are
set to '' - although this can be
changed. (e.g. to return None so you
can test for missing keywords - where
'' is a valid answer but to have the
field missing isn't.) It also takes a
keyword argument 'nolist'. If this is
True list values only return their
first value.
"""
data = {}
for field in valuelist:
if not theform.has_key(field):
# if the field is not present (or was empty)
data[field] = notpresent
else:
# the field is present
print type(theform[field])
if type(theform[field]) != type([]):
# is it a list or a single item
print type(theform[field])
data[field] = theform[field].value
else:
if not nolist:
# do we want a list ?
data[field] = theform.getlist(field)
else:
data[field] = theform.getfirst(field)
# just fetch the first item
return data
def getall(theform, nolist=False):
"""
Passed a form (cgi.FieldStorage
instance) return *all* the values.
This doesn't take into account
multipart form data (file uploads).
It also takes a keyword argument
'nolist'. If this is True list values
only return their first value.
"""
data = {}
for field in theform.keys():
# we can't just iterate over it, but must use the keys() method
if type(theform[field]) == type([]):
if not nolist:
data[field] = theform.getlist(field)
else:
data[field] = theform.getfirst(field)
else:
data[field] = theform[field].value
return data
def isblank(indict):
"""
Passed an indict of values it checks
if any of the values are set. Returns
True if the indict is empty, else
returns False. I use it on the a form
processed with getform to tell if my
CGI has been activated without any
form values.
"""
for key in indict.keys():
if indict[key]:
return False
return True
def go_backup(myhost):
myhostname = socket.gethostname()
results="Nothing happened"
if myhostname.strip() == myhost.strip():
command= "sudo /usr/MythVantage/bin/backup_job"
command2="sleep 1; /data/srv/hobbit/server/ext/hbnotes.py"
results=os.popen(command,'r')
os.popen(command2,'r')
return results
def go_restore(restorefile,myhost):
myhostname = socket.gethostname()
if myhostname.strip() == myhost.strip():
localcommand="sudo /usr/bin/restore_job.sh "
command= localcommand + restorefile
else:
sshcmd="ssh -o StrictHostKeyChecking=no -o ConnectTimeout=1 -i /data/srv/.nobody_ssh/id_dsa mythtv@"
sshcmd+=myhost.strip()
cmd=' "sudo /usr/bin/restore_job.sh " '
command=sshcmd + cmd + restorefile + " 2>&1 "
results=os.popen(command,'r')
return results
def go_optimize(myhost):
myhostname = socket.gethostname()
if myhostname.strip() == myhost.strip():
command="/usr/LH/bin/optimize_mythdb.py"
results=os.popen(command,'r')
else:
results='This host does not run a database'
return results
def go_update(myhost,update_type):
myhostname = socket.gethostname()
if myhostname.strip() == myhost.strip():
cmd="sudo /usr/bin/update_system "
command=cmd + update_type
else:
cmd=" call pacman update_system "
cmd+=update_type
command="/usr/bin/func \"" + myhost.strip() + "*\" " + cmd
print command
#print command
results=os.popen(command,'r')
return results
def go_updateall(allupdate_type):
cmd=" sudo /usr/bin/update_system_all "
cmd+=allupdate_type
command= cmd + " 2>&1 "
results=os.popen(command,'r')
return results
def go_shutdown(myhost):
myhostname = socket.gethostname()
message="Shutdown from MBE"
command="/usr/bin/func \"" + myhost.strip() + "*\" call msg display \"%s\" " %message
results=os.popen(command,'r')
time.sleep(3)
command="/usr/bin/func \"" + myhost.strip() + "*\" call power poweroff "
results=os.popen(command,'r')
return results
def go_shutdownall():
import MySQLdb
#import mysql
#find all hosts(minus myself)
#loop through results shutdown as we go.
db = MySQLdb.connect(host="localhost", user="mythtv", passwd="mythtv", db="mythconverg")
# create a cursor
cursor = db.cursor()
# execute SQL statement
myhostname = socket.gethostname()
results=["Sent shutdown command to: \n "]
cursor.execute("SELECT distinct(hostname) from settings where not hostname = %s ; ",(myhostname))
result = cursor.fetchall()
for row in result:
go_shutdown(row[0])
results.append(row[0])
#shutdown myself.
#go_shutdown(myhostname)
#results.append(myhostname)
return results
def go_reboot(myhost):
myhostname = socket.gethostname()
message="Reboot from MBE"
command="/usr/bin/func \"" + myhost.strip() + "*\" call msg display \"%s\" " %message
results=os.popen(command,'r')
time.sleep(3)
command="/usr/bin/func \"" + myhost.strip() + "*\" call power reboot "
results=os.popen(command,'r')
return results
def go_wake(myhost):
command="/usr/MythVantage/bin/wakeonlan.sh " + myhost.strip()
results=os.popen(command,'r')
return results
def go_kill(myhost,kill_type):
myhostname = socket.gethostname()
cmd=" call fe_restart "
cmd+=kill_type
command="/usr/bin/func \"" + myhost.strip() + "*\" " + cmd
results=os.popen(command,'r')
return results
mainpage = '''
Receiving a Form
%s'''
error = '''
Error
No Form Submission Was Received
'''
result = '''
%s of %s
'''
possible_parameters = ['param1', 'param2', 'param3', 'param4','hiddenparam','param5','param6']
if __name__ == '__main__':
cgiprint(contentheader) # content header
cgiprint() # finish headers with blank line
theform = cgi.FieldStorage()
#print theform
formdict = getform(theform, possible_parameters)
#print possible_parameters
if isblank(formdict):
body = error
else:
name = formdict['param1']
radio = formdict['param2'] # should be 'this' or 'that'
if radio != 'Restore' :
name = ""
update_type = formdict['param3']
oldurl = formdict['param4']
hidden = formdict['hiddenparam']
all_update_type=formdict['param5']
kill_type = formdict['param6']
body = result % (radio, hidden)
print mainpage % body
selection=radio
myhost=hidden
if selection == "Restore":
mylogfile=go_restore(name,myhost)
elif selection == "Backup":
mylogfile=go_backup(myhost)
elif selection == "Update":
mylogfile=go_update(myhost,update_type)
elif selection == "Shutdown":
mylogfile=go_shutdown(myhost)
elif selection == "Reboot":
mylogfile=go_reboot(myhost)
elif selection == "Optimize":
mylogfile=go_optimize(myhost)
elif selection == "WOL":
mylogfile=go_wake(myhost)
elif selection == "UpdateAll":
mylogfile=go_updateall(all_update_type)
elif selection == "ShutdownAll":
mylogfile=go_shutdownall()
elif selection == "Kill":
mylogfile=go_kill(myhost,kill_type)
box='''
'''
endbox='''
'''
javascript='''
'''
print box
for line in mylogfile:
print line + ' \r\n'
print endbox
print ' Back '
#print oldurl
print javascript