caffeinism-utils 0.0.168__py3-none-any.whl → 0.0.170__py3-none-any.whl

caffeinism_utils/aiteration.py

@@ -26,6 +26,8 @@ async def rate_limit_iterator(aiterator, iters_per_second):
 
 T = TypeVar("T")
 
+__CLOSE = object()
+
 
 class StreamQueue(Generic[T]):
     def __init__(self):
@@ -35,6 +37,9 @@ class StreamQueue(Generic[T]):
         return self.queue.put(data)
 
     async def __aiter__(self) -> AsyncGenerator[T]:
-        while (iterator := await self.queue.get()) is not None:
+        while (iterator := await self.queue.get()) is not __CLOSE:
             async for it in iterator:
                 yield it
+
+    async def close(self):
+        await self.queue.put(__CLOSE)

caffeinism_utils/av/io.py

@@ -388,10 +388,9 @@ class PyAVWriter(PyAVInterface):
             if self.future is not None:
                 self.future.result()
                 del self.future
-
             self.future = self.pool.submit(func, *args, **kwargs)
 
-        return func
+        return _func
 
     def write(self, array):
         frame = self.array_to_frame(array)
@@ -420,8 +419,15 @@ class PyAVWriter(PyAVInterface):
 
         return cc
 
-    def _encode_video_frame(self, stream: av.VideoStream, frame: av.VideoFrame):
+    def _encode_video_frame(self, stream: av.VideoStream, frame: av.VideoFrame | None):
         cc = self._codex_contexts.get(stream.index)
+        if frame is None:
+            if cc is not None:
+                for packet in cc.encode_lazy():
+                    packet.stream = stream
+                    yield packet
+            return
+
         if cc is None:
             stream.width = frame.width
             stream.height = frame.height
@@ -437,7 +443,7 @@ class PyAVWriter(PyAVInterface):
         elif (stream.width, stream.height) != (frame.width, frame.height):
             stream.width = frame.width
             stream.height = frame.height
-            for packet in cc.encode():
+            for packet in cc.encode_lazy():
                 packet.stream = stream
                 yield packet
             cc = self.create_codec_context(stream)
@@ -498,15 +504,16 @@ class PyAVWriter(PyAVInterface):
             del self.future
             self.future = None
 
-        if self.alpha_extractor is not None:
-            self.alpha_extractor.close()
         for stream in self.streams:
-            cc = self._codex_contexts.get(stream.index)
-            if cc is not None:
-                self.container.mux(cc.encode())
+            for packet in self._encode_video_frame(stream, None):
+                self.container.mux_one(packet)
+
         if self.audio_stream is not None:
             self.container.mux(self.audio_stream.encode())
 
+        if self.alpha_extractor is not None:
+            self.alpha_extractor.close()
+
     def __exit__(
         self,
         t: type[BaseException] | None,

caffeinism_utils/prefetch.py

@@ -103,7 +103,7 @@ async def _aprefetch_aiterator(
     while True:
         if prefetched is not None:
             try:
-                ret = await prefetched
+                ret = await asyncio.shield(prefetched)
             except StopAsyncIteration:
                 break
 

{caffeinism_utils-0.0.168.dist-info → caffeinism_utils-0.0.170.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: caffeinism-utils
-Version: 0.0.168
+Version: 0.0.170
 Summary: 
 Author: Kim Minjong
 Author-email: make.dirty.code@gmail.com

{caffeinism_utils-0.0.168.dist-info → caffeinism_utils-0.0.170.dist-info}/RECORD

@@ -1,16 +1,16 @@
 caffeinism_utils/__init__.py,sha256=MER0q-NVosxQWibvx3a3bKwxRWGVXWeso9-qaU6oH_E,43
-caffeinism_utils/aiteration.py,sha256=uVMBXj27oUWNU6nsAv6VrQaDGj8krroCblGdl-gnxSU,1041
+caffeinism_utils/aiteration.py,sha256=PD2PMU4rkjz2xlVDyBiNd19_-2zBpF3KWrIGS3QNl_c,1130
 caffeinism_utils/asyncio.py,sha256=33yJ51vY0JvS6HJiBsEXXxAE3UlpLeudK8tanlu599Y,4811
 caffeinism_utils/av/__init__.py,sha256=J7dtpfW9R9BH-HJmlubAaFMqcEOMOu--zH64XJ7A2hI,423
 caffeinism_utils/av/codecs.py,sha256=gkOeazEqJWC1fRTLGpjdQbeorpVZpRHDzN7Ky01AML8,3360
 caffeinism_utils/av/filters.py,sha256=P2cMPTOMTU-oDKkqgPv-5YGO0JPFpMdNVrYdRv1OD0w,15381
-caffeinism_utils/av/io.py,sha256=VlEbGkJZA1vt1dgxak27xSHs-LVEUYBD41iViTYxmt8,16531
+caffeinism_utils/av/io.py,sha256=RwlyYRRluDBn9JA6_Q_KChMELUZYzYqBJ1rFwp3hJXo,16724
 caffeinism_utils/http.py,sha256=ZDcdHxaSPzbo6Jy2RfWl6Ib5Hii2170AkXlb-KKuI4I,424
 caffeinism_utils/io.py,sha256=cgA7LwV8zgfEd3Qaubogow3V21IR4yp2drRmOgql6Tw,3771
 caffeinism_utils/iteration.py,sha256=7BddGudGRBmNyJi3cTpmcQ2zWplu5zjxYCN4J_cWe3c,3026
-caffeinism_utils/prefetch.py,sha256=e6RH78doS9YWCAeTJSbZi7avgRsbMLXt8lieNpzmS54,4128
+caffeinism_utils/prefetch.py,sha256=yNTbBaSF-BA5iA1zUfnWVvcMm3br5nlBmPgvBMXHO7I,4144
 caffeinism_utils/utils.py,sha256=-g0DDuiCrck4xjN9T5ZUaklClXqDLCfOB6PKN4D8mlE,927
 caffeinism_utils/zip.py,sha256=L1d8qfphao-iD5G7bDu4LMQNg6ULc4kCqUk5gJbhSdE,1679
-caffeinism_utils-0.0.168.dist-info/METADATA,sha256=AhZSaNIBeHN5KCSdcJodsKc1C6KaBBCOGq5RsgBsVjA,899
-caffeinism_utils-0.0.168.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
-caffeinism_utils-0.0.168.dist-info/RECORD,,
+caffeinism_utils-0.0.170.dist-info/METADATA,sha256=f0uoS1I0M3QiGiOm4c6n3K9X7SJJIuihVUcbBVcWMkQ,899
+caffeinism_utils-0.0.170.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+caffeinism_utils-0.0.170.dist-info/RECORD,,

{caffeinism_utils-0.0.168.dist-info → caffeinism_utils-0.0.170.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.2.0
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any