Quick improvements and polishing\

Also improved __performance() method accuracy

Signed-off-by: Shahm Najeeb <Nirt_12023@outlook.com>

Author: DefinetlyNotAI

CODE/Logicytics.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import gc
 import os
 import shutil
 import subprocess
@@ -197,14 +198,15 @@ def __performance(self):
 
         execution_times = []
         memory_usage = []
-        process = psutil.Process()
+        process = psutil.Process(os.getpid())
 
         for file in range(len(self.execution_list)):
+            gc.collect()
             start_time = datetime.now()
-            start_memory = process.memory_info().rss / 1024 / 1024  # MB
+            start_memory = process.memory_full_info().uss / 1024 / 1024  # MB
             log.execution(Execute.script(self.execution_list[file]))
             end_time = datetime.now()
-            end_memory = process.memory_info().rss / 1024 / 1024  # MB
+            end_memory = process.memory_full_info().uss / 1024 / 1024  # MB
             elapsed_time = end_time - start_time
             memory_delta = end_memory - start_memory
             memory_usage.append((self.execution_list[file], str(memory_delta)))
@@ -510,8 +512,8 @@ def Logicytics():
     try:
         Logicytics()
     except KeyboardInterrupt:
-        log.warning("Shutting down Logicytics utility with force causes many files to remain where they shouldn't")
-        log.warning("Please don't force shut Logicytics again - As we don't have a cleanup function yet.")
+        log.warning("⚠️ Force shutdown detected! Some temporary files might be left behind.")
+        log.warning("💡 Pro tip: Next time, let the program finish naturally.")
         # TODO v3.4.2 -> Cleanup function
         exit(0)
 else:

CODE/logicytics/Get.py

@@ -1,37 +1,10 @@
 from __future__ import annotations
 
-import functools
 import os
-import time
-
-
-def time_execution(func):
-    """
-    A decorator that logs the execution time of a function.
-
-    Usage:
-    @time_execution
-    def my_function():
-        ...
-
-    Or:
-    time_execution(my_function, *args, **kwargs)
-    """
-
-    @functools.wraps(func)
-    def wrapper(*args, **kwargs):
-        start_time = time.time()
-        result = func(*args, **kwargs)
-        end_time = time.time()
-        print(f"Function '{func.__name__}' executed in {end_time - start_time:.6f} seconds")
-        return result
-
-    return wrapper
 
 
 class Get:
     @staticmethod
-    @time_execution
     def list_of_files(
             directory: str,
             only_extensions: list[str] = None,

CODE/logicytics/__init__.py

@@ -25,7 +25,17 @@
 class ObjectLoadError(Exception):
     """Raised when an Object fails to load."""
 
-    def __init__(self, message="Failed to load object"):
+    def __init__(self, message="Failed to load object", object_name=None):
+        """
+        Initialize the exception with a custom message and object details.
+
+        Args:
+            message (str): The error message
+            object_name (str, optional): Name of the object that failed to load
+        """
+        self.object_name = object_name
+        if object_name:
+            message = f"{message} (Object: {object_name})"
         super().__init__(message)