49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import getpass, time, requests
|
|
|
|
def fetch(site, node, start, stop, user, psw):
|
|
start_str = time.strftime("%Y-%m-%dT%H:%M", time.localtime(int(start)))
|
|
stop_str = time.strftime("%Y-%m-%dT%H:%M", time.localtime(int(stop+61)))
|
|
auth = requests.auth.HTTPBasicAuth(user, psw)
|
|
resp = requests.get(f"https://api.grid5000.fr/stable/sites/{site}/metrics?nodes={node}&metrics=bmc_node_power_watt&start_time={start_str}&end_time={stop_str}", auth=auth)
|
|
if resp.status_code != 200:
|
|
print("Status: ", resp.status_code)
|
|
print(resp.content)
|
|
exit(1)
|
|
data = resp.json()
|
|
|
|
energy = 0
|
|
last_power = None
|
|
last_time = 0
|
|
for sample in data:
|
|
t = time.mktime(time.strptime(sample["timestamp"], "%Y-%m-%dT%H:%M:%S.%f%z"))
|
|
if last_power != None:
|
|
energy += last_power * (t - last_time)
|
|
last_power = None
|
|
if t >= start and t <= stop:
|
|
last_power = sample["value"]
|
|
last_time = t
|
|
return energy
|
|
|
|
def get_psw():
|
|
psw = None
|
|
try:
|
|
f = open("/dev/shm/g5k_psw", "r")
|
|
psw = f.read()
|
|
f.close()
|
|
except:
|
|
pass
|
|
if psw:
|
|
return psw
|
|
psw = getpass.getpass("Psw: ")
|
|
f = open("/dev/shm/g5k_psw", "w")
|
|
f.write(psw)
|
|
f.close()
|
|
return psw
|
|
|
|
def main():
|
|
user = "pengelib"
|
|
psw = get_psw()
|
|
print(fetch("nancy", "gros-69", 1772205368.593937, 1772206568.6941307, user, psw))
|
|
|
|
if __name__ == "__main__":
|
|
main()
|