JudgeServer/server/utils.py
zema1 cbcc9cb313 Token获取方式从文件改为环境变量 (#5)
token获取方式改为环境变量
2017-05-30 13:14:49 +08:00

40 lines
981 B
Python

# coding=utf-8
from __future__ import unicode_literals
import _judger
import psutil
import socket
import logging
import hashlib
import os
from exception import JudgeClientError
logger = logging.getLogger(__name__)
handler = logging.FileHandler("/log/judge_server.log")
formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.setLevel(logging.WARNING)
def server_info():
ver = _judger.VERSION
return {"hostname": socket.gethostname(),
"cpu": psutil.cpu_percent(),
"cpu_core": psutil.cpu_count(),
"memory": psutil.virtual_memory().percent,
"judger_version": ".".join([str((ver >> 16) & 0xff), str((ver >> 8) & 0xff), str(ver & 0xff)])}
def get_token():
token = os.environ.get("token")
if token:
return token
else:
raise JudgeClientError("ENV token not found")
token = hashlib.sha256(get_token()).hexdigest()