Cleanup

Author: savannahostrowski

src/fastapi_cli/utils/cli.py

@@ -12,10 +12,9 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.toolkit = get_rich_toolkit()
 
     def formatMessage(self, record: logging.LogRecord) -> str:
-        message = record.getMessage()
-        result = self.toolkit.print_as_string(message, tag=record.levelname)
+        result = self.toolkit.print_as_string(record.getMessage(), tag=record.levelname)
         # Prepend newline to fix alignment after ^C is printed by the terminal
-        if message == "Shutting down":
+        if record.message == "Shutting down":
             result = "\n" + result
         return result