def machine(value): ##define host and group info.
HInfo = host_info.hostInfo()
machineInfo = []
if value == 'host' or value == 'view':
host = HInfo.hostInfo()
elif value == 'group':
host = HInfo.hostInfo("select distinct groupname from groups")
print "\n----------------------------%s list----------------------------\n" % value
for i in host:
machineInfo.append(i[0])
if value == 'group':
gInfo = host_info.hostInfo()
sql = "select hostname from host where hostname = any(select hostname from groups where groupname = '%s')" % i[0]
g = gInfo.hostInfo(sql)
print '\n\tGroup\t\t\tHost'
print "\t", i[0]
for i in g:
print '\t\t\t\t',i[0]
else:
print "\t", i[0]
print "\n"
if value == 'view':
print "\n"
return 0
ChooseInfo = "Please input you want to choose or input 'quit' will exit: "
return input(ChooseInfo,machineInfo)
def nameInfo(sql): ##check group or hostname info
HInfo = host_info.hostInfo()
groups = HInfo.hostInfo(sql)
groupInfo = []
for i in groups:
groupInfo.append(i[0])
return groupInfo
def BeSure(name,choose): ## choose hosts or groups execute commands.
HInfo = host_info.hostInfo()
run = conn_host.execute()
subTitle = ['Execute commands', 'Copy Files','Exit']
subMenu = menu.title(subTitle)
null = 0
if choose == 1:
host = HInfo.hostInfo()
sql = "select * from host where hostname = '%s'" % name
elif choose == 2:
host = HInfo.hostInfo("select distinct groupname from groups")
sql = "select * from host where hostname = any(select hostname from groups where groupname = '%s')" % name
for i in host:
if i[0] == name:
null = 1
if subMenu == '1':
cmd = raw_input("Please input you need to run command or input 'quit' will exit: ")
if cmd == 'quit':return 0
run.run(cmd,sql)
elif subMenu == '2':
filename = raw_input("Please input you need to copy of the file name or input 'quit' will exit: ")
if filename == 'quit':return 0
pathname = raw_input("Please input you need to save path or input 'quit' will exit: ")
if pathname == 'quit':return 0
run.scp(filename,sql,pathname)
elif subMenu == '3':break
if null == 0:
print "The %s is absent." % name
def scp(self,filename,sql,pathname='/root'): print time.ctime() HInfo = host_info.hostInfo() host = HInfo.hostInfo(sql) for i in host: t = threading.Thread(target=self.__connectScp,args=(i[0],i[1],i[2],i[3],filename,pathname,)) t.start() print "current has %d threads" % (threading.activeCount() - 1)
def run(self,cmd,sql): print time.ctime() HInfo = host_info.hostInfo() host = HInfo.hostInfo(sql) for i in host: #print i t = threading.Thread(target=self.__connectHost,args=(cmd,i[0],i[1],i[2],i[3],)) t.start() print "current has %d threads" % (threading.activeCount() - 1)
def addInfo(): ##add hosts or groups info.
HInfo = host_info.hostInfo()
addSQLCommands = HInfo.hostInfo()
title_name = ['Add host', 'Add group','Del host','Del group','Host added to group','Exit']
v = menu.title(title_name)
if v == '1':
addHostHostname = raw_input("Please input you want to add Host's hostname: ").strip()
addHostIp = raw_input("Please input you want to add Host's ip address: ").strip()
addHostUsername = raw_input("Please input you want to add Host's username: "******"Please input you want to add Host's password: "******"insert into host values('%s','%s','%s','%s')" %(addHostHostname,addHostIp,addHostUsername,addHostPassword)
addSQLCommands = HInfo.hostInfo(sql)
if addSQLCommands == 0:
print "Add '%s' host is failed." % addHostHostname
else:
print "Add '%s' host is successful." % addHostHostname
elif v == '2':
while True:
addGroupname = raw_input("Please input you want to add Group's groupname or input 'quit' will exit: ").strip()
sql = "select distinct groupname from groups"
if addGroupname in nameInfo(sql):
print "The '%s' group is exist,add failed and please input groupname again." % addGroupname
continue
elif len(addGroupname) == 0:continue
elif addGroupname == 'quit':break
sql = "insert into groups (groupname) values('%s')" % addGroupname
addSQLCommands = HInfo.hostInfo(sql)
if addSQLCommands == 0:
print "Add '%s' group is failed." % addGroupname
else:
print "Add '%s' group is successful." % addGroupname
elif v == '3':
while True:
delHost = "Please input you need to delete hostname or input 'quit' will exit: "
sql = "select hostname from host"
result = input(delHost,nameInfo(sql))
if result == 'quit':return 0
sql = "delete from host where hostname='%s'" % result
addSQLCommands = HInfo.hostInfo(sql)
if addSQLCommands == 0:
print "Delete '%s' host is failed." % result
else:
print "Delete '%s' host is successful." % result
elif v == '4':
while True:
delGroup = "Please input you need to delete groupname or input 'quit' will exit: "
sql = "select groupname from groups"
result = input(delGroup,nameInfo(sql))
if result == 'quit':return 0
sql = "delete from groups where groupname='%s'" % result
addSQLCommands = HInfo.hostInfo(sql)
if addSQLCommands == 0:
print "Delete '%s' host is failed." % result
else:
print "Delete '%s' host is successful." % result
elif v == '5':
sql = "select hostname from host"
h = nameInfo(sql)
print '-----------------------------------------------------------------The host list----------------------------------------------------------------------------'
print '|\t%s\t|' %('\t|\t'.join(h))
print '----------------------------------------------------------------------------------------------------------------------------------------------------------\n'
hostName = "Please input you need to added hostname to group: "
hostNameValue = input(hostName,h)
if hostNameValue == 'quit':return 0
sql = "select distinct groupname from groups"
g = nameInfo(sql)
print '-----------------------------------------------------------------The group list----------------------------------------------------------------------------'
print '|\t%s\t|' %('\t|\t'.join(g))
print '-----------------------------------------------------------------------------------------------------------------------------------------------------------\n'
groupName = "Please input your groupname: "
groupNameValue = input(groupName,g)
if groupNameValue == 'quit':return 0
checkSql = "select hostname from groups where hostname='%s' and groupname='%s'" % (hostNameValue,groupNameValue)
checkName = nameInfo(checkSql)
if hostNameValue in checkName:
print "The '%s' host already in '%s' group,No added." % (hostNameValue,groupNameValue)
return 0
sql = "insert into groups values('%s','%s')" % (groupNameValue,hostNameValue)
addSQLCommands = HInfo.hostInfo(sql)
if addSQLCommands == 0:
print "The '%s' host added to %s group is failed." % (hostNameValue,groupNameValue)
else:
print "The '%s' host added to %s group is successful." % (hostNameValue,groupNameValue)
elif v == '6':return 0