def runMenu(): ##define the main program
title_name = ['Choose the host', 'Choose the group','View all hosts','Added(Delete) info(Host or group)','Exit']
m = menu.title(title_name)
while True:
if m == '1':
h = machine('host')
if h == 'quit':break
BeSure(h,1)
break
elif m == '2':
h = machine('group')
if h == 'quit':break
BeSure(h,2)
break
elif m == '3':
machine('view')
break
elif m == '4':
addInfo()
break
elif m == '5':
break
else:
m = menu.title(title_name)
def check_db():
path_db = os.path.join(os.path.dirname(__file__), 'database/calendar.db')
if os.path.exists(path_db) == False:
db_schema.schema_db(path_db)
db_schema.insert_default(path_db)
elif os.path.exists(path_db) == True:
db_schema.schema_db(path_db)
menu.title()
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 insertMenu(self, menu, before="&Help"):
"""Insert a menu in the menubar before the specified menu.
The new menu can be inserted BEFORE any of the existing menus.
By default the new menu will be inserted before the Help menu.
Also, the menu's title should be unique.
"""
if not self.menus.has_key(str(menu.title())):
self.menu.insertMenu(self.menus[before], menu)
self.menus = dict([[str(a.text()), a] for a in self.menu.actions()])
from os.path import join
from constants import *
from globalvars import *
from world import *
import sprites
import pc
from monsters import *
import menu
import pygame
# initialization
pygame.mixer.init()
pygame.display.set_caption("Mechanical Failure")
pygame.display.set_icon(globalvars.images["misc" + "logo"])
menu.title()
while True:
if globalvars.newgame:
# load save game
globalvars.hero = pc.PC("Cole", CENTERCENTER)
globalvars.heroGroup.empty()
globalvars.heroGroup.add(globalvars.hero)
globalvars.solidGroup.empty()
globalvars.solidGroup.add(globalvars.hero)
globalvars.attackGroup.empty()
globalvars.itemGroup.empty()
path = join(getcwd(), "savedata")
savelist = listdir(path)
if len(savelist) > 1:
if menu.dialogue("Do you want to load the saved game?"):
import convert as con
import menu
import rates
menu.title()
while True:
menu.menu()
ch = input("Enter a Choice: ")
if ch == '1':
con.usd()
p = input("Do you want to return to Menu(M) or Quit(Q)?")
while True:
if p == 'M' or p == 'm': continue
if p == 'Q' or p == 'q': break
break
if ch == '2':
con.eur()
p = input("Do you want to return to Menu(M) or Quit(Q)?")
while True:
if p == 'M' or p == 'm': continue
if p == 'Q' or p == 'q': break
break
if ch == '3':
con.gbp()
p = input("Do you want to return to Menu(M) or Quit(Q)?")
while True:
if p == 'M' or p == 'm': continue
if p == 'Q' or p == 'q': break
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
