Refactor Logger class message method to improve readability and add t…

…imer functionality

logmagix/logger.py

@@ -39,10 +39,8 @@ def warning(self, message: str, start: int = None, end: int = None, level: str =
 
     def message(self, level: str, message: str, start: int = None, end: int = None) -> None:
         time = self.get_time()
-        if start is not None and end is not None:
-            print(f"{self.prefix}[{self.MAGENTAA}{time}{self.PINK}] {self.PINK}[{self.CYAN}{level}{self.PINK}] -> {Fore.RESET} {self.CYAN}{message}{Fore.RESET} [{Fore.CYAN}{end - start}s{Style.RESET_ALL}]")
-        else:
-            print(f"{self.prefix}[{self.MAGENTAA}{time}{self.PINK}] [{Fore.BLUE}{level}{self.PINK}] -> {Fore.RESET} {self.CYAN}{message}{Fore.RESET}")
+        timer = f" {self.MAGENTAA}In{self.WHITE} -> {self.MAGENTAA}{str(end - start)[:5]} Seconds" if start and end else ""
+        print(f"{self.prefix}[{self.MAGENTAA}{time}{self.PINK}] [{self.CYAN}{level}{self.PINK}] -> [{self.CYAN}{message}{self.PINK}]{timer}")
 
     def message2(self, level: str, message: str, start: int = None, end: int = None) -> None:
         time = self.get_time()

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="logmagix",                   
-    version="1.0.5",                           
+    version="1.0.6",                           
     packages=find_packages(),               
     install_requires=["colorama"],           
     author="Sexfrance",                     

test.py

@@ -4,15 +4,18 @@
 import uuid
 
 log = Logger(prefix="custom/log/prefix")
+start_time = time.time()
 
 # Log messages
 log.success("Everything is running smoothly!")
 log.warning("Watch out, something might happen!")
 log.failure("Critical error occurred!")
 log.info("System is working properly")
 log.debug(f"The system uuid is {uuid.getnode()}")
+log.message("Dad", f"How are you? I'm gonna come soon!", start=start_time, end=time.time())
 log.question("How old are you? ")
 
+
 # Use loader with custom prefix and context manager
 with Loader(prefix="custom/loader/prefix", desc="Processing data..."):
     time.sleep(2)  # Simulate task
@@ -33,4 +36,5 @@
 
 home_screen.display()
 
+
 log.success("Processing completed!")