mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
171 lines
6.1 KiB
Python
171 lines
6.1 KiB
Python
import logging
|
|
import datetime
|
|
import typing as t
|
|
from peewee import (
|
|
CharField,
|
|
AutoField,
|
|
DateTimeField,
|
|
BooleanField,
|
|
IntegerField,
|
|
)
|
|
from playhouse.shortcuts import model_to_dict
|
|
|
|
from app.classes.shared.main_models import DatabaseShortcuts
|
|
from app.classes.models.base_model import BaseModel
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# **********************************************************************************
|
|
# Servers Class
|
|
# **********************************************************************************
|
|
class Servers(BaseModel):
|
|
server_id = AutoField()
|
|
created = DateTimeField(default=datetime.datetime.now)
|
|
server_uuid = CharField(default="", index=True)
|
|
server_name = CharField(default="Server", index=True)
|
|
path = CharField(default="")
|
|
backup_path = CharField(default="")
|
|
executable = CharField(default="")
|
|
log_path = CharField(default="")
|
|
execution_command = CharField(default="")
|
|
auto_start = BooleanField(default=0)
|
|
auto_start_delay = IntegerField(default=10)
|
|
crash_detection = BooleanField(default=0)
|
|
stop_command = CharField(default="stop")
|
|
executable_update_url = CharField(default="")
|
|
server_ip = CharField(default="127.0.0.1")
|
|
server_port = IntegerField(default=25565)
|
|
logs_delete_after = IntegerField(default=0)
|
|
type = CharField(default="minecraft-java")
|
|
|
|
class Meta:
|
|
table_name = "servers"
|
|
|
|
|
|
# **********************************************************************************
|
|
# Servers Class
|
|
# **********************************************************************************
|
|
class HelperServers:
|
|
def __init__(self, database):
|
|
self.database = database
|
|
|
|
# **********************************************************************************
|
|
# Generic Servers Methods
|
|
# **********************************************************************************
|
|
@staticmethod
|
|
def create_server(
|
|
name: str,
|
|
server_uuid: str,
|
|
server_dir: str,
|
|
backup_path: str,
|
|
server_command: str,
|
|
server_file: str,
|
|
server_log_file: str,
|
|
server_stop: str,
|
|
server_type: str,
|
|
server_port: int = 25565,
|
|
server_host: str = "127.0.0.1",
|
|
) -> int:
|
|
"""Create a server in the database
|
|
|
|
Args:
|
|
name: The name of the server
|
|
server_uuid: This is the UUID of the server
|
|
server_dir: The directory where the server is located
|
|
backup_path: The path to the backup folder
|
|
server_command: The command to start the server
|
|
server_file: The name of the server file
|
|
server_log_file: The path to the server log file
|
|
server_stop: This is the command to stop the server
|
|
server_type: This is the type of server you're creating.
|
|
server_port: The port the server will be monitored on, defaults to 25565
|
|
server_host: The host the server will be monitored on, defaults to 127.0.0.1
|
|
|
|
Returns:
|
|
int: The new server's id
|
|
|
|
Raises:
|
|
PeeweeException: If the server already exists
|
|
"""
|
|
return Servers.insert(
|
|
{
|
|
Servers.server_name: name,
|
|
Servers.server_uuid: server_uuid,
|
|
Servers.path: server_dir,
|
|
Servers.executable: server_file,
|
|
Servers.execution_command: server_command,
|
|
Servers.auto_start: False,
|
|
Servers.auto_start_delay: 10,
|
|
Servers.crash_detection: False,
|
|
Servers.log_path: server_log_file,
|
|
Servers.server_port: server_port,
|
|
Servers.server_ip: server_host,
|
|
Servers.stop_command: server_stop,
|
|
Servers.backup_path: backup_path,
|
|
Servers.type: server_type,
|
|
}
|
|
).execute()
|
|
|
|
@staticmethod
|
|
def get_server_obj(server_id):
|
|
return Servers.get_by_id(server_id)
|
|
|
|
@staticmethod
|
|
def get_server_type_by_id(server_id):
|
|
server_type = Servers.select().where(Servers.server_id == server_id).get()
|
|
return server_type.type
|
|
|
|
@staticmethod
|
|
def update_server(server_obj):
|
|
return server_obj.save()
|
|
|
|
def remove_server(self, server_id):
|
|
Servers.delete().where(Servers.server_id == server_id).execute()
|
|
|
|
@staticmethod
|
|
def get_server_data_by_id(server_id):
|
|
query = Servers.select().where(Servers.server_id == server_id).limit(1)
|
|
try:
|
|
return DatabaseShortcuts.return_rows(query)[0]
|
|
except IndexError:
|
|
return {}
|
|
|
|
@staticmethod
|
|
def get_server_columns(
|
|
server_id: t.Union[str, int], column_names: t.List[str]
|
|
) -> t.List[t.Any]:
|
|
columns = [getattr(Servers, column) for column in column_names]
|
|
return model_to_dict(
|
|
Servers.select(*columns).where(Servers.server_id == server_id).get(),
|
|
only=columns,
|
|
)
|
|
|
|
@staticmethod
|
|
def get_server_column(server_id: t.Union[str, int], column_name: str) -> t.Any:
|
|
column = getattr(Servers, column_name)
|
|
return model_to_dict(
|
|
Servers.select(column).where(Servers.server_id == server_id).get(),
|
|
only=[column],
|
|
)[column_name]
|
|
|
|
# **********************************************************************************
|
|
# Servers Methods
|
|
# **********************************************************************************
|
|
@staticmethod
|
|
def get_all_defined_servers():
|
|
query = Servers.select()
|
|
return DatabaseShortcuts.return_rows(query)
|
|
|
|
@staticmethod
|
|
def get_all_server_ids() -> t.List[int]:
|
|
return [server.server_id for server in Servers.select(Servers.server_id)]
|
|
|
|
@staticmethod
|
|
def get_server_friendly_name(server_id):
|
|
server_data = HelperServers.get_server_data_by_id(server_id)
|
|
friendly_name = (
|
|
f"{server_data.get('server_name', None)} "
|
|
f"with ID: {server_data.get('server_id', 0)}"
|
|
)
|
|
return friendly_name
|