slurmaccount.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	# © All rights reserved. ECOLE POLYTECHNIQUE FEDERALE DE LAUSANNE,
	# Switzerland
	# SCITAS - Scientific IT and Application Support, 2021
	# See the LICENSE.txt file for more details.

	import falcon
	from sausage.esoperations import ESOp
	from sausage.dboperations import DBOp
	from sausage.printer import PrintJson


	class SlurmAccount(object):
	def on_get_account(self, req, resp, account):
	response = {
	"name": account,
	"null": {
	"time": 0,
	"chf": 0,
	"co2": 0,
	}
	}

	try:
	newdb = DBOp()
	record = newdb.get_account(account)
	if record:
	pjson = PrintJson("account", record)
	response = pjson.response

	except BaseException:
	response = {
	"except": {
	"time": -1,
	"chf": -1,
	"co2": -1,
	}
	}

	resp.status = falcon.HTTP_200
	resp.media = response

	def on_get_user(self, req, resp, username, account='null'):
	response = {
	"name": username,
	account : {
	"null": {
	"time": 0,
	"chf": 0,
	"co2": 0,
	}
	}
	}
	try:
	newdb = DBOp()
	record = newdb.get_user(username, account)
	if record:
	if username == "all":
	pjson = PrintJson("all", record)
	else:
	pjson = PrintJson("user", record)
	response = pjson.response
	except BaseException:
	response = {
	"name": username,
	account : {
	"null": {
	"time": 0,
	"chf": 0,
	"co2": 0,
	}
	}
	}
	resp.status = falcon.HTTP_200
	resp.media = response