class Server(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80))
type = db.Column(db.String(40))
ip = db.Column(db.String(40))
port = db.Column(db.Integer)
game_type = db.Column(db.String(40))
rcon = db.Column(db.String(80))
location = db.Column(db.String(256))
_server = None
is_restarting = db.Column(db.Boolean)
def __init__(self, name, ip, port, rcon, location, game_type, local=True):
'''Creates a new server object.
Params:
name - Friendly name of server.
type - Type of server, TF2 as an example.
ip - Ip of the server.
port - Port this server is running on.
rcon - RCON password
local - Boolean; Is this a local server?
location - Location on disk, absolute path.
game_type - TF2, DOD:S, etc.
'''
self.name = name
self.ip = ip
try:
self.port = int(port)
except ValueError:
self.port = 0
self.location = location
self.rcon = rcon
self.game_type = game_type
self._get_source_binding()
self._server = None
def get_players(self):
'''Return the number of players on the server.'''
try:
if not self._server:
self._get_source_binding()
return self._server.info()['numplayers']
except socket.error:
return "??"
def get_current_map(self):
'''Return the current map.'''
try:
if not self._server:
self._get_source_binding()
return self._server.info()['map']
except socket.error:
return ""
def get_actual_name(self):
'''Return the name the server is reporting.'''
try:
if not self._server:
self._get_source_binding()
return self._server.info()['hostname']
except socket.error:
return ""
def _get_source_binding(self):
self._server = SourceQuery(self.ip, self.port)
def get_friendly_time(self, time):
try:
float_time = float(time)
if float_time < 60:
return "%.0f seconds" % float_time
float_time = float_time / 60.0
if float_time < 60:
return "%.1f minutes" % float_time
float_time = float_time / 60.0
if float_time < 60:
return "%.1f hours" % float_time
except ValueError:
pass
return "N/A"
def get_all_players(self):
try:
if not self._server:
self._get_source_binding()
return self._server.player()
except socket.error:
return []
def get_server_value(key):
'''Returns a tuple of the key given, and the value it is set to on this server.'''
matcher = re.compile('"([^"]+)" = "([^"]+)"')
matches = matcher.match(self._send_rcon(key).split('\n')[0])
if matches is not None:
return matcher.match(self._send_rcon(key).split('\n')[0]).groups()
return (key, None)
def get_mode(self):
if not self.location:
return "Error - Unknown Server Location"
if not os.path.exists(location):
return "Error - Server Location (%s) was not found." % location
def restart(self, message=None):
# TODO: When we fix the 10 second wait, we need to fix this message too.
if not message:
message = "Server is restarting in... NOW!"
self._send_rcon("sm_csay '%s'" % message)
self.is_restarting = True
db.session.commit()
self._send_rcon('quit')
# TODO: Still need to fix this.
#Timer(10, self._send_rcon, ['quit']).start()
return True
def _send_rcon(self, cmd, user=None):
for restricted in RESTRICTED_CMDS:
if restricted in cmd.lower():
print "Restricted command issued"
return ""
server = SourceRcon(self.ip, self.port, self.rcon)
try:
return server.rcon(cmd)
except:
# This just throws a generic exception.
return ""
@staticmethod
def get(id):
server = Server.query.filter_by(id=id).first()
if not server:
return False
server._get_source_binding()
# Verify that this server didn't come back online
if server.is_restarting and not server.get_players() == "??":
server.is_restarting = False
db.session.commit()
return server
@staticmethod
def create(name, address, port, rcon, servertype, path):
server = Server.query.filter_by(name=name).first()
if server:
return (False, "Error: A server named '%s' already exists." % name)
new_server = Server(name, address, port, rcon, path, servertype)
db.session.add(new_server)
return (True, "Success! A server named '%s' was created!" % name)
@staticmethod
def get_all():
allservers = Server.query.all()
for server in allservers:
if server.is_restarting and not server.get_players() == "??":
server.is_restarting = False
db.session.commit()
return allservers
@staticmethod
def delete(id):
server = Server.query.filter_by(id=id).first()
if not server:
return False
db.session.delete(server)
db.session.commit()
return True
def _get_source_binding(self):
self._server = SourceQuery(self.ip, self.port)