slurmaccount.py
79a2ce481ff7master

slurmaccount.py
View Options

			# © 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:
			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