Delete Modules/cli_handler.py
This commit is contained in:
@ -1,50 +0,0 @@
|
|||||||
import argparse
|
|
||||||
from TTPs.menu import MENU_OPTIONS as BASE_MENU_OPTIONS
|
|
||||||
from Modules.Imports.all_imports import *
|
|
||||||
from Modules.config import *
|
|
||||||
|
|
||||||
def display_help():
|
|
||||||
help_text = """
|
|
||||||
Usage:
|
|
||||||
python3 dco.py Start the program
|
|
||||||
python3 dco.py -v Display the current version
|
|
||||||
python3 dco.py -h Show this help page
|
|
||||||
python3 dco.py -f <file> Add the last session so users can collaborate
|
|
||||||
"""
|
|
||||||
print(help_text)
|
|
||||||
|
|
||||||
def get_version():
|
|
||||||
"""Display the version of the application."""
|
|
||||||
print(f"DCO Version: {VERSION}")
|
|
||||||
|
|
||||||
def load_session(file_path):
|
|
||||||
"""Load a session file."""
|
|
||||||
try:
|
|
||||||
with open(file_path, "r") as file:
|
|
||||||
session_data = file.read()
|
|
||||||
print(f"Session loaded successfully from {file_path}")
|
|
||||||
print(session_data) # Optionally display the session data
|
|
||||||
except FileNotFoundError:
|
|
||||||
print(f"Error: File {file_path} not found.")
|
|
||||||
except Exception as e:
|
|
||||||
print(f"Error: {e}")
|
|
||||||
|
|
||||||
def handle_arguments():
|
|
||||||
"""Parse and handle command-line arguments."""
|
|
||||||
parser = argparse.ArgumentParser(
|
|
||||||
description="DCO - Threat AI Command Interface",
|
|
||||||
usage="python3 dco.py [-v | -f <file>]"
|
|
||||||
)
|
|
||||||
parser.add_argument("-v", "--version", action="store_true", help="Display the current version")
|
|
||||||
parser.add_argument("-f", "--file", type=str, help="Add the last session file for collaboration")
|
|
||||||
|
|
||||||
# Parse the arguments
|
|
||||||
args = parser.parse_args()
|
|
||||||
|
|
||||||
# Handle arguments
|
|
||||||
if args.version:
|
|
||||||
get_version()
|
|
||||||
exit(0) # Exit after displaying the version
|
|
||||||
elif args.file:
|
|
||||||
load_session(args.file)
|
|
||||||
exit(0) # Exit after loading the session
|
|
Reference in New Issue
Block a user