Delete Modules/submenu.py
This commit is contained in:
@ -1,185 +0,0 @@
|
|||||||
import sys # For exiting the program
|
|
||||||
from Modules.common_ui import *
|
|
||||||
from Modules.wrappers import *
|
|
||||||
|
|
||||||
|
|
||||||
# Global action status to persist across all submenus and their nested submenus
|
|
||||||
GLOBAL_ACTION_STATUS = {}
|
|
||||||
GLOBAL_TOTAL_PROGRESS = {"completed": 0, "total": 0} # Tracks progress across all sections
|
|
||||||
GLOBAL_HIGHLIGHTS = {} # Tracks highlighted actions
|
|
||||||
|
|
||||||
|
|
||||||
def build_submenu(menu_title, actions=None, module=None):
|
|
||||||
"""
|
|
||||||
Builds a submenu dynamically by extracting functions from the given module,
|
|
||||||
or using a provided `actions` dictionary for pre-defined submenus.
|
|
||||||
Includes an `[ALL]` option to execute all functions or submenus on the page.
|
|
||||||
Allows actions to be unmarked and highlighted.
|
|
||||||
"""
|
|
||||||
if not actions and not module:
|
|
||||||
raise ValueError("No module or actions provided to extract submenu items.")
|
|
||||||
|
|
||||||
# If actions are not provided, dynamically extract from the module
|
|
||||||
if not actions:
|
|
||||||
actions = {
|
|
||||||
str(idx + 1): {
|
|
||||||
"description": func.__name__.replace("_", " ").title(),
|
|
||||||
"function": func,
|
|
||||||
}
|
|
||||||
for idx, (name, func) in enumerate(module.items())
|
|
||||||
if (
|
|
||||||
callable(func)
|
|
||||||
and func.__module__ == module["__name__"] # Ensure the function is from this module
|
|
||||||
and "submenu" not in func.__name__ # Exclude functions with 'submenu' in their name
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
# Reassign sequential numbering starting at 1
|
|
||||||
numbered_actions = {str(idx + 1): action for idx, action in enumerate(actions.values())}
|
|
||||||
|
|
||||||
# Add the ALL option with key "ALL"
|
|
||||||
numbered_actions["ALL"] = {
|
|
||||||
"description": f"Run All {menu_title}",
|
|
||||||
"function": lambda: run_all_actions(numbered_actions, module, GLOBAL_ACTION_STATUS[menu_title]),
|
|
||||||
}
|
|
||||||
|
|
||||||
# Initialize global action status and highlights for this submenu
|
|
||||||
if menu_title not in GLOBAL_ACTION_STATUS:
|
|
||||||
GLOBAL_ACTION_STATUS[menu_title] = {key: False for key in numbered_actions}
|
|
||||||
GLOBAL_TOTAL_PROGRESS["total"] += len(numbered_actions) - 1 # Exclude the ALL option from the progress count
|
|
||||||
|
|
||||||
if menu_title not in GLOBAL_HIGHLIGHTS:
|
|
||||||
GLOBAL_HIGHLIGHTS[menu_title] = {key: False for key in numbered_actions}
|
|
||||||
|
|
||||||
action_status = GLOBAL_ACTION_STATUS[menu_title]
|
|
||||||
highlight_status = GLOBAL_HIGHLIGHTS[menu_title]
|
|
||||||
|
|
||||||
while True:
|
|
||||||
# Check if all individual actions are complete; if so, mark "ALL" as complete
|
|
||||||
action_status["ALL"] = all(status for key, status in action_status.items() if key != "ALL")
|
|
||||||
|
|
||||||
clear_screen()
|
|
||||||
header() # Display global app header
|
|
||||||
|
|
||||||
print("=" * 70)
|
|
||||||
print(f" {menu_title} Menu ")
|
|
||||||
print("=" * 70, "\n")
|
|
||||||
|
|
||||||
for key, action in numbered_actions.items():
|
|
||||||
# Determine the status symbol with color
|
|
||||||
status = f"{Fore.GREEN}✔️ " if action_status[key] else f"❌"
|
|
||||||
# Highlight the key and description based on highlight status
|
|
||||||
if key == "ALL":
|
|
||||||
print(f"{Fore.BLUE}[{key}]{Style.RESET_ALL} {status} {action['description']}{Style.RESET_ALL}")
|
|
||||||
elif highlight_status[key]:
|
|
||||||
print(f"{Style.RESET_ALL}[{key}] {Fore.YELLOW}💡 {action['description']}{Style.RESET_ALL}")
|
|
||||||
else:
|
|
||||||
print(f"{Style.RESET_ALL}[{key}] {status} {action['description']}")
|
|
||||||
|
|
||||||
print("[0] Return to Previous Menu")
|
|
||||||
print("[q] EXIT\n")
|
|
||||||
|
|
||||||
|
|
||||||
# Get user choice
|
|
||||||
choice = input("Enter your choice: ").strip().upper()
|
|
||||||
|
|
||||||
# Handle global commands
|
|
||||||
if choice in ["NOTE", "ABOUT"]:
|
|
||||||
global_command_handler(choice)
|
|
||||||
continue
|
|
||||||
|
|
||||||
# Handle highlighting (toggle with "!")
|
|
||||||
if choice.endswith("!") and choice[:-1] in numbered_actions:
|
|
||||||
key_to_toggle = choice[:-1]
|
|
||||||
highlight_status[key_to_toggle] = not highlight_status[key_to_toggle]
|
|
||||||
if highlight_status[key_to_toggle]:
|
|
||||||
print(f"{Fore.YELLOW}Action [{key_to_toggle}] has been highlighted.")
|
|
||||||
else:
|
|
||||||
print(f"{Fore.YELLOW}Highlight removed from action [{key_to_toggle}].")
|
|
||||||
input("Press Enter to continue...")
|
|
||||||
continue
|
|
||||||
|
|
||||||
# Handle unchecking of actions (with "-")
|
|
||||||
if choice.startswith("-") and choice[1:] in numbered_actions:
|
|
||||||
key_to_uncheck = choice[1:]
|
|
||||||
if action_status[key_to_uncheck]:
|
|
||||||
action_status[key_to_uncheck] = False
|
|
||||||
GLOBAL_TOTAL_PROGRESS["completed"] -= 1
|
|
||||||
print(f"{Fore.RED}Unchecked action [{key_to_uncheck}].")
|
|
||||||
else:
|
|
||||||
print(f"{Fore.RED}Action [{key_to_uncheck}] was already unchecked.")
|
|
||||||
input("Press Enter to continue...")
|
|
||||||
continue
|
|
||||||
|
|
||||||
if choice in numbered_actions:
|
|
||||||
if "function" in numbered_actions[choice]:
|
|
||||||
# Call the action function
|
|
||||||
numbered_actions[choice]["function"]()
|
|
||||||
if choice != "ALL" and not action_status[choice]: # Exclude ALL from progress tracking here
|
|
||||||
action_status[choice] = True
|
|
||||||
GLOBAL_TOTAL_PROGRESS["completed"] += 1
|
|
||||||
elif "submenu" in numbered_actions[choice]:
|
|
||||||
# Call the nested submenu
|
|
||||||
numbered_actions[choice]["submenu"]()
|
|
||||||
if is_submenu_complete(numbered_actions[choice]["submenu"]):
|
|
||||||
if not action_status[choice]:
|
|
||||||
action_status[choice] = True
|
|
||||||
GLOBAL_TOTAL_PROGRESS["completed"] += 1
|
|
||||||
elif choice == "0":
|
|
||||||
break
|
|
||||||
elif choice in ["Q", "EXIT", "QUIT"]:
|
|
||||||
sys.exit()
|
|
||||||
else:
|
|
||||||
print(f"{Fore.RED}Invalid choice. Press Enter to continue...")
|
|
||||||
input()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def run_all_actions(actions, module, action_status):
|
|
||||||
"""
|
|
||||||
Executes all actions in the submenu sequentially, excluding the "ALL" option itself.
|
|
||||||
"""
|
|
||||||
for key, action in actions.items():
|
|
||||||
if action["description"].startswith("Run All"): # Skip the "ALL" option
|
|
||||||
continue
|
|
||||||
|
|
||||||
if "function" in action:
|
|
||||||
if not action_status[key]:
|
|
||||||
action["function"]()
|
|
||||||
action_status[key] = True
|
|
||||||
GLOBAL_TOTAL_PROGRESS["completed"] += 1
|
|
||||||
elif "submenu" in action: # Recursively handle submenus
|
|
||||||
actions[key]["submenu"]()
|
|
||||||
if is_submenu_complete(actions[key]["submenu"]):
|
|
||||||
if not action_status[key]:
|
|
||||||
action_status[key] = True
|
|
||||||
GLOBAL_TOTAL_PROGRESS["completed"] += 1
|
|
||||||
|
|
||||||
|
|
||||||
def is_submenu_complete(submenu_function):
|
|
||||||
"""
|
|
||||||
Check if all actions in a submenu are complete.
|
|
||||||
"""
|
|
||||||
submenu_title = get_submenu_title(submenu_function)
|
|
||||||
if submenu_title in GLOBAL_ACTION_STATUS:
|
|
||||||
return all(GLOBAL_ACTION_STATUS[submenu_title].values())
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
def propagate_status_to_parent(menu_title):
|
|
||||||
"""
|
|
||||||
Propagates the status to the parent menu when all submenu items are completed.
|
|
||||||
"""
|
|
||||||
for parent_title, statuses in GLOBAL_ACTION_STATUS.items():
|
|
||||||
if menu_title in statuses:
|
|
||||||
statuses[menu_title] = all(
|
|
||||||
GLOBAL_ACTION_STATUS[menu_title].values()
|
|
||||||
) # Mark parent complete if all are done
|
|
||||||
|
|
||||||
|
|
||||||
def get_submenu_title(submenu_function):
|
|
||||||
"""
|
|
||||||
Converts a submenu function name to a readable submenu title.
|
|
||||||
"""
|
|
||||||
return submenu_function.__name__.replace("_submenu", "").replace("_", " ").title()
|
|
||||||
Reference in New Issue
Block a user