def run_as_handyrep(self, servername, commands):
# runs a set of commands as the "handyrep" user
# exiting when the first command fails
# returns a dic with the results of the last command
# run
lock_fabric()
env.key_filename = self.servers[servername]["ssh_key"]
env.user = self.servers[servername]["ssh_user"]
env.disable_known_hosts = True
env.host_string = self.servers[servername]["hostname"]
rundict = { "result": "SUCCESS",
"details" : "no commands provided",
"return_code" : None }
for command in commands:
try:
runit = run(command, warn_only=True, quiet=True)
rundict.update({ "details" : runit ,
"return_code" : runit.return_code })
if runit.succeeded:
rundict.update({"result":"SUCCESS"})
else:
rundict.update({"result":"FAIL"})
break
except Exception as ex:
rundict = { "result" : "FAIL",
"details" : "connection failure: %s" % exstr(ex),
"return_code" : None }
break
self.disconnect_and_unlock()
return rundict
def sudorun(self, servername, commands, runas, passwd="", sshpass=None):
# generic function to run one or more commands
# as a specific remote user. returns the results
# of the last command run. aborts when any
# command fails
lock_fabric()
if sshpass:
env.password = sshpass
else:
env.key_filename = self.servers[servername]["ssh_key"]
env.user = self.servers[servername]["ssh_user"]
env.disable_known_hosts = True
env.host_string = self.servers[servername]["hostname"]
rundict = return_dict(True, "no commands provided",
{"return_code": None})
if passwd is None:
pgpasswd = ""
else:
pgpasswd = passwd
for command in commands:
try:
with shell_env(PGPASSWORD=pgpasswd):
runit = sudo(command,
user=runas,
warn_only=True,
pty=False,
quiet=True)
rundict.update({
"details": runit,
"return_code": runit.return_code
})
if runit.succeeded:
rundict.update({"result": "SUCCESS"})
else:
rundict.update({"result": "FAIL"})
break
except Exception as ex:
rundict = {
"result": "FAIL",
"details": "connection failure: %s" % self.exstr(ex),
"return_code": None
}
break
self.disconnect_and_unlock()
return rundict
def push_template(self,
servername,
templatename,
destination,
template_params,
new_owner=None,
file_mode=700):
# renders a template file and pushes it to the
# target location on an external server
# not implemented for writing to localhost at this time
lock_fabric()
env.key_filename = self.servers[servername]["ssh_key"]
env.user = self.servers[servername]["ssh_user"]
env.disable_known_hosts = True
env.host_string = self.servers[servername]["hostname"]
try:
upload_template(
templatename,
destination,
use_jinja=True,
context=template_params,
template_dir=self.conf["handyrep"]["templates_dir"],
use_sudo=True)
if file_mode:
sudo("chmod %d %s" % (
file_mode,
destination,
), quiet=True)
if new_owner:
sudo("chown %s %s" % (
new_owner,
destination,
), quiet=True)
except:
self.log(
'PLUGIN', 'could not push template %s to server %s - %s' %
(templatename, servername, traceback.format_exc()), True)
retdict = return_dict(
False, "could not push template %s to server %s" % (
templatename,
servername,
))
else:
retdict = return_dict(True, "pushed template")
finally:
self.disconnect_and_unlock()
return retdict
def sudorun(self, servername, commands, runas, passwd="", sshpass=None):
# generic function to run one or more commands
# as a specific remote user. returns the results
# of the last command run. aborts when any
# command fails
lock_fabric()
if sshpass:
env.password = sshpass
else:
env.key_filename = self.servers[servername]["ssh_key"]
env.user = self.servers[servername]["ssh_user"]
env.disable_known_hosts = True
env.host_string = self.servers[servername]["hostname"]
rundict = return_dict(True, "no commands provided", {"return_code" : None })
if passwd is None:
pgpasswd = ""
else:
pgpasswd = passwd
for command in commands:
try:
with shell_env(PGPASSWORD=pgpasswd):
runit = sudo(command, user=runas, warn_only=True,pty=False, quiet=True)
rundict.update({ "details" : runit ,
"return_code" : runit.return_code })
if runit.succeeded:
rundict.update({"result":"SUCCESS"})
else:
rundict.update({"result":"FAIL"})
break
except Exception as ex:
rundict = { "result" : "FAIL",
"details" : "connection failure: %s" % self.exstr(ex),
"return_code" : None }
break
self.disconnect_and_unlock()
return rundict
def push_template(self, servername, templatename, destination, template_params, new_owner=None, file_mode=700):
# renders a template file and pushes it to the
# target location on an external server
# not implemented for writing to localhost at this time
lock_fabric()
env.key_filename = self.servers[servername]["ssh_key"]
env.user = self.servers[servername]["ssh_user"]
env.disable_known_hosts = True
env.host_string = self.servers[servername]["hostname"]
try:
upload_template( templatename, destination, use_jinja=True, context=template_params, template_dir=self.conf["handyrep"]["templates_dir"], use_sudo=True )
if file_mode:
sudo("chmod %d %s" % (file_mode, destination,), quiet=True)
if new_owner:
sudo("chown %s %s" % (new_owner, destination,), quiet=True)
except:
retdict = return_dict(False, "could not push template %s to server %s" % (templatename, servername,))
else:
retdict = return_dict(True, "pushed template")
finally:
self.disconnect_and_unlock()
return retdict
def run_as_handyrep(self, servername, commands):
# runs a set of commands as the "handyrep" user
# exiting when the first command fails
# returns a dic with the results of the last command
# run
lock_fabric()
env.key_filename = self.servers[servername]["ssh_key"]
env.user = self.servers[servername]["ssh_user"]
env.disable_known_hosts = True
env.host_string = self.servers[servername]["hostname"]
rundict = {
"result": "SUCCESS",
"details": "no commands provided",
"return_code": None
}
for command in commands:
try:
runit = run(command, warn_only=True, quiet=True)
rundict.update({
"details": runit,
"return_code": runit.return_code
})
if runit.succeeded:
rundict.update({"result": "SUCCESS"})
else:
rundict.update({"result": "FAIL"})
break
except Exception as ex:
rundict = {
"result": "FAIL",
"details": "connection failure: %s" % exstr(ex),
"return_code": None
}
break
self.disconnect_and_unlock()
return rundict
def disconnect_and_unlock(self):
disconnect_all()
lock_fabric(False)
return True
def disconnect_and_unlock(self):
disconnect_all()
lock_fabric(False)
return True