fix VideoFromFile stream source to _ReentrantBytesIO for parallel async use (with pickle support)

Author: bigcat88

comfy_api/latest/_input_impl/video_types.py

@@ -22,8 +22,8 @@ def __init__(self, data: bytes):
         if data is None:
             raise TypeError("data must be bytes, not None")
         self._data = data
-        self._view = memoryview(data)
         self._pos = 0
+        self._len = len(data)
 
     def getvalue(self) -> bytes:
         if self.closed:
@@ -33,7 +33,7 @@ def getvalue(self) -> bytes:
     def getbuffer(self) -> memoryview:
         if self.closed:
             raise ValueError("I/O operation on closed file.")
-        return memoryview(self._data)  # return a NEW view; external .release() won't break our internal _view.
+        return memoryview(self._data)
 
     def readable(self) -> bool:
         return True
@@ -55,7 +55,7 @@ def seek(self, offset: int, whence: int = io.SEEK_SET) -> int:
         elif whence == io.SEEK_CUR:
             new_pos = self._pos + offset
         elif whence == io.SEEK_END:
-            new_pos = len(self._view) + offset
+            new_pos = self._len + offset
         else:
             raise ValueError(f"Invalid whence: {whence}")
         if new_pos < 0:
@@ -70,10 +70,10 @@ def readinto(self, b) -> int:
         if mv.readonly:
             raise TypeError("readinto() argument must be writable")
         mv = mv.cast("B")
-        if self._pos >= len(self._view):
+        if self._pos >= self._len:
             return 0
-        n = min(len(mv), len(self._view) - self._pos)
-        mv[:n] = self._view[self._pos:self._pos + n]
+        n = min(len(mv), self._len - self._pos)
+        mv[:n] = self._data[self._pos:self._pos + n]
         self._pos += n
         return n
 
@@ -84,10 +84,10 @@ def read(self, size: int = -1) -> bytes:
         if self.closed:
             raise ValueError("I/O operation on closed file.")
         if size is None or size < 0:
-            size = len(self._view) - self._pos
-        if self._pos >= len(self._view):
+            size = self._len - self._pos
+        if self._pos >= self._len:
             return b""
-        end = min(self._pos + size, len(self._view))
+        end = min(self._pos + size, self._len)
         out = self._data[self._pos:end]
         self._pos = end
         return out
@@ -98,9 +98,9 @@ def read1(self, size: int = -1) -> bytes:
     def readline(self, size: int = -1) -> bytes:
         if self.closed:
             raise ValueError("I/O operation on closed file.")
-        if self._pos >= len(self._view):
+        if self._pos >= self._len:
             return b""
-        end_limit = len(self._view) if size is None or size < 0 else min(len(self._view), self._pos + size)
+        end_limit = self._len if size is None or size < 0 else min(self._len, self._pos + size)
         nl = self._data.find(b"\n", self._pos, end_limit)
         end = (nl + 1) if nl != -1 else end_limit
         out = self._data[self._pos:end]
@@ -131,11 +131,6 @@ def writelines(self, lines) -> None:
     def truncate(self, size: int | None = None) -> int:
         raise io.UnsupportedOperation("not writable")
 
-    def close(self) -> None:
-        with suppress(Exception):
-            self._view.release()
-        super().close()
-
 
 def container_to_output_format(container_format: str | None) -> str | None:
     """