Fix type of encoding since it could be None

plesk · May 27, 2024 · 281f7d5 · 281f7d5
1 parent 2dcb2c3
commit 281f7d5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pleskdistup/common/src/log.py b/pleskdistup/common/src/log.py
@@ -18,7 +18,7 @@ def _re_decode_message(message: str) -> str:
 
     @staticmethod
     def init_logger(logfiles: typing.List[str], streams: typing.List[typing.Any],
-                    console: bool = False, loglevel: int = logging.INFO, encoding: str = None) -> None:
+                    console: bool = False, loglevel: int = logging.INFO, encoding: typing.Optional[str] = None) -> None:
         if encoding is None:
             logger.encoding = locale.getpreferredencoding()
         else:
@@ -50,7 +50,7 @@ def init_logger(logfiles: typing.List[str], streams: typing.List[typing.Any],
 
     @staticmethod
     def reinit_logger(logfiles: typing.List[str], streams: typing.List[typing.Any],
-                      console: bool = False, loglevel: int = logging.INFO, encoding: str = None) -> None:
+                      console: bool = False, loglevel: int = logging.INFO, encoding: typing.Optional[str] = None) -> None:
         logger.files_logger = logging.getLogger("distupgrade_files")
         logger.streams_logger = logging.getLogger("distupgrade_streams")
         logger.init_logger(logfiles, streams, console, loglevel, encoding=encoding)
@@ -93,12 +93,12 @@ def err(msg: str, to_file: bool = True, to_stream: bool = True) -> None:
 
 
 def init_logger(logfiles: typing.List[str], streams: typing.List[typing.Any],
-                console: bool = False, loglevel: int = logging.INFO, encoding: str = None) -> None:
+                console: bool = False, loglevel: int = logging.INFO, encoding: typing.Optional[str] = None) -> None:
     logger.init_logger(logfiles, streams, console, loglevel, encoding=encoding)
 
 
 def reinit_logger(logfiles: typing.List[str], streams: typing.List[typing.Any],
-                  console: bool = False, loglevel: int = logging.INFO, encoding: str = None) -> None:
+                  console: bool = False, loglevel: int = logging.INFO, encoding: typing.Optional[str] = None) -> None:
     logger.reinit_logger(logfiles, streams, console, loglevel, encoding=encoding)