github/nzbToMedia
1
0
Fork 0
mirror of https://github.com/clinton-hall/nzbToMedia.git synced 2025-08-14 10:36:52 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix shutil.copyfileobj monkey patching

Browse source
This commit is contained in:
Labrys of Knossos 2019-01-05 17:02:03 -05:00
commit 5c644890e8
2 changed files with 13 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

11
core/utils/__init__.py
View file

@ -19,6 +19,7 @@ import subliminal
import core
from core import extractor, logger
from core.utils.download_info import get_download_info, update_download_info_status
from core.utils import shutil_custom
from core.utils.network import test_connection, wake_on_lan, wake_up
from core.utils.parsers import (
parse_args,
@ -47,15 +48,7 @@ except ImportError:
requests.packages.urllib3.disable_warnings()
# Monkey Patch shutil.copyfileobj() to adjust the buffer length to 512KB rather than 4KB
shutil.copyfileobjOrig = shutil.copyfileobj
def copyfileobj_fast(fsrc, fdst, length=512 * 1024):
shutil.copyfileobjOrig(fsrc, fdst, length=length)
shutil.copyfileobj = copyfileobj_fast
shutil_custom.monkey_patch()
def sanitize_name(name):

11
core/utils/shutil_custom.py Normal file
View file

@ -0,0 +1,11 @@
from functools import partial
import shutil
from six import PY2
def monkey_patch(length=512 * 1024):
if PY2:
# On Python 2 monkey patch shutil.copyfileobj()
# to adjust the buffer length to 512KB rather than 4KB
original_copyfileobj = shutil.copyfileobj
shutil.copyfileobj = partial(original_copyfileobj, length=length)