def __init__(self, site):
cmd.Cmd.__init__(self)
self.suffix = "=>> "
self.intro = "Welcome to gsh! "
self.prompt = site + " " + self.suffix
self.keywords = ["!", "hist", "exit", "help", "setsite", "getsite", "EOF", "vdt_location", "version"]
self.home = os.environ["HOME"]
self.history = "%s/.gsh_history" % self.home
self.workfile = "%s/gsh_workfile" % self.home
call(["/bin/touch", self.workfile])
call(["/bin/touch", self.history])
# site specific variables
self.site_env = {}
self.site = site
self.site_name = getSiteNameFromFQDN(self.site)
# register command handler
self.commandHandler = CommandHandler(self)
pwd = self.commandHandler.get_pwd(empty_cwd=True)
self.cwd = pwd.split("/")
self.old_cwd = None
def __init__(self, site, gsh_location):
cmd.Cmd.__init__(self)
self.suffix = "=>> "
self.intro = "Welcome to gsh! "
self.prompt = site + " " + self.suffix
self.keywords = ["!", "hist", "exit", "help", "setsite", "getsite",
"EOF", "vdt_location", "version"]
self.home = os.environ['HOME']
self.history = "%s/.gsh_history" % self.home
self.workfile = "%s/gsh_workfile" % self.home
call(["/bin/touch", self.workfile])
call(["/bin/touch", self.history])
# site specific variables
self.site_env = {}
self.site_env["PWD"] = ""
self.site = site
self.site_name = getSiteNameFromFQDN(self.site)
# register command handler
self.commandHandler = CommandHandler(self)
class Console(cmd.Cmd):
def __init__(self, site, gsh_location):
cmd.Cmd.__init__(self)
self.suffix = "=>> "
self.intro = "Welcome to gsh! "
self.prompt = site + " " + self.suffix
self.keywords = ["!", "hist", "exit", "help", "setsite", "getsite",
"EOF", "vdt_location", "version"]
self.home = os.environ['HOME']
self.history = "%s/.gsh_history" % self.home
self.workfile = "%s/gsh_workfile" % self.home
call(["/bin/touch", self.workfile])
call(["/bin/touch", self.history])
# site specific variables
self.site_env = {}
self.site_env["PWD"] = ""
self.site = site
self.site_name = getSiteNameFromFQDN(self.site)
# register command handler
self.commandHandler = CommandHandler(self)
## Command definitions to support Cmd object functionality ##
def do_EOF(self, args):
"""Exit on system end of file character"""
readline.write_history_file(self.history)
call(["/bin/rm", "-f", self.workfile])
return self.do_exit(args)
def do_shell(self, args):
"""Pass command to a system shell when line begins with '!'"""
if isinstance(args, types.StringType):
args = shlex.split(args)
call(args)
def do_help(self, args):
"""Get help on commands
'help' or '?' with no arguments prints a list of commands for which help is available
'help <command>' or '? <command>' gives help on <command>
"""
## The only reason to define this method is for the help text in the doc string
cmd.Cmd.do_help(self, args)
## Override methods in Cmd object ##
def preloop(self):
"""Initialization before prompting user for commands.
Despite the claims in the Cmd documentaion, Cmd.preloop() is not a stub.
"""
cmd.Cmd.preloop(self) ## sets up command completion
# need to read history and set self._hist
f = open(self.history)
hist = f.readlines()
f.close()
self._hist = [a[:-1] for a in hist]
readline.read_history_file(self.history)
self._locals = {} ## Initialize execution namespace for user
self._globals = {}
def postloop(self):
"""Take care of any unfinished business.
Despite the claims in the Cmd documentaion, Cmd.postloop() is not a stub.
"""
cmd.Cmd.postloop(self) ## Clean up command completion
print "Exiting..."
def precmd(self, line):
""" This method is called after the line has been input but before
it has been interpreted. If you want to modify the input line
before execution (for example, variable substitution) do it here.
"""
self._hist += [ line.strip() ]
if len(line) > 1: # added so that gsh doesn't bomb when user presses enter without typing anything else
if not (line.split()[0] in self.keywords):
if len(self.site) < 1:
print "You did not specify a site. Please execute setsite <fqdn> with a valid fqdn."
line = ""
else:
line = self.commandHandler.customCommand(line, self.workfile)
return line
def postcmd(self, stop, line):
"""If you want to stop the console, return something that evaluates to true.
If you want to do some post command processing, do it here.
"""
return stop
def emptyline(self):
"""Do nothing on empty input line"""
pass
def default(self, line):
"""Called on an input line when the command prefix is not recognized.
In that case we execute the line as Python code.
"""
try:
exec(line) in self._locals, self._globals
except Exception, e:
print e.__class__, ":", e
