blackteay/ffmpeg-python
1
0
Fork 0
mirror of https://github.com/kkroening/ffmpeg-python.git synced 2025-04-05 04:22:51 +08:00
Code Issues Projects Releases Wiki Activity

Fixed monkey patching in examples/show_progress.py

Browse Source
This commit is contained in:
David Singh 2024-03-29 22:58:29 +07:00
parent 6330535dab
commit 0e26e2dd29
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
examples/show_progress.py
View File

@ -5,7 +5,7 @@ import argparse
import contextlib
import ffmpeg
import gevent
import gevent.monkey; gevent.monkey.patch_all(thread=False)
import gevent.monkey
import os
import shutil
import socket
@ -62,6 +62,16 @@ def _do_watch_progress(filename, sock, handler):
connection.close()
class PatchSelectors():
def __enter__(self):
gevent.monkey.patch_selectors(aggressive=False)
def __exit__(self, *args, **kwargs):
import importlib
import selectors
importlib.reload(selectors)
@contextlib.contextmanager
def _watch_progress(handler):
"""Context manager for creating a unix-domain socket and listen for
@ -78,7 +88,7 @@ def _watch_progress(handler):
Yields:
socket_filename: the name of the socket file.
"""
with _tmpdir_scope() as tmpdir:
with _tmpdir_scope() as tmpdir, PatchSelectors():
socket_filename = os.path.join(tmpdir, 'sock')
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
with contextlib.closing(sock):