Delete Modules/All_Pages/wrappers.py
This commit is contained in:
@ -1,83 +0,0 @@
|
|||||||
import subprocess
|
|
||||||
from colorama import Fore, Style
|
|
||||||
|
|
||||||
from Assets.ascii_text_prompts import ascii_art, full_ascii_art, infinitei
|
|
||||||
from Modules.All_Pages.clear_screen import clear_screen
|
|
||||||
from Modules.All_Pages.random_tip import get_random_tip_with_color
|
|
||||||
|
|
||||||
from Modules.All_Pages.center_text import *
|
|
||||||
|
|
||||||
from Modules.Login.scan import BACKGROUND_TASK
|
|
||||||
from Modules.Login.check_ip import get_target_ip
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def highlight_ports(port, open_ports): # Highlight a port if it is in the list of open ports.
|
|
||||||
return f"{Fore.GREEN}{port}{Style.RESET_ALL}" if port in open_ports else f"{port}"
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def display_background_task():
|
|
||||||
if BACKGROUND_TASK:
|
|
||||||
print("\nBackground Task:")
|
|
||||||
for task in BACKGROUND_TASK:
|
|
||||||
print(f" - {task}")
|
|
||||||
else:
|
|
||||||
print("\nBackground Task: None")
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def header(target_ip, open_ports):
|
|
||||||
clear_screen()
|
|
||||||
print(center_text(ascii_art))
|
|
||||||
print(center_text(get_random_tip_with_color()) + "\n")
|
|
||||||
print(center_text(f"Target IP: {target_ip}\n"))
|
|
||||||
if open_ports:
|
|
||||||
print(center_text(f"Open Ports: {', '.join(open_ports)}\n"))
|
|
||||||
else:
|
|
||||||
print(center_text("Open Ports: None\n"))
|
|
||||||
display_background_task() # Include the background tasks section
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def display_menu(menu_options, open_ports, target_ip):
|
|
||||||
while True:
|
|
||||||
header(target_ip, open_ports) # Pass target_ip and open_ports to the header
|
|
||||||
print("\nStandard Enumeration")
|
|
||||||
for key, value in menu_options.items():
|
|
||||||
ports = "/".join([highlight_ports(port, open_ports) for port in value["ports"]])
|
|
||||||
print(f"[{key}] {value['name'].upper()}: Ports {ports}")
|
|
||||||
print("[0] Logout\n")
|
|
||||||
|
|
||||||
choice = input("Enter your choice: ").strip().lower()
|
|
||||||
|
|
||||||
# Match the choice to a menu option
|
|
||||||
if choice in menu_options:
|
|
||||||
menu_options[choice]["submenu"](target_ip, open_ports) # Pass arguments directly
|
|
||||||
elif choice in ["0", "q", "exit", "quit"]:
|
|
||||||
print("Logging out...")
|
|
||||||
break
|
|
||||||
else:
|
|
||||||
print("Invalid choice. Please try again.")
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def run_command(title, content, target_ip, open_ports):
|
|
||||||
header(target_ip, open_ports) # Display the header at the top
|
|
||||||
print("=" * 40)
|
|
||||||
print(f"{title.center(40)}")
|
|
||||||
print("=" * 40)
|
|
||||||
print("\n")
|
|
||||||
print("Executing Command:\n")
|
|
||||||
print(content)
|
|
||||||
print("\n")
|
|
||||||
|
|
||||||
try:
|
|
||||||
# Execute the command and display its output in real-time
|
|
||||||
subprocess.run(content, shell=True, check=True)
|
|
||||||
except subprocess.CalledProcessError as e:
|
|
||||||
print(f"Error executing command:\n{e}")
|
|
||||||
except Exception as ex:
|
|
||||||
print(f"An unexpected error occurred: {ex}")
|
|
||||||
finally:
|
|
||||||
input("\nPress Enter to return...")
|
|
Reference in New Issue
Block a user