Improve type stub

av/__main__.py

@@ -7,13 +7,11 @@ def main():
     parser.add_argument("--version", action="store_true")
     args = parser.parse_args()
 
-    # ---
-
     if args.version:
         import av
         import av._core
 
-        print("PyAV v" + av.__version__)
+        print(f"PyAV v{av.__version__}")
 
         by_config = {}
         for libname, config in sorted(av._core.library_meta.items()):

av/codec/context.pyi

@@ -73,6 +73,3 @@ class CodecContext:
     @staticmethod
     def create(codec, mode=None) -> CodecContext: ...
     def parse(self, raw_input=None) -> list: ...
-    def encode(self, frame: Frame | None = None) -> list[Packet]: ...
-    def encode_lazy(self, frame: Frame | None = None) -> Iterator[Packet]: ...
-    def decode(self, packet: Packet | None = None) -> list[Frame]: ...

av/filter/__init__.pyi

@@ -1,2 +1,2 @@
-from .filter import Filter, filters_available
-from .graph import Graph
+from .filter import *
+from .graph import *

av/packet.pyi

@@ -14,5 +14,4 @@ class Packet:
     is_keyframe: bool
     is_corrupt: bool
 
-    def __init__(self, input=None): ...
-    def decode(self) -> list: ...
+    def __init__(self, input: int | None = None): ...

av/video/frame.pyi

@@ -1,7 +1,21 @@
 from typing import Any
 
+from PIL import Image
+
+from av.enum import EnumItem
+
 from .plane import VideoPlane
 
+class PictureType(EnumItem):
+    NONE: int
+    I: int
+    P: int
+    B: int
+    S: int
+    SI: int
+    SP: int
+    BI: int
+
 class VideoFrame:
     planes: tuple[VideoPlane]
     width: int
@@ -10,7 +24,8 @@ class VideoFrame:
     interlaced_frame: bool
     pict_type: Any
 
-    def from_image(self, img: Any) -> VideoFrame: ...
+    def from_image(self, img: Image.Image) -> VideoFrame: ...
+    def to_image(self, **kwargs) -> Image.Image: ...
     def __init__(
         self, name: str, width: int = 0, height: int = 0, format: str = "yuv420p"
     ): ...

av/video/stream.pyi

@@ -1,3 +1,8 @@
+from av.packet import Packet
 from av.stream import Stream
 
-class VideoStream(Stream): ...
+from .frame import VideoFrame
+
+class VideoStream(Stream):
+    def encode(self, frame: VideoFrame | None = None) -> list[Packet]: ...
+    def decode(self, packet: Packet | None = None) -> list[VideoFrame]: ...