my_host = socket.gethostname()
if "ernie" not in my_host:
sys.exit()
now = int(time.time())
SLURM_FILE = "/tmp/slurm.xml"
xml_file = open(SLURM_FILE, "w", encoding="iso-8859-1")
##################
# Get controllers
##################
primary, backup = pyslurm.get_controllers()
xml_file.write('<?xml version="1.0" encoding="iso-8859-1" ?>\n')
xml_file.write("<slurm>\n")
xml_file.write(f"\t<lastUpdate>{now}</lastUpdate>\n")
####################
# XML output of Jobs
####################
a = pyslurm.job()
jobs = a.get()
xml_file.write("\t<jobs>\n")
for key, value in jobs.items():
xml_file.write("\t\t<job>\n")
if __name__ == "__main__":
print()
print("PySLURM\t\t{0}".format(pyslurm.version()))
print("SLURM API\t{0}-{1}-{2}\n".format(*pyslurm.slurm_api_version()))
host = socket.gethostname()
print("Checking host.....{0}".format(host))
try:
a = pyslurm.is_controller(host)
print("\tHost is controller ({0})\n".format(a))
print("Querying SLURM controllers")
primary, backup = pyslurm.get_controllers()
print("\tPrimary - {0}".format(primary))
print("\tBackup - {0}".format(backup))
print("\nPinging SLURM controllers")
if primary:
print("\tPrimary .....", end=' ')
controller_up()
if backup:
print("\tBackup .....", end=' ')
controller_up(2)
except ValueError as e:
print("Error - {0}".format(e.args[0]))
def test_slurm_get_controllers():
"""Misc: Test slurm_get_controllers()."""
controllers = pyslurm.get_controllers()
assert_equals(controllers[0], gethostname())
if __name__ == "__main__":
print()
print(f"PySLURM\t\t{pyslurm.version()}")
major, minor, micro = pyslurm.slurm_api_version()
print(f"SLURM API\t{major}-{minor}-{micro}\n")
host = socket.gethostname()
print(f"Checking host.....{host}")
try:
a = pyslurm.is_controller(host)
print(f"\tHost is controller ({a})\n")
print("Querying SLURM controllers")
control_machs = pyslurm.get_controllers()
X = 0
for machine in control_machs:
if X == 0:
print(f"\tPrimary - {machine}")
print("\t\tPing .....", end=" ")
controller_up()
else:
print(f"\tBackup{X} - {machine}")
print("\t\tPing .....", end=" ")
controller_up(X)
X = X + 1
except ValueError as value_error:
print(f"Error - {value_error.args[0]}")
def GetSlurmServer(self):
return slurm.get_controllers()[0]
