import math import os import signal import sys import threading import time import psutil from common.constants import bcolors from common.log import logUtils as log from constants import serverPackets from helpers import consoleHelper from objects import glob def dispose(): """ Perform some clean up. Called on shutdown. :return: """ print("> Disposing server... ") glob.fileBuffers.flushAll() consoleHelper.printColored("Goodbye!", bcolors.GREEN) def runningUnderUnix(): """ Get if the server is running under UNIX or NT :return: True if running under UNIX, otherwise False """ return True if os.name == "posix" else False def scheduleShutdown(sendRestartTime, restart, message = "", delay=20): """ Schedule a server shutdown/restart :param sendRestartTime: time (seconds) to wait before sending server restart packets to every client :param restart: if True, server will restart. if False, server will shudown :param message: if set, send that message to every client to warn about the shutdown/restart :param delay: additional restart delay in seconds. Default: 20 :return: """ # Console output log.info("Pep.py will {} in {} seconds!".format("restart" if restart else "shutdown", sendRestartTime+delay)) log.info("Sending server restart packets in {} seconds...".format(sendRestartTime)) # Send notification if set if message != "": glob.streams.broadcast("main", serverPackets.notification(message)) # Schedule server restart packet threading.Timer(sendRestartTime, glob.streams.broadcast, ["main", serverPackets.banchoRestart(delay*2*1000)]).start() glob.restarting = True # Restart/shutdown if restart: action = restartServer else: action = shutdownServer # Schedule actual server shutdown/restart some seconds after server restart packet, so everyone gets it threading.Timer(sendRestartTime+delay, action).start() def restartServer(): """ Restart pep.py :return: """ log.info("Restarting pep.py...") dispose() os.execv(sys.executable, [sys.executable] + sys.argv) def shutdownServer(): """ Shutdown pep.py :return: """ log.info("Shutting down pep.py...") dispose() sig = signal.SIGKILL if runningUnderUnix() else signal.CTRL_C_EVENT os.kill(os.getpid(), sig) def getSystemInfo(): """ Get a dictionary with some system/server info :return: ["unix", "connectedUsers", "webServer", "cpuUsage", "totalMemory", "usedMemory", "loadAverage"] """ data = {"unix": runningUnderUnix(), "connectedUsers": len(glob.tokens.tokens), "matches": len(glob.matches.matches)} # General stats delta = time.time()-glob.startTime days = math.floor(delta/86400) delta -= days*86400 hours = math.floor(delta/3600) delta -= hours*3600 minutes = math.floor(delta/60) delta -= minutes*60 seconds = math.floor(delta) data["uptime"] = "{}d {}h {}m {}s".format(days, hours, minutes, seconds) data["cpuUsage"] = psutil.cpu_percent() memory = psutil.virtual_memory() data["totalMemory"] = "{0:.2f}".format(memory.total/1074000000) data["usedMemory"] = "{0:.2f}".format(memory.active/1074000000) # Unix only stats if data["unix"]: data["loadAverage"] = os.getloadavg() else: data["loadAverage"] = (0,0,0) return data