Implement initial version of CLI logs (using rich)

Author: ArtyomVancyan

setup.cfg

@@ -35,6 +35,7 @@ install_requires =
     imageio-ffmpeg>=0.4.7
     imageio>=2.23.0
     pillow>=8.4.0
+    rich>=13.0.0
 python_requires = >=3.7
 package_dir =
     =src

src/thumbnails/progress.py

@@ -0,0 +1,21 @@
+from rich.live import Live
+from rich.status import Status
+
+
+class Progress:
+    def __init__(self, status, done="Process completed"):
+        self.status = Status(status)
+        self.live = Live(self.status)
+        self.done = done
+
+    def update(self, status):
+        self.status.update(status)
+        self.live.update(self.status)
+
+    def __enter__(self):
+        self.live.start()
+        return self
+
+    def __exit__(self, *_):
+        self.live.update("[white]%s ... [/white][green]done[/green]" % self.done)
+        self.live.stop()

src/thumbnails/thumbnail.py

@@ -12,6 +12,7 @@
 from .pathtools import ensure_tree
 from .pathtools import extract_name
 from .pathtools import metadata_path
+from .progress import Progress
 
 
 def register_thumbnail(typename):
@@ -45,7 +46,9 @@ def __init__(self, video, base, skip, output):
         self.thumbnail_dir = self.calc_thumbnail_dir()
         self.metadata_path = self._get_metadata_path()
         self._perform_skip()
-        self.extract_frames()
+
+        with Progress("Extracting frames ...", "Frames have been extracted"):
+            self.extract_frames()
 
     def _get_metadata_path(self):
         """Initiates the name of the thumbnail metadata file."""
@@ -100,12 +103,15 @@ def prepare_frames(self):
         master = Image.new(mode="RGBA", size=next(thumbnails))
         master_path = os.path.join(self.thumbnail_dir, extract_name(self.filepath) + ".png")
 
-        for frame, *_, x, y in thumbnails:
-            with Image.open(frame) as image:
-                image = image.resize((self.width, self.height), Image.ANTIALIAS)
-                master.paste(image, (x, y))
+        with Progress("Processing frames ...", "Frames have been merged into one") as progress:
+            for frame, *_, x, y in thumbnails:
+                progress.update("Processing '%s'" % frame)
+                with Image.open(frame) as image:
+                    image = image.resize((self.width, self.height), Image.ANTIALIAS)
+                    master.paste(image, (x, y))
 
-        master.save(master_path)
+            progress.update("Saving the result '%s'" % master_path)
+            master.save(master_path)
 
     def generate(self):
         def format_time(secs):
@@ -117,12 +123,14 @@ def format_time(secs):
         route = os.path.join(prefix, extract_name(self.filepath) + ".png")
         route = pathlib.Path(route).as_posix()
 
-        for _, start, end, x, y in self.thumbnails():
-            thumbnail_data = "%s --> %s\n%s#xywh=%d,%d,%d,%d\n\n" % (
-                format_time(start), format_time(end),
-                route, x, y, self.width, self.height,
-            )
-            metadata.append(thumbnail_data)
+        with Progress("Generating thumbnail metadata ...", "Thumbnail metadata has been generated") as progress:
+            for _, start, end, x, y in self.thumbnails():
+                progress.update("Generating metadata for '%s'" % route)
+                thumbnail_data = "%s --> %s\n%s#xywh=%d,%d,%d,%d\n\n" % (
+                    format_time(start), format_time(end),
+                    route, x, y, self.width, self.height,
+                )
+                metadata.append(thumbnail_data)
 
         with open(self.metadata_path, "w") as fp:
             fp.writelines(metadata)

src/thumbnails/video.py

@@ -10,6 +10,7 @@
 
 from .ffmpeg import _FFMpeg
 from .frame import _Frame
+from .progress import Progress
 
 ffmpeg_bin = get_ffmpeg_exe()
 
@@ -41,8 +42,9 @@ def __init__(self, filepath, compress, interval):
         self.__frames_count = None
         self.__columns = None
 
-        _FFMpeg.__init__(self, filepath)
-        _Frame.__init__(self, self.size)
+        with Progress("Parsing the video metadata ...", "Metadata has been parsed"):
+            _FFMpeg.__init__(self, filepath)
+            _Frame.__init__(self, self.size)
 
     @property
     def filepath(self):