mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
353 lines
12 KiB
Python
353 lines
12 KiB
Python
import os
|
|
import logging
|
|
import datetime
|
|
import typing as t
|
|
|
|
from playhouse.shortcuts import model_to_dict
|
|
|
|
import typing as t
|
|
|
|
from peewee import DoesNotExist
|
|
from playhouse.shortcuts import model_to_dict
|
|
|
|
|
|
from app.classes.models.servers import Servers, HelperServers
|
|
from app.classes.shared.helpers import Helpers
|
|
from app.classes.shared.migration import MigrationManager
|
|
|
|
try:
|
|
from peewee import (
|
|
SqliteDatabase,
|
|
Model,
|
|
ForeignKeyField,
|
|
CharField,
|
|
AutoField,
|
|
DateTimeField,
|
|
BooleanField,
|
|
IntegerField,
|
|
FloatField,
|
|
DoesNotExist,
|
|
)
|
|
|
|
except ModuleNotFoundError as e:
|
|
Helpers.auto_installer_fix(e)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
peewee_logger = logging.getLogger("peewee")
|
|
peewee_logger.setLevel(logging.INFO)
|
|
|
|
# **********************************************************************************
|
|
# Servers Stats Class
|
|
# **********************************************************************************
|
|
class ServerStats(Model):
|
|
stats_id = AutoField()
|
|
created = DateTimeField(default=datetime.datetime.now)
|
|
server_id = ForeignKeyField(Servers, backref="server", index=True)
|
|
started = CharField(default="")
|
|
running = BooleanField(default=False)
|
|
cpu = FloatField(default=0)
|
|
mem = FloatField(default=0)
|
|
mem_percent = FloatField(default=0)
|
|
world_name = CharField(default="")
|
|
world_size = CharField(default="")
|
|
server_port = IntegerField(default=25565)
|
|
int_ping_results = CharField(default="")
|
|
online = IntegerField(default=0)
|
|
max = IntegerField(default=0)
|
|
players = CharField(default="")
|
|
desc = CharField(default="Unable to Connect")
|
|
version = CharField(default="")
|
|
updating = BooleanField(default=False)
|
|
waiting_start = BooleanField(default=False)
|
|
first_run = BooleanField(default=True)
|
|
crashed = BooleanField(default=False)
|
|
downloading = BooleanField(default=False)
|
|
|
|
class Meta:
|
|
table_name = "server_stats"
|
|
|
|
|
|
# **********************************************************************************
|
|
# Servers_Stats Methods
|
|
# **********************************************************************************
|
|
class HelperServerStats:
|
|
server_id: int
|
|
database = None
|
|
|
|
def __init__(self, server_id):
|
|
self.server_id = int(server_id)
|
|
self.init_database(self.server_id)
|
|
|
|
def init_database(self, server_id):
|
|
try:
|
|
server = HelperServers.get_server_data_by_id(server_id)
|
|
db_folder = os.path.join(f"{server['path']}", "db_stats")
|
|
db_file = os.path.join(
|
|
db_folder,
|
|
"crafty_server_stats.sqlite",
|
|
)
|
|
self.database = SqliteDatabase(
|
|
db_file, pragmas={"journal_mode": "wal", "cache_size": -1024 * 10}
|
|
)
|
|
if not os.path.exists(db_file):
|
|
try:
|
|
os.mkdir(db_folder)
|
|
except Exception as ex:
|
|
logger.warning(
|
|
f"Error try to create the db_stats folder for server : {ex}"
|
|
)
|
|
helper_stats = Helpers()
|
|
helper_stats.migration_dir = os.path.join(
|
|
f"{helper_stats.migration_dir}", "stats"
|
|
)
|
|
helper_stats.db_path = db_file
|
|
migration_manager = MigrationManager(self.database, helper_stats)
|
|
migration_manager.up() # Automatically runs migrations
|
|
except Exception as ex:
|
|
logger.warning(
|
|
f"Error try to look for the db_stats files for server : {ex}"
|
|
)
|
|
return None
|
|
|
|
def select_database(self):
|
|
try:
|
|
server = HelperServers.get_server_data_by_id(self.server_id)
|
|
db_file = os.path.join(
|
|
f"{server['path']}",
|
|
"db_stats",
|
|
"crafty_server_stats.sqlite",
|
|
)
|
|
self.database = SqliteDatabase(
|
|
db_file, pragmas={"journal_mode": "wal", "cache_size": -1024 * 10}
|
|
)
|
|
except Exception as ex:
|
|
logger.warning(
|
|
f"Error try to look for the db_stats files for server : {ex}"
|
|
)
|
|
return None
|
|
|
|
def get_all_servers_stats(self):
|
|
servers = HelperServers.get_all_defined_servers()
|
|
server_data = []
|
|
try:
|
|
for server in servers:
|
|
latest = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == server.get("server_id"))
|
|
.order_by(ServerStats.created.desc())
|
|
.limit(1)
|
|
)
|
|
latest._database = self.database
|
|
server_data.append(
|
|
{
|
|
"server_data": server,
|
|
"stats": stats,
|
|
"user_command_permission": True,
|
|
}
|
|
)
|
|
except IndexError as ex:
|
|
logger.error(
|
|
f"Stats collection failed with error: {ex}. Was a server just created?"
|
|
)
|
|
return server_data
|
|
|
|
def insert_server_stats(self, server):
|
|
server_id = server.get("id", 0)
|
|
|
|
if server_id == 0:
|
|
logger.warning("Stats saving failed with error: Server unknown (id = 0)")
|
|
return
|
|
|
|
ServerStats.insert(
|
|
{
|
|
ServerStats.server_id: server.get("id", 0),
|
|
ServerStats.started: server.get("started", ""),
|
|
ServerStats.running: server.get("running", False),
|
|
ServerStats.cpu: server.get("cpu", 0),
|
|
ServerStats.mem: server.get("mem", 0),
|
|
ServerStats.mem_percent: server.get("mem_percent", 0),
|
|
ServerStats.world_name: server.get("world_name", ""),
|
|
ServerStats.world_size: server.get("world_size", ""),
|
|
ServerStats.server_port: server.get("server_port", 0),
|
|
ServerStats.int_ping_results: server.get("int_ping_results", False),
|
|
ServerStats.online: server.get("online", False),
|
|
ServerStats.max: server.get("max", False),
|
|
ServerStats.players: server.get("players", False),
|
|
ServerStats.desc: server.get("desc", False),
|
|
ServerStats.version: server.get("version", False),
|
|
}
|
|
).execute(self.database)
|
|
|
|
def remove_old_stats(self, last_week):
|
|
# self.select_database(self.server_id)
|
|
ServerStats.delete().where(ServerStats.created < last_week).execute(
|
|
self.database
|
|
)
|
|
|
|
def get_latest_server_stats(self):
|
|
latest = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.order_by(ServerStats.created.desc())
|
|
.limit(1)
|
|
.get(self.database)
|
|
)
|
|
try:
|
|
return model_to_dict(latest)
|
|
except IndexError:
|
|
return {}
|
|
|
|
def get_server_stats(self):
|
|
stats = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.order_by(ServerStats.created.desc())
|
|
.limit(1)
|
|
.first(self.database)
|
|
)
|
|
return model_to_dict(stats)
|
|
|
|
def server_id_exists(self):
|
|
# self.select_database(self.server_id)
|
|
if not HelperServers.get_server_data_by_id(self.server_id):
|
|
return False
|
|
return True
|
|
|
|
def sever_crashed(self):
|
|
# self.select_database(self.server_id)
|
|
ServerStats.update(crashed=True).where(
|
|
ServerStats.server_id == self.server_id
|
|
).execute(self.database)
|
|
|
|
def set_download(self):
|
|
# self.select_database(self.server_id)
|
|
ServerStats.update(downloading=True).where(
|
|
ServerStats.server_id == self.server_id
|
|
).execute(self.database)
|
|
|
|
def finish_download(self):
|
|
# self.select_database(self.server_id)
|
|
ServerStats.update(downloading=False).where(
|
|
ServerStats.server_id == self.server_id
|
|
).execute(self.database)
|
|
|
|
def get_download_status(self):
|
|
# self.select_database(self.server_id)
|
|
download_status = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.get(self.database)
|
|
)
|
|
return download_status.downloading
|
|
|
|
def server_crash_reset(self):
|
|
if self.server_id is None:
|
|
return
|
|
|
|
# self.select_database(self.server_id)
|
|
ServerStats.update(crashed=False).where(
|
|
ServerStats.server_id == self.server_id
|
|
).execute(self.database)
|
|
|
|
def is_crashed(self):
|
|
# self.select_database(self.server_id)
|
|
svr: ServerStats = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.get(self.database)
|
|
)
|
|
return svr.crashed
|
|
|
|
def set_update(self, value):
|
|
if self.server_id is None:
|
|
return
|
|
|
|
# self.select_database(self.server_id)
|
|
try:
|
|
# Checks if server even exists
|
|
ServerStats.select().where(ServerStats.server_id == self.server_id).execute(
|
|
self.database
|
|
)
|
|
except DoesNotExist as ex:
|
|
logger.error(f"Database entry not found! {ex}")
|
|
return
|
|
ServerStats.update(updating=value).where(
|
|
ServerStats.server_id == self.server_id
|
|
).execute(self.database)
|
|
|
|
def get_update_status(self):
|
|
# self.select_database(self.server_id)
|
|
update_status = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.get(self.database)
|
|
)
|
|
return update_status.updating
|
|
|
|
def set_first_run(self):
|
|
# self.select_database(self.server_id)
|
|
# Sets first run to false
|
|
try:
|
|
# Checks if server even exists
|
|
ServerStats.select().where(ServerStats.server_id == self.server_id).execute(
|
|
self.database
|
|
)
|
|
except Exception as ex:
|
|
logger.error(f"Database entry not found! {ex}")
|
|
return
|
|
ServerStats.update(first_run=False).where(
|
|
ServerStats.server_id == self.server_id
|
|
).execute(self.database)
|
|
|
|
def get_first_run(self):
|
|
# self.select_database(self.server_id)
|
|
first_run = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.get(self.database)
|
|
)
|
|
return first_run.first_run
|
|
|
|
def get_ttl_without_player(self):
|
|
# self.select_database(self.server_id)
|
|
last_stat = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.order_by(ServerStats.created.desc())
|
|
.first(self.database)
|
|
)
|
|
last_stat_with_player = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.where(ServerStats.online > 0)
|
|
.order_by(ServerStats.created.desc())
|
|
.first(self.database)
|
|
)
|
|
return last_stat.created - last_stat_with_player.created
|
|
|
|
def can_stop_no_players(self, time_limit):
|
|
ttl_no_players = self.get_ttl_without_player()
|
|
return (time_limit == -1) or (ttl_no_players > time_limit)
|
|
|
|
def set_waiting_start(self, value):
|
|
# self.select_database(self.server_id)
|
|
try:
|
|
# Checks if server even exists
|
|
ServerStats.select().where(ServerStats.server_id == self.server_id).execute(
|
|
self.database
|
|
)
|
|
except DoesNotExist as ex:
|
|
logger.error(f"Database entry not found! {ex}")
|
|
return
|
|
ServerStats.update(waiting_start=value).where(
|
|
ServerStats.server_id == self.server_id
|
|
).execute(self.database)
|
|
|
|
def get_waiting_start(self):
|
|
waiting_start = (
|
|
ServerStats.select()
|
|
.where(ServerStats.server_id == self.server_id)
|
|
.get(self.database)
|
|
)
|
|
return waiting_start.waiting_start
|