forked from hipstercat/giantsd
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from dpnet.netserver import Netserver
|
|
from giants.masterserver import MasterServer
|
|
import asyncio
|
|
|
|
## stats
|
|
import pymysql
|
|
|
|
|
|
class Server:
|
|
def __init__(self, **kwargs):
|
|
self.listen_ip = kwargs.get("ip", "0.0.0.0")
|
|
self.listen_port = kwargs.get("port", 19711)
|
|
self.register_with_ms = kwargs.get("register", True)
|
|
## stats
|
|
self.db = self.db_connect()
|
|
|
|
@classmethod
|
|
def db_connect(cls):
|
|
return pymysql.connect(host='192.168.40.2', user="gck", passwd="gck", db="gck", autocommit=True)
|
|
|
|
def db_execute(self, sql, args=None):
|
|
try:
|
|
with self.db.cursor() as cursor:
|
|
cursor.execute(sql, args)
|
|
except:
|
|
self.db = self.db_connect()
|
|
return self.db_execute(sql, args)
|
|
|
|
def db_query(self, sql, args=None):
|
|
try:
|
|
with self.db.cursor() as cursor:
|
|
cursor.execute(sql, args)
|
|
t = cursor.fetchall()
|
|
return t
|
|
except:
|
|
self.db = self.db_connect()
|
|
return self.db_query(sql, args)
|
|
|
|
async def on_connection_received(self, ip, port):
|
|
version = 1.5
|
|
map_name = "Players last 24h/last 48h"
|
|
server_name = "\x03!! V1.5 IS OUT !! \x04 Download on Discord: \x06https://discord.gg/Avj4azU\x01"
|
|
|
|
self.db_execute("INSERT INTO multiplayer_stats(ip, date) VALUES (%s, NOW())", (ip,))
|
|
stats_24 = self.db_query("SELECT COUNT(DISTINCT ip) FROM multiplayer_stats WHERE date + INTERVAL 1 DAY > NOW() AND ip not in ('192.168.40.1')")[0][0]
|
|
stats_48 = self.db_query("SELECT COUNT(DISTINCT ip) FROM multiplayer_stats WHERE date + INTERVAL 2 DAY > NOW() AND ip not in ('192.168.40.1')")[0][0]
|
|
|
|
player_count = stats_24
|
|
max_player_count = stats_48
|
|
|
|
return version, server_name, player_count, max_player_count, map_name
|
|
|
|
|
|
if __name__ == '__main__':
|
|
server = Server(port=19110)
|
|
if server.register_with_ms:
|
|
ms = MasterServer(server)
|
|
register_thread = ms.register_and_run()
|
|
|
|
loop = asyncio.get_event_loop()
|
|
listen = loop.create_datagram_endpoint(lambda: Netserver(server), local_addr=(server.listen_ip, server.listen_port))
|
|
transport, protocol = loop.run_until_complete(listen)
|
|
try:
|
|
loop.run_forever()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
transport.close()
|
|
loop.close()
|