Add Graph.link_nodes

av/filter/graph.pyi

@@ -17,6 +17,7 @@ class Graph:
 
     def __init__(self) -> None: ...
     def configure(self, auto_buffer: bool = True, force: bool = False) -> None: ...
+    def link_nodes(self, *nodes: FilterContext) -> Graph: ...
     def add(
         self, filter: str | Filter, args: Any = None, **kwargs: str
     ) -> FilterContext: ...

av/filter/graph.pyx

@@ -45,6 +45,13 @@ cdef class Graph:
         # We get auto-inserted stuff here.
         self._auto_register()
 
+    def link_nodes(self, *nodes):
+        """
+        Links nodes together for simple filter graphs.
+        """
+        for c, n in zip(nodes, nodes[1:]):
+            c.link_to(n)
+        return self
 
     def add(self, filter, args=None, **kwargs):
         cdef Filter cy_filter
@@ -68,7 +75,7 @@ cdef class Graph:
 
         # There might have been automatic contexts added (e.g. resamplers,
         # fifos, and scalers). It is more likely to see them after the graph
-        # is configured, but we wan't to be safe.
+        # is configured, but we want to be safe.
         self._auto_register()
 
         return ctx

tests/test_filters.py

@@ -128,14 +128,9 @@ def test_audio_buffer_sink(self):
             if e.errno != errno.EAGAIN:
                 raise
 
-    @staticmethod
-    def link_nodes(*nodes):
-        for c, n in zip(nodes, nodes[1:]):
-            c.link_to(n)
-
     def test_audio_buffer_resample(self):
         graph = Graph()
-        self.link_nodes(
+        graph.link_nodes(
             graph.add_abuffer(
                 format="fltp",
                 sample_rate=48000,
@@ -146,8 +141,7 @@ def test_audio_buffer_resample(self):
                 "aformat", "sample_fmts=s16:sample_rates=44100:channel_layouts=stereo"
             ),
             graph.add("abuffersink"),
-        )
-        graph.configure()
+        ).configure()
 
         graph.push(
             generate_audio_frame(
@@ -161,7 +155,7 @@ def test_audio_buffer_resample(self):
 
     def test_audio_buffer_volume_filter(self):
         graph = Graph()
-        self.link_nodes(
+        graph.link_nodes(
             graph.add_abuffer(
                 format="fltp",
                 sample_rate=48000,
@@ -170,8 +164,7 @@ def test_audio_buffer_volume_filter(self):
             ),
             graph.add("volume", volume="0.5"),
             graph.add("abuffersink"),
-        )
-        graph.configure()
+        ).configure()
 
         input_frame = generate_audio_frame(
             0, input_format="fltp", layout="stereo", sample_rate=48000