X-Git-Url: http://git.scottworley.com/srec/blobdiff_plain/00da1af9213f1f880aca532d80217e9f19e8708e..b3399b18f91a2f2e12ac9858f8f8e008a60138cd:/srec.py diff --git a/srec.py b/srec.py index 50f311c..8207c91 100644 --- a/srec.py +++ b/srec.py @@ -8,6 +8,7 @@ from dataclasses import dataclass from datetime import datetime import os import subprocess +import sys from typing import Any, Callable import gi @@ -19,18 +20,21 @@ from gi.repository import GLib # nopep8 pylint: disable=wrong-import-position @dataclass -class Recording: +class Stream: process: subprocess.Popen[bytes] def stop(self) -> None: stdin = self.process.stdin assert stdin is not None - stdin.write(b'q') - stdin.flush() + try: + stdin.write(b'q') + stdin.flush() + except BrokenPipeError: + print("Stream already stopped?", file=sys.stderr) self.process.wait() -recording: Recording | None = None +recording: Stream | None = None def make_filename() -> str: @@ -82,7 +86,7 @@ def on_start_recording(_: Gtk.Button, stack: Gtk.Stack) -> None: command = (['ffmpeg', '-framerate', '25'] + video_source(stack) + ['-f', 'pulse', '-ac', '2', '-i', 'default', filename]) # pylint: disable=consider-using-with - recording = Recording( + recording = Stream( process=subprocess.Popen(command, stdin=subprocess.PIPE)) stack.set_visible_child_name("recording")