mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
import cmd
|
||
|
|
import time
|
||
|
|
|
||
|
|
import logging
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
from app.classes.shared.console import console
|
||
|
|
from app.classes.shared.helpers import helper
|
||
|
|
from app.classes.shared.tasks import tasks_manager
|
||
|
|
|
||
|
|
try:
|
||
|
|
import requests
|
||
|
|
|
||
|
|
except ModuleNotFoundError as e:
|
||
|
|
logger.critical("Import Error: Unable to load {} module".format(e, e.name))
|
||
|
|
console.critical("Import Error: Unable to load {} module".format(e, e.name))
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
class MainPrompt(cmd.Cmd):
|
||
|
|
|
||
|
|
# overrides the default Prompt
|
||
|
|
prompt = "Crafty Controller v{} > ".format(helper.get_version_string())
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def emptyline():
|
||
|
|
pass
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _clean_shutdown():
|
||
|
|
exit_file = os.path.join(helper.root_dir, "exit.txt")
|
||
|
|
try:
|
||
|
|
with open(exit_file, 'w') as f:
|
||
|
|
f.write("exit")
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
logger.critical("Unable to write exit file due to error: {}".format(e))
|
||
|
|
console.critical("Unable to write exit file due to error: {}".format(e))
|
||
|
|
|
||
|
|
def do_exit(self, line):
|
||
|
|
logger.info("Stopping all server daemons / threads")
|
||
|
|
console.info("Stopping all server daemons / threads")
|
||
|
|
self._clean_shutdown()
|
||
|
|
while True:
|
||
|
|
if 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")
|