def send_nrpecfg(sonda_pk, tasklog_pk=None):
sonda = Sonda.objects.get(pk=sonda_pk)
if SondaTask.objects.filter(name="send_nrpecfg").count() == 0:
task = SondaTask()
task.name = "send_nrpecfg"
task.description = "send the nrpe configuration to the sonda"
task.save()
if tasklog_pk is None:
tasklog = SondaTasksLog()
tasklog.sonda = sonda
tasklog.task = SondaTask.objects.get(name="send_nrpecfg")
tasklog.save()
else:
tasklog = SondaTasksLog.objects.get(pk=tasklog_pk)
taskstatus = SondaTaskStatus()
taskstatus.tasklog = tasklog
try:
data = {"NAGIOS_SERVER": sonda.servidor_nagios, "checks": []}
for nagiosnrpecheckopts in NagiosNrpeCheckOpts.objects.filter(sonda=sonda):
data["checks"].append(
"["
+ sonda.slug
+ "_"
+ nagiosnrpecheckopts.service.name
+ "]="
+ sonda.dir_checks
+ "/"
+ nagiosnrpecheckopts.get_full_check()
+ " -H $ARG1$"
)
nrpe_cfg_template = open("monitoring/sondas/templates/nrpe.cfg", "r")
template = Template(nrpe_cfg_template.read())
nrpe_cfg_template.close()
nrpecfg = open("tmp/nrpe_" + sonda.name + ".cfg", "w")
nrpecfg.write(template.render(Context(data)))
nrpecfg.close()
env.skip_bad_hosts = True
env.abort_on_prompts = True
env.user = "******"
env.host_string = str(sonda.unrackable_networked_device.main_ip.addr)
env.key_filename = settings.PROJECT_ROOT + "/keys/id_rsa"
if sonda.script_inicio is not None:
f = open("tmp/script_inicio_" + sonda.name, "w")
f.write(str(sonda.script_inicio.replace("\r", "")))
f.close()
put("tmp/script_inicio_" + sonda.name, "/tmp/script_inicio")
run("chmod 700 /tmp/script_inicio")
run("/tmp/script_inicio")
put("tmp/nrpe_" + sonda.name + ".cfg", "/etc/nagios/nrpe.cfg")
run("service " + sonda.nrpe_service_name + " restart")
if sonda.script_end is not None:
f = open("tmp/script_end_" + sonda.name, "w")
f.write(str(sonda.script_end.replace("\r", "")))
f.close()
put("tmp/script_end_" + sonda.name, "/tmp/script_end")
run("chmod 700 /tmp/script_end")
run("/tmp/script_end")
taskstatus.message = "nrpe.cfg send to " + sonda.name
taskstatus.status = 0
except Exception as e:
taskstatus.message = str(e)
taskstatus.status = 1
except:
fail = "Unknow exeption !\n sys exc info : \n"
for fails in sys.exc_info()[0:5]:
fail += str(fails) + "\n"
taskstatus.message = fail
taskstatus.status = 2
taskstatus.timestamp = datetime.datetime.now()
taskstatus.save()
def ssh_key_send_task(sonda_pk, user, passwd, tasklog_pk):
sonda = Sonda.objects.get(pk=sonda_pk)
if SondaTask.objects.filter(name="ssh_key").count() == 0:
task = SondaTask()
task.name = "ssh_key"
task.description = "send the ssh key to a sonda"
task.save()
if tasklog_pk is None:
tasklog = SondaTasksLog()
tasklog.sonda = sonda
tasklog.task = SondaTask.objects.get(name="ssh_key")
tasklog.save()
else:
tasklog = SondaTasksLog.objects.get(pk=tasklog_pk)
taskstatus = SondaTaskStatus()
taskstatus.tasklog = tasklog
try:
env.skip_bad_hosts = True
env.abort_on_prompts = True
env.user = user
env.password = passwd
env.host_string = str(sonda.unrackable_networked_device.main_ip.addr)
if sonda.script_inicio != "":
f = open("tmp/script_inicio_" + sonda.name, "w")
f.write(sonda.script_inicio.replace("\r", ""))
f.close()
put("tmp/script_inicio_" + sonda.name, "/tmp/script_inicio")
run("chmod 700 /tmp/script_inicio")
run("/tmp/script_inicio")
if not exists("/root/.ssh/"):
run("mkdir /root/.ssh")
if not exists("/root/.ssh/authorized/"):
run("mkdir /root/.ssh/authorized")
put(settings.PROJECT_ROOT + "/keys/id_rsa.pub", "/root/.ssh/authorized/id_rsa.pub")
run("cat /root/.ssh/authorized/id_rsa.pub >> /root/.ssh/authorized_keys")
if sonda.script_end is not None:
f = open("tmp/script_end_" + sonda.name, "w")
f.write(str(sonda.script_end.replace("\r", "")))
f.close()
put("tmp/script_end_" + sonda.name, "/tmp/script_end")
run("chmod 700 /tmp/script_end")
run("/tmp/script_end")
sonda.ssh = True
sonda.save()
print("Config done with " + sonda.name)
taskstatus.message = "Config done with " + sonda.name
taskstatus.status = 0
env.password = ""
except Exception as e:
taskstatus.message = str(e)
taskstatus.status = 1
except:
fail = "Unknow exeption !\n sys exc info : \n"
for fails in sys.exc_info()[0:5]:
fail += str(fails) + "\n"
taskstatus.message = fail
taskstatus.status = 2
taskstatus.timestamp = datetime.datetime.now()
taskstatus.save()