mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2025-01-20 10:15:29 +01:00
91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
import sys
|
|
import cmd
|
|
import time
|
|
import threading
|
|
import logging
|
|
|
|
from app.classes.shared.console import console
|
|
from app.classes.shared.helpers import helper
|
|
from app.classes.shared.import3 import import3
|
|
from app.classes.web.websocket_helper import websocket_helper
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class MainPrompt(cmd.Cmd):
|
|
def __init__(self, tasks_manager, migration_manager):
|
|
super().__init__()
|
|
self.tasks_manager = tasks_manager
|
|
self.migration_manager = migration_manager
|
|
|
|
# overrides the default Prompt
|
|
prompt = f"Crafty Controller v{helper.get_version_string()} > "
|
|
|
|
# see MR !233 for pylint exemptino reason
|
|
@staticmethod
|
|
def emptyline(): # pylint: disable=arguments-differ
|
|
pass
|
|
|
|
def do_exit(self, _line):
|
|
self.tasks_manager._main_graceful_exit()
|
|
self.universal_exit()
|
|
|
|
def do_migrations(self, line):
|
|
if line == "up":
|
|
self.migration_manager.up()
|
|
elif line == "down":
|
|
self.migration_manager.down()
|
|
elif line == "done":
|
|
console.info(self.migration_manager.done)
|
|
elif line == "todo":
|
|
console.info(self.migration_manager.todo)
|
|
elif line == "diff":
|
|
console.info(self.migration_manager.diff)
|
|
elif line == "info":
|
|
console.info(f"Done: {self.migration_manager.done}")
|
|
console.info(f"FS: {self.migration_manager.todo}")
|
|
console.info(f"Todo: {self.migration_manager.diff}")
|
|
elif line.startswith("add "):
|
|
migration_name = line[len("add ") :]
|
|
self.migration_manager.create(migration_name, False)
|
|
else:
|
|
console.info("Unknown migration command")
|
|
|
|
@staticmethod
|
|
def do_threads(_line):
|
|
for thread in threading.enumerate():
|
|
if sys.version_info >= (3, 8):
|
|
print(
|
|
f"Name: {thread.name} Identifier: "
|
|
f"{thread.ident} TID/PID: {thread.native_id}"
|
|
)
|
|
else:
|
|
print(f"Name: {thread.name} Identifier: {thread.ident}")
|
|
|
|
def do_import3(self, _line):
|
|
import3.start_import()
|
|
|
|
def universal_exit(self):
|
|
logger.info("Stopping all server daemons / threads")
|
|
console.info(
|
|
"Stopping all server daemons / threads - This may take a few seconds"
|
|
)
|
|
websocket_helper.disconnect_all()
|
|
console.info("Waiting for main thread to stop")
|
|
while True:
|
|
if self.tasks_manager.get_main_thread_run_status():
|
|
sys.exit(0)
|
|
time.sleep(1)
|
|
|
|
@staticmethod
|
|
def help_exit():
|
|
console.help("Stops the server if running, Exits the program")
|
|
|
|
@staticmethod
|
|
def help_migrations():
|
|
console.help("Only for advanced users. Use with caution")
|
|
|
|
@staticmethod
|
|
def help_import3():
|
|
console.help("Import users and servers from Crafty 3")
|