from twisted.web import server, resource
from twisted.internet import reactor
from twisted.internet.error import ReactorAlreadyRunning
from multiprocessing.managers import BaseManager
import time
import datetime
class GameSpyServerDatabase(BaseManager):
pass
GameSpyServerDatabase.register("get_server_list")
class StatsPage(resource.Resource):
isLeaf = True
def __init__(self, stats):
self.stats = stats
def render_GET(self, request):
# TODO: Make this easier to modify later
server_list = self.stats.get_server_list()
output = ""
output += ""
output += ""
output += "| Game ID | # Players | "
output += "
"
if server_list != None:
for game in server_list:
if not server_list[game]:
continue
output += ""
output += "| " + game + " | "
output += "%d | " % (len(server_list[game]))
output += "
"
output += "
"
output += "
"
output += "Last updated: %s
" % (self.stats.get_last_update_time())
output += ""
return output
class InternalStatsServer(object):
def __init__(self):
self.last_update = 0
self.next_update = 0
self.server_list = None
self.seconds_per_update = 60 # The number of seconds to wait before updating the server list
def start(self):
manager_address = ("127.0.0.1", 27500)
manager_password = ""
self.server_manager = GameSpyServerDatabase(address = manager_address, authkey= manager_password)
self.server_manager.connect()
site = server.Site(StatsPage(self))
reactor.listenTCP(9001, site)
try:
if reactor.running == False:
reactor.run(installSignalHandlers=0)
except ReactorAlreadyRunning:
pass
def get_server_list(self):
if self.next_update == 0 or self.next_update - time.time() <= 0:
self.last_update = time.time()
self.next_update = time.time() + self.seconds_per_update
self.server_list = self.server_manager.get_server_list()._getvalue()
print self.server_list
return self.server_list
def get_last_update_time(self):
return str(datetime.datetime.fromtimestamp(self.last_update))
if __name__ == "__main__":
stats = InternalStatsServer()
stats.start()