SantaSpeen/KuiToi-Server
1
0
Fork 0
mirror of https://github.com/kuitoi/kuitoi-Server.git synced 2025-08-18 00:35:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Non CMD mode.

Browse Source
This commit is contained in:
Maxim Khomutov 2023-07-25 19:28:49 +03:00
parent 6d01f0ef8d
commit f52c73ab76
1 changed files with 26 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/modules/ConsoleSystem/console_system.py
View File

@ -14,6 +14,7 @@ from prompt_toolkit import PromptSession, print_formatted_text, HTML
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
from prompt_toolkit.completion import NestedCompleter from prompt_toolkit.completion import NestedCompleter
from prompt_toolkit.history import FileHistory from prompt_toolkit.history import FileHistory
from prompt_toolkit.output.win32 import NoConsoleScreenBufferError
from prompt_toolkit.patch_stdout import patch_stdout from prompt_toolkit.patch_stdout import patch_stdout
from core import get_logger from core import get_logger
@ -28,6 +29,7 @@ class Console:
debug=False) -> None: debug=False) -> None:
self.__logger = get_logger("console") self.__logger = get_logger("console")
self.__is_run = False self.__is_run = False
self.no_cmd = False
self.__prompt_in = prompt_in self.__prompt_in = prompt_in
self.__prompt_out = prompt_out self.__prompt_out = prompt_out
self.__not_found = not_found self.__not_found = not_found
@ -128,10 +130,18 @@ class Console:
return self.__alias.copy() return self.__alias.copy()
def _write(self, t): def _write(self, t):
if self.no_cmd:
print(t)
return
try:
if t.startswith("html:"): if t.startswith("html:"):
print_formatted_text(HTML(t[5:])) print_formatted_text(HTML(t[5:]))
else: else:
print_formatted_text(t) print_formatted_text(t)
except NoConsoleScreenBufferError:
print("Works in non cmd mode.")
self.no_cmd = True
print(t)
def write(self, s: AnyStr): def write(self, s: AnyStr):
if isinstance(s, (list, tuple)): if isinstance(s, (list, tuple)):
@ -202,11 +212,18 @@ class Console:
while True: while True:
try: try:
with patch_stdout(): with patch_stdout():
if self.no_cmd:
cmd_in = input(self.__prompt_in)
else:
try:
cmd_in = await session.prompt_async( cmd_in = await session.prompt_async(
self.__prompt_in, self.__prompt_in,
completer=self.completer, completer=self.completer,
auto_suggest=AutoSuggestFromHistory() auto_suggest=AutoSuggestFromHistory()
) )
except NoConsoleScreenBufferError:
print("Works in non cmd mode.")
self.no_cmd = True
cmd_s = cmd_in.split(" ") cmd_s = cmd_in.split(" ")
cmd = cmd_s[0] cmd = cmd_s[0]
if cmd == "": if cmd == "":