blackteay/ffmpeg-python
1
0
Fork 0
mirror of https://github.com/kkroening/ffmpeg-python.git synced 2025-04-06 04:15:44 +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 contextlib
import ffmpeg import ffmpeg
import gevent import gevent
import gevent.monkey; gevent.monkey.patch_all(thread=False) import gevent.monkey
import os import os
import shutil import shutil
import socket import socket
@ -62,6 +62,16 @@ def _do_watch_progress(filename, sock, handler):
connection.close() 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 @contextlib.contextmanager
def _watch_progress(handler): def _watch_progress(handler):
"""Context manager for creating a unix-domain socket and listen for """Context manager for creating a unix-domain socket and listen for
@ -78,7 +88,7 @@ def _watch_progress(handler):
Yields: Yields:
socket_filename: the name of the socket file. 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') socket_filename = os.path.join(tmpdir, 'sock')
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
with contextlib.closing(sock): with contextlib.closing(sock):