added unified tool 2

Author: ThatSINEWAVE

BETA-Unified-Tool-2.py

@@ -0,0 +1,174 @@
+import json
+import time
+from datetime import datetime
+
+# Constants
+JSON_FILE_PATH = "../Database-Files/Edit-Database/Compromised-Discord-Accounts.json"
+
+
+# ======================
+# Utility Functions
+# ======================
+
+def print_header(title):
+    """Print a formatted header for tool sections."""
+    print(f"\n{'=' * 50}")
+    print(f"{title.upper():^50}")
+    print(f"{'=' * 50}\n")
+
+
+def log_message(message):
+    """Print formatted log messages with timestamp."""
+    print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] {message}")
+
+
+# ======================
+# Tool Functions
+# ======================
+
+def chronological_case_sorter():
+    """Sort cases chronologically by FOUND_ON date (Tool-02)."""
+    print_header("chronological case sorter")
+
+    # Load existing data from the JSON file
+    log_message("Loading JSON data...")
+    try:
+        with open(JSON_FILE_PATH, "r", encoding="utf-8") as file:
+            data = json.load(file)
+    except FileNotFoundError:
+        log_message("JSON file not found. Exiting.")
+        return
+    except Exception as e:
+        log_message(f"Error loading JSON file: {str(e)}")
+        return
+
+    # Convert data into a list of tuples (key, value) and sort by date
+    log_message("Sorting cases by date...")
+    date_count = 0
+    unknown_dates = 0
+
+    def parse_date(account):
+        nonlocal date_count, unknown_dates
+        date_str = account.get("FOUND_ON", "Unknown")
+        if date_str == "Unknown":
+            unknown_dates += 1
+            return datetime.min  # Assign minimum date for unknown values
+        try:
+            date_count += 1
+            return datetime.strptime(date_str, "%Y-%m-%d")
+        except ValueError:
+            unknown_dates += 1
+            return datetime.min
+
+    sorted_data = dict(sorted(data.items(), key=lambda item: parse_date(item[1])))
+
+    # Print sorting statistics
+    log_message(f"Total dates found: {date_count + unknown_dates}")
+    log_message(f"Dates sorted: {date_count}")
+    log_message(f"Unknown dates: {unknown_dates}")
+
+    # Write the sorted data back to the JSON file
+    log_message("Saving sorted JSON data...")
+    try:
+        with open(JSON_FILE_PATH, "w", encoding="utf-8") as file:
+            json.dump(sorted_data, file, indent=4, ensure_ascii=False)
+        log_message("Sorting complete.")
+    except Exception as e:
+        log_message(f"Error saving JSON file: {str(e)}")
+
+
+def case_number_normalizer():
+    """Normalize case numbers and account keys (Tool-03)."""
+    print_header("case number normalizer")
+
+    def get_current_timestamp():
+        return datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+    log_message(f"Starting to fix account numbers in the file: {JSON_FILE_PATH}")
+
+    try:
+        with open(JSON_FILE_PATH, "r", encoding="utf-8") as f:
+            data = json.load(f)
+        log_message("Successfully loaded the data from the file.")
+    except Exception as e:
+        log_message(f"Error loading the file: {e}")
+        return
+
+    total_accounts = len(data)
+    updated_count = 0
+    updated_data = {}
+
+    # Iterate through the accounts and update both the keys and the CASE_NUMBER
+    for index, (old_key, account) in enumerate(data.items(), start=1):
+        # Generate the new key (e.g., ACCOUNT_NUMBER_1, ACCOUNT_NUMBER_2, etc.)
+        new_key = f"ACCOUNT_NUMBER_{index}"
+
+        # Remove the 'ACCOUNT_NUMBER_' field if it exists
+        if "ACCOUNT_NUMBER_" in account:
+            del account["ACCOUNT_NUMBER_"]
+
+        # Update the 'CASE_NUMBER' field
+        if account["CASE_NUMBER"] != str(index):  # If the CASE_NUMBER is incorrect
+            updated_count += 1
+        account["CASE_NUMBER"] = str(index)
+
+        # Add the updated account data with the new key to the updated_data dictionary
+        updated_data[new_key] = account
+
+    try:
+        with open(JSON_FILE_PATH, "w", encoding="utf-8") as f:
+            json.dump(updated_data, f, indent=4, ensure_ascii=False)
+        log_message("Successfully saved the corrected data back to the file.")
+    except Exception as e:
+        log_message(f"Error saving the file: {e}")
+        return
+
+    log_message(f"Found {total_accounts} accounts in the file.")
+    log_message(f"{updated_count} accounts were updated.")
+
+
+# ======================
+# Main Menu
+# ======================
+
+def display_menu():
+    """Display the main menu and get user choice."""
+    print("\n" + "=" * 50)
+    print("CASE MANAGEMENT TOOL".center(50))
+    print("=" * 50)
+    print("\nMain Menu:")
+    print("1. Run All Tools (Full Processing)")
+    print("2. Chronological Case Sorter")
+    print("3. Case Number Normalizer")
+    print("4. Exit")
+
+    while True:
+        choice = input("\nEnter your choice (1-4): ")
+        if choice.isdigit() and 1 <= int(choice) <= 4:
+            return int(choice)
+        print("Invalid input. Please enter a number between 1 and 4.")
+
+
+def main():
+    """Main function to run the combined tool."""
+    while True:
+        choice = display_menu()
+
+        if choice == 1:  # Run all tools
+            print_header("running all tools")
+            chronological_case_sorter()
+            case_number_normalizer()
+            print_header("all tools completed")
+        elif choice == 2:
+            chronological_case_sorter()
+        elif choice == 3:
+            case_number_normalizer()
+        elif choice == 4:
+            print("\nExiting the program. Goodbye!")
+            break
+
+        input("\nPress Enter to return to the main menu...")
+
+
+if __name__ == "__main__":
+    main()