logging-ext2 0.2.4__tar.gz → 0.3.0__tar.gz

{logging_ext2-0.2.4 → logging_ext2-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: logging-ext2
-Version: 0.2.4
+Version: 0.3.0
 Summary: extend python logging library
 Project-URL: Documentation, https://github.com/ramwin/logging-ext#readme
 Project-URL: Issues, https://github.com/ramwin/logging-ext/issues

{logging_ext2-0.2.4 → logging_ext2-0.3.0}/src/logging_ext2/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2025-present Xiang Wang <ramwin@qq.com>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.2.4"
+__version__ = "0.3.0"

{logging_ext2-0.2.4 → logging_ext2-0.3.0}/src/logging_ext2/handlers.py

@@ -6,6 +6,7 @@
 import datetime
 import subprocess
 import time
+from typing import List
 
 from filelock import FileLock
 
@@ -21,18 +22,19 @@ class TimedRotatingFileHandler(Handler):
     """
 
     def __init__(self, level=NOTSET, filename="log.log",
-                 datetime_formatter="%Y-%m-%d", max_keep=10, flat_keep=2):
+                 datetime_formatter="%Y-%m-%d", max_keep=10, flat_keep=2) -> None:
         """
         params:
             max_keep: how many files will the rotation keep
             flat_keep: how many files will stay in text mod, the other (max_keep-flat_keep) will be compressed using gzip
         """
-        self.filepath: Path = None
+        self.filepath: Path = None  # type: ignore[assignment]
         self.filename = Path(filename).name
         self.base_dir = Path(filename).parent
         self.datetime_formatter = datetime_formatter
         self.current_time_str = self.get_time_str()
         self.stream = self.init_stream()
+        assert self.filepath
         self.max_keep = max_keep
         self.flat_keep = flat_keep
         super().__init__(level)
@@ -55,9 +57,10 @@ class TimedRotatingFileHandler(Handler):
         new_time_str = self.get_time_str()
         return new_time_str != self.current_time_str
 
-    def do_rollover(self):
+    def do_rollover(self) -> None:
         if self.filepath.exists():
             self.stream.close()
+        self.stream = self.init_stream()
         paths: List[Path] = [
                 i
                 for i in self.base_dir.iterdir()
@@ -66,10 +69,19 @@ class TimedRotatingFileHandler(Handler):
         paths.sort(key=lambda x: x.name, reverse=True)
         to_delete: Path
         for to_delete in paths[self.max_keep:]:
+            if to_delete == self.filepath:
+                continue
             to_delete.unlink(missing_ok=True)
+
+    def gzip_file(self, paths) -> None:
+        """
+        TODO use async function to gzip file
+        """
         for to_gzip_path in paths[self.flat_keep:self.max_keep]:
             # here use the system gzip command to ignore exceptions like
             # permission denied or file not found
+            if to_gzip_path == self.filepath:
+                continue
             try:
                 if not to_gzip_path.stat().st_size:
                     to_gzip_path.unlink(missing_ok=True)
@@ -90,4 +102,3 @@ class TimedRotatingFileHandler(Handler):
                     ).with_suffix(".gz"))
                 cmds = ["gzip", to_gzip_path]
                 subprocess.run(cmds)
-        self.stream = self.init_stream()