NussNougate/nhentai
4
0
Fork 0
mirror of https://github.com/RicterZ/nhentai.git synced 2025-04-19 10:21:19 +02:00
Code Issues Packages Projects Releases Wiki Activity

global timeout

Browse Source
This commit is contained in:
ricterz 2015-05-09 14:00:03 +08:00
parent 0583c7910c
commit 7152ec2e1a
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
hentai/downloader.py
View File

@ -1,6 +1,7 @@
#coding: utf-8
import os
import sys
import socket
import threading
import Queue
import requests
@ -8,6 +9,11 @@ from urlparse import urlparse
from hentai.logger import logger
# global timeout
timeout = 30
socket.setdefaulttimeout(timeout)
class Downloader(object):
_instance = None
@ -34,7 +40,7 @@ class Downloader(object):
filename = filename if filename else os.path.basename(urlparse(url).path)
try:
with open(os.path.join(folder, filename), "wb") as f:
response = requests.get(url, stream=True)
response = requests.get(url, stream=True, timeout=10)
length = response.headers.get('content-length')
if length is None:
f.write(response.content)