WIP: more override decorators

pyinfra_cli/exceptions.py

@@ -2,6 +2,7 @@
 import sys
 from inspect import getframeinfo
 from traceback import format_exception, format_tb, walk_tb
+from typing import override
 from types import TracebackType
 
 import click
@@ -37,6 +38,7 @@ def __init__(self, e: Exception):
             message = repr(message)
         self.message = message
 
+    @override
     def show(self, file=None):
         name = "unknown error"
 
@@ -65,6 +67,7 @@ def show(self, file=None):
 
 
 class CliError(click.ClickException):
+    @override
     def show(self, file=None):
         logger.warning(
             "--> {0}: {1}".format(
@@ -96,6 +99,7 @@ def __init__(self, e, filename):
         self.exception = e
         self.filename = filename
 
+    @override
     def show(self, file=None):
         logger.warning(
             "--> {0}:\n".format(
@@ -118,6 +122,7 @@ def __init__(self, e):
         e._traceback = traceback
         self.exception = e
 
+    @override
     def show(self, file=None):
         click.echo(
             "--> {0}:\n".format(

pyinfra_cli/log.py

@@ -1,12 +1,15 @@
 import logging
 
+from typing import override
+
 import click
 
 from pyinfra import logger, state
 from pyinfra.context import ctx_state
 
 
 class LogHandler(logging.Handler):
+    @override
     def emit(self, record):
         try:
             message = self.format(record)
@@ -25,6 +28,7 @@ class LogFormatter(logging.Formatter):
         logging.CRITICAL: lambda s: click.style(s, "red", bold=True),
     }
 
+    @override
     def format(self, record):
         message = record.msg