Upload files to "Modules"

This commit is contained in:
2024-11-26 08:05:34 -05:00
parent 7e5cb3ba31
commit 55c2f96ba8
2 changed files with 51 additions and 0 deletions

50
Modules/cli_handler.py Normal file
View File

@ -0,0 +1,50 @@
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

1
Modules/config.py Normal file
View File

@ -0,0 +1 @@
VERSION = "1.0.3"