Improve subs in av/codec/codec and av/codec/context

av/codec/codec.pyi

@@ -2,9 +2,12 @@ from fractions import Fraction
 from typing import Literal
 
 from av.audio.format import AudioFormat
+from av.descriptor import Descriptor
 from av.enum import EnumFlag
 from av.video.format import VideoFormat
 
+from .context import CodecContext
+
 class Properties(EnumFlag):
     NONE: int
     INTRA_ONLY: int
@@ -41,15 +44,20 @@ class UnknownCodecError(ValueError): ...
 
 class Codec:
     is_decoder: bool
+    descriptor: Descriptor
     name: str
     long_name: str
     type: Literal["video", "audio", "data", "subtitle", "attachment"]
+    id: int
     frame_rates: list[Fraction] | None
     audio_rates: list[int] | None
     video_formats: list[VideoFormat] | None
     audio_formats: list[AudioFormat] | None
+    properties: Properties
+    capabilities: Capabilities
 
-    def __init__(self, name: str, mode: Literal["r", "w"]): ...
+    def __init__(self, name: str, mode: Literal["r", "w"]) -> None: ...
+    def create(self) -> CodecContext: ...
 
 class codec_descriptor:
     name: str

av/codec/context.pyi

@@ -1,9 +1,10 @@
-from typing import Any, Iterator, Literal
+from typing import Any, Literal
 
 from av.enum import EnumFlag, EnumItem
-from av.frame import Frame
 from av.packet import Packet
 
+from .codec import Codec
+
 class ThreadType(EnumFlag):
     NONE: int
     FRAME: int
@@ -71,7 +72,9 @@ class CodecContext:
     def open(self, strict: bool = True) -> None: ...
     def close(self, strict: bool = True) -> None: ...
     @staticmethod
-    def create(codec, mode=None) -> CodecContext: ...
+    def create(
+        codec: str | Codec, mode: Literal["r", "w"] | None = None
+    ) -> CodecContext: ...
     def parse(
         self, raw_input: bytes | bytearray | memoryview | None = None
     ) -> list[Packet]: ...