Apr 3, 2025 · Apr 3, 2025 · Apr 3, 2025 · Apr 3, 2025 · Apr 3, 2025
diff --git a/pwncat/commands/__init__.py b/pwncat/commands/__init__.py
 import pkgutil
 import termios
 import argparse
 import importlib
 from io import TextIOWrapper
 from enum import Enum, auto
 from typing import Dict, List, Type, Callable, Iterable
        for loader, module_name, is_pkg in pkgutil.walk_packages(__path__):
            if module_name == "base":
                continue
            self.commands.append(
                loader.find_module(module_name)
                .load_module(module_name)
                .Command(manager)
            )
            full_module_name = f"{__name__}.{module_name}"
            module = importlib.import_module(full_module_name)
            self.commands.append(module.Command(manager))

        self.prompt: PromptSession = None
        self.toolbar: PromptSession = None
        self.loading_complete = False
        self.aliases: Dict[str, CommandDefinition] = {}
        self.shortcuts: Dict[str, CommandDefinition] = {}
        self.found_prefix: bool = False
        # Saved terminal state to support switching between raw and normal
        # mode.
        # Saved terminal state to support switching between raw and normal mode.
        self.saved_term_state = None


    def setup_prompt(self):
        """This needs to happen after __init__ when the database is fully
        initialized."""
Original file line number	Diff line number	Diff line change
Expand Up		@@ -43,6 +43,7 @@ def run(self, manager: "pwncat.manager.Manager", args: "argparse.Namespace"):
		import pkgutil
		import termios
		import argparse
		import importlib
		from io import TextIOWrapper
		from enum import Enum, auto
		from typing import Dict, List, Type, Callable, Iterable
Expand DownExpand Up		@@ -432,22 +433,20 @@ def __init__(self, manager: "pwncat.manager.Manager"):
		for loader, module_name, is_pkg in pkgutil.walk_packages(__path__):
		if module_name == "base":
		continue
		self.commands.append(
		loader.find_module(module_name)
		.load_module(module_name)
		.Command(manager)
		)
		full_module_name = f"{__name__}.{module_name}"
		module = importlib.import_module(full_module_name)
		self.commands.append(module.Command(manager))

		self.prompt: PromptSession = None
		self.toolbar: PromptSession = None
		self.loading_complete = False
		self.aliases: Dict[str, CommandDefinition] = {}
		self.shortcuts: Dict[str, CommandDefinition] = {}
		self.found_prefix: bool = False
		# Saved terminal state to support switching between raw and normal
		# mode.
		# Saved terminal state to support switching between raw and normal mode.
		self.saved_term_state = None


		def setup_prompt(self):
		"""This needs to happen after __init__ when the database is fully
		initialized."""
Expand Down