Added enumeratable device listing functions

tikuma-lsuhsc · tikuma-lsuhsc · commit 9e75772c53f0 · 2022-03-15T20:44:15.000-05:00
diff --git a/src/ffmpegio/devices.py b/src/ffmpegio/devices.py
@@ -82,12 +82,28 @@ def gather_device_info(dev_type, hook):
     SINKS = gather_device_info("sinks", "device_sink_api")
 
 
-    plugin_devices = {name: api for name, api in plugins.get_hook().device_source_api()}
-    SOURCES = {k: check_plugin(plugin_devices, k, v) for k, v in get_devices("sources")}
+def _list_devices(devs, mtype):
+    return [
+        dev
+        for dev, info in devs.items()
+        if "list" in info and any((k.startswith(mtype) for k in info["list"].keys()))
+    ]
 
-    plugin_devices = {name: api for name, api in plugins.get_hook().device_sink_api()}
-    SINKS = {k: check_plugin(plugin_devices, k, v) for k, v in get_devices("sinks")}
 
+def list_video_sources():
+    return _list_devices(SOURCES, "v")
+
+
+def list_audio_sources():
+    return _list_devices(SOURCES, "a")
+
+
+def list_video_sinks():
+    return _list_devices(SINKS, "v")
+
+
+def list_audio_sinks():
+    return _list_devices(SINKS, "a")
 
 def resolve_source(url, opts):
     try:
