functionalize get_videos

Author: scivision

demo.py

@@ -74,7 +74,9 @@ def mp_worker(i: int, Niter: int, tic: float):
     P = ArgumentParser(
         description="Demonstrate differences between coroutines, threads and proceses."
     )
-    P.add_argument("method", choices=["c", "t", "p"], help="c: coroutine, t: threading, p: multiprocessing")
+    P.add_argument(
+        "method", choices=["c", "t", "p"], help="c: coroutine, t: threading, p: multiprocessing"
+    )
     P.add_argument("-Nworker", help="number of workers", type=int, default=4)
     P.add_argument(
         "-Niter", help="number of loop iterations (arbitrary)", type=int, default=5000000,
@@ -90,9 +92,7 @@ def mp_worker(i: int, Niter: int, tic: float):
             t.start()
             ts.append(t)
         elif A.method == "p":
-            p = multiprocessing.Process(
-                target=mp_worker, args=(i, A.Niter, tic)
-            )
+            p = multiprocessing.Process(target=mp_worker, args=(i, A.Niter, tic))
             p.start()
             ps.append(p)
 

play_coroutine.py

@@ -1,9 +1,9 @@
-#!/usr/bin/env python
-from pathlib import Path
+#!/usr/bin/env python3
 from argparse import ArgumentParser
 
 import asyncioffmpeg.ffplay as play
 from asyncioffmpeg.runner import runner
+from asyncioffmpeg import get_videos
 
 
 if __name__ == "__main__":
@@ -17,10 +17,7 @@
     )
     P = p.parse_args()
 
-    path = Path(P.path).expanduser()
-    if not path.is_dir():
-        raise NotADirectoryError(path)
-
-    flist = (f for f in path.iterdir() if f.is_file() and f.suffix in P.suffix)
+    flist = get_videos(P.path, P.suffix)
+    print("found", len(flist), "files in", P.path)
 
     runner(play.main, flist)

play_sync.py renamed to play_queue.py

@@ -1,13 +1,14 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 """
 example of using queue with non-asyncio subprocess and threading
 """
-from pathlib import Path
+
 from argparse import ArgumentParser
 import queue
 import threading
 
 import asyncioffmpeg.ffplay as play
+from asyncioffmpeg import get_videos
 
 NTHREADS = 2
 
@@ -16,16 +17,12 @@
     p = ArgumentParser(description="Plays media files with FFplay")
     p.add_argument("path", help="directory where media files are kept")
     p.add_argument(
-        "-suffix",
-        help="file suffixes of desired media file types",
-        nargs="+",
-        default=[".mp4", ".avi", ".ogv", ".wmv", ".flv", ".mov"],
+        "-suffix", help="file suffixes of desired media file types", nargs="+",
     )
     P = p.parse_args()
 
-    path = Path(P.path).expanduser()
-
-    flist = (f for f in path.iterdir() if f.is_file() and f.suffix in P.suffix)
+    flist = get_videos(P.path, P.suffix)
+    print("found", len(flist), "files in", P.path)
 
     qin = queue.Queue()  # type: ignore
     for fn in flist:

play_threadpool.py

@@ -1,15 +1,23 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 from pathlib import Path
 import subprocess
 from argparse import ArgumentParser
 import concurrent.futures
-from asyncioffmpeg.ffplay import FFPLAY
+import shutil
+
+from asyncioffmpeg import get_videos
+
+EXE = shutil.which("ffplay")
+if not EXE:
+    raise FileNotFoundError("ffplay")
 
 
 def ffplay(filein: Path):
-    assert filein.is_file()
 
-    cmd = [FFPLAY, "-v", "warning", "-autoexit", str(filein)]
+    if not filein.is_file():
+        raise FileNotFoundError(filein)
+
+    cmd = [EXE, "-v", "warning", "-autoexit", str(filein)]
 
     subprocess.check_call(cmd)
 
@@ -18,18 +26,12 @@ def ffplay(filein: Path):
     p = ArgumentParser(description="Asynchronous playback with ThreadPool and FFplay")
     p.add_argument("path", help="directory where media files are kept")
     p.add_argument(
-        "-suffix",
-        help="file suffixes of desired media file types",
-        nargs="+",
-        default=[".mp4", ".avi", ".ogv", ".wmv", ".flv", ".mov"],
+        "-suffix", help="file suffixes of desired media file types", nargs="+",
     )
     P = p.parse_args()
 
-    path = Path(P.path).expanduser()
-    if not path.is_dir():
-        raise FileNotFoundError(f"{path} is not a directory")
-
-    flist = (f for f in path.iterdir() if f.is_file() and f.suffix in P.suffix)
+    flist = get_videos(P.path, P.suffix)
+    print("found", len(flist), "files in", P.path)
 
     with concurrent.futures.ThreadPoolExecutor(max_workers=2, thread_name_prefix="ffplay") as pool:
         pool.map(ffplay, flist)

src/asyncioffmpeg/__init__.py

@@ -0,0 +1,18 @@
+from pathlib import Path
+import typing as T
+
+
+def get_videos(path: Path, suffixes: T.Sequence[str] = None) -> T.List[Path]:
+
+    if not suffixes:
+        suffixes = [".mp4", ".avi", ".ogv", ".wmv", ".flv", ".mov"]
+
+    path = Path(path).expanduser()
+
+    if not path.is_dir():
+        raise FileNotFoundError(f"{path} is not a directory")
+
+    flist = [f for f in path.iterdir() if f.is_file() and f.suffix in suffixes]
+    if len(flist) == 0:
+        raise FileNotFoundError(f"No files found in {path} with {suffixes}")
+    return flist